PORTNAME=	tvm
DISTVERSIONPREFIX=	v
DISTVERSION=	0.6.1
CATEGORIES=	misc # machine-learning

MAINTAINER=	yuri@FreeBSD.org
COMMENT=	Deep learning compiler stack for cpu, gpu and specialized accelerators

LICENSE=	APACHE20
LICENSE_FILE=	${WRKSRC}/LICENSE

BUILD_DEPENDS=	dmlc-core>0:devel/dmlc-core \
		rang>0:devel/rang

USES=		cmake compiler:c++11-lang localbase
USE_GITHUB=	yes
GH_ACCOUNT=	apache
GH_PROJECT=	incubator-tvm
GH_TUPLE=	dmlc:dlpack:0acb731:dlpack/3rdparty/dlpack
USE_LDCONFIG=	yes

CMAKE_ARGS=	-DDMLC_PATH=${LOCALBASE}/include -DRANG_PATH=${LOCALBASE}/include

LDFLAGS+=	/usr/lib/libexecinfo.so

OPTIONS_DEFINE=		LLVM COMPILER
OPTIONS_DEFAULT=	LLVM COMPILER
OPTIONS_SUB=		yes

LLVM_DESC=		With LLVM support for model compilation
LLVM_CMAKE_ON=		-DUSE_LLVM=${LOCALBASE}/bin/llvm-config${LLVM_DEFAULT}
LLVM_BUILD_DEPENDS=	llvm-config${LLVM_DEFAULT}:devel/llvm${LLVM_DEFAULT}
LLVM_RUN_DEPENDS=	llvm-config${LLVM_DEFAULT}:devel/llvm${LLVM_DEFAULT}

COMPILER_DESC=		Install compiler infrastructure (as opposed to runtime-only)
COMPILER_CMAKE_BOOL=	INSTALL_DEV

.include <bsd.port.mk>