summaryrefslogtreecommitdiff
path: root/misc/tvm/Makefile
blob: a5b5c70cead466deca55d1a1568a81968993d289 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39

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>