misc/tvm: Add the COMPILER option enabling NN model compilation C++ API.

2 files changed, 130 insertions, 4 deletions

diff --git a/misc/tvm/Makefile b/misc/tvm/Makefile
index c36d43709033..fcd90d4b5257 100644
--- a/misc/tvm/Makefile
+++ b/misc/tvm/Makefile
@@ -3,7 +3,7 @@
 PORTNAME=	tvm
 DISTVERSIONPREFIX=	v
 DISTVERSION=	0.6.0
-PORTREVISION=	1
+PORTREVISION=	2
 CATEGORIES=	misc # machine-learning
 
 MAINTAINER=	yuri@FreeBSD.org
@@ -25,12 +25,19 @@ USE_LDCONFIG=	yes
 
 LDFLAGS+=	/usr/lib/libexecinfo.so
 
-OPTIONS_DEFINE=		LLVM
-OPTIONS_DEFAULT=	LLVM
+OPTIONS_DEFINE=		LLVM COMPILER
+OPTIONS_DEFAULT=	LLVM COMPILER
+OPTIONS_SUB=		yes
 
 LLVM_DESC=		With LLVM support for model compillation
 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
+
+post-install-COMPILER-on: # delete headers that conflict with dmlc-core: https://github.com/apache/incubator-tvm/issues/4530
+	@${RM} -r ${STAGEDIR}${PREFIX}/include/dmlc
+
 .include <bsd.port.mk>
diff --git a/misc/tvm/pkg-plist b/misc/tvm/pkg-plist
index 456e37a0f7c8..398701549dc4 100644
--- a/misc/tvm/pkg-plist
+++ b/misc/tvm/pkg-plist
@@ -1,8 +1,122 @@
+%%COMPILER%%include/dlpack/dlpack.h
+%%COMPILER%%include/nnvm/base.h
+%%COMPILER%%include/nnvm/c_api.h
+%%COMPILER%%include/nnvm/compiler/op_attr_types.h
+%%COMPILER%%include/nnvm/compiler/packed_func_ext.h
+%%COMPILER%%include/nnvm/compiler/util.h
+%%COMPILER%%include/nnvm/graph.h
+%%COMPILER%%include/nnvm/graph_attr_types.h
+%%COMPILER%%include/nnvm/layout.h
+%%COMPILER%%include/nnvm/node.h
+%%COMPILER%%include/nnvm/op.h
+%%COMPILER%%include/nnvm/op_attr_types.h
+%%COMPILER%%include/nnvm/pass.h
+%%COMPILER%%include/nnvm/pass_functions.h
+%%COMPILER%%include/nnvm/symbolic.h
+%%COMPILER%%include/nnvm/top/nn.h
+%%COMPILER%%include/nnvm/top/tensor.h
+%%COMPILER%%include/nnvm/tuple.h
+%%COMPILER%%include/topi/broadcast.h
+%%COMPILER%%include/topi/contrib/cublas.h
+%%COMPILER%%include/topi/contrib/rocblas.h
+%%COMPILER%%include/topi/cuda/dense.h
+%%COMPILER%%include/topi/cuda/injective.h
+%%COMPILER%%include/topi/cuda/normalization.h
+%%COMPILER%%include/topi/cuda/pooling.h
+%%COMPILER%%include/topi/cuda/reduction.h
+%%COMPILER%%include/topi/cuda/softmax.h
+%%COMPILER%%include/topi/detail/array_utils.h
+%%COMPILER%%include/topi/detail/broadcast.h
+%%COMPILER%%include/topi/detail/constant_utils.h
+%%COMPILER%%include/topi/detail/extern.h
+%%COMPILER%%include/topi/detail/fuse.h
+%%COMPILER%%include/topi/detail/pad_utils.h
+%%COMPILER%%include/topi/detail/ravel_unravel.h
+%%COMPILER%%include/topi/elemwise.h
+%%COMPILER%%include/topi/generic/default.h
+%%COMPILER%%include/topi/generic/extern.h
+%%COMPILER%%include/topi/generic/injective.h
+%%COMPILER%%include/topi/image/resize.h
+%%COMPILER%%include/topi/nn.h
+%%COMPILER%%include/topi/nn/batch_matmul.h
+%%COMPILER%%include/topi/nn/bias_add.h
+%%COMPILER%%include/topi/nn/bnn.h
+%%COMPILER%%include/topi/nn/dense.h
+%%COMPILER%%include/topi/nn/dilate.h
+%%COMPILER%%include/topi/nn/flatten.h
+%%COMPILER%%include/topi/nn/l2_normalize.h
+%%COMPILER%%include/topi/nn/local_response_norm.h
+%%COMPILER%%include/topi/nn/mapping.h
+%%COMPILER%%include/topi/nn/pooling.h
+%%COMPILER%%include/topi/nn/softmax.h
+%%COMPILER%%include/topi/nn/upsampling.h
+%%COMPILER%%include/topi/reduction.h
+%%COMPILER%%include/topi/rocm/dense.h
+%%COMPILER%%include/topi/rocm/normalization.h
+%%COMPILER%%include/topi/tags.h
+%%COMPILER%%include/topi/transform.h
+%%COMPILER%%include/topi/vision/reorg.h
+%%COMPILER%%include/topi/x86/bnn.h
+%%COMPILER%%include/topi/x86/default.h
+%%COMPILER%%include/topi/x86/injective.h
+%%COMPILER%%include/tvm/api_registry.h
+%%COMPILER%%include/tvm/arithmetic.h
+%%COMPILER%%include/tvm/attrs.h
+%%COMPILER%%include/tvm/base.h
+%%COMPILER%%include/tvm/buffer.h
+%%COMPILER%%include/tvm/build_module.h
+%%COMPILER%%include/tvm/channel.h
+%%COMPILER%%include/tvm/codegen.h
+%%COMPILER%%include/tvm/data_layout.h
+%%COMPILER%%include/tvm/dtype.h
+%%COMPILER%%include/tvm/expr.h
+%%COMPILER%%include/tvm/expr_operator.h
+%%COMPILER%%include/tvm/ir.h
+%%COMPILER%%include/tvm/ir_functor_ext.h
+%%COMPILER%%include/tvm/ir_mutator.h
+%%COMPILER%%include/tvm/ir_pass.h
+%%COMPILER%%include/tvm/ir_visitor.h
+%%COMPILER%%include/tvm/logging.h
+%%COMPILER%%include/tvm/lowered_func.h
+%%COMPILER%%include/tvm/node/container.h
+%%COMPILER%%include/tvm/node/functor.h
+%%COMPILER%%include/tvm/node/node.h
+%%COMPILER%%include/tvm/node/reflection.h
+%%COMPILER%%include/tvm/node/serialization.h
+%%COMPILER%%include/tvm/operation.h
+%%COMPILER%%include/tvm/packed_func_ext.h
+%%COMPILER%%include/tvm/relay/adt.h
+%%COMPILER%%include/tvm/relay/analysis.h
+%%COMPILER%%include/tvm/relay/attrs/algorithm.h
+%%COMPILER%%include/tvm/relay/attrs/annotation.h
+%%COMPILER%%include/tvm/relay/attrs/bitserial.h
+%%COMPILER%%include/tvm/relay/attrs/debug.h
+%%COMPILER%%include/tvm/relay/attrs/device_copy.h
+%%COMPILER%%include/tvm/relay/attrs/image.h
+%%COMPILER%%include/tvm/relay/attrs/memory.h
+%%COMPILER%%include/tvm/relay/attrs/nn.h
+%%COMPILER%%include/tvm/relay/attrs/reduce.h
+%%COMPILER%%include/tvm/relay/attrs/transform.h
+%%COMPILER%%include/tvm/relay/attrs/vision.h
+%%COMPILER%%include/tvm/relay/base.h
+%%COMPILER%%include/tvm/relay/error.h
+%%COMPILER%%include/tvm/relay/expr.h
+%%COMPILER%%include/tvm/relay/expr_functor.h
+%%COMPILER%%include/tvm/relay/feature.h
+%%COMPILER%%include/tvm/relay/interpreter.h
+%%COMPILER%%include/tvm/relay/module.h
+%%COMPILER%%include/tvm/relay/op.h
+%%COMPILER%%include/tvm/relay/op_attr_types.h
+%%COMPILER%%include/tvm/relay/pattern_functor.h
+%%COMPILER%%include/tvm/relay/qnn/attrs.h
+%%COMPILER%%include/tvm/relay/qnn/transform.h
+%%COMPILER%%include/tvm/relay/transform.h
+%%COMPILER%%include/tvm/relay/type.h
 include/tvm/runtime/c_backend_api.h
 include/tvm/runtime/c_runtime_api.h
 include/tvm/runtime/device_api.h
-include/tvm/runtime/micro/standalone/utvm_runtime.h
 include/tvm/runtime/memory.h
+include/tvm/runtime/micro/standalone/utvm_runtime.h
 include/tvm/runtime/module.h
 include/tvm/runtime/ndarray.h
 include/tvm/runtime/object.h
@@ -12,6 +126,11 @@ include/tvm/runtime/serializer.h
 include/tvm/runtime/threading_backend.h
 include/tvm/runtime/util.h
 include/tvm/runtime/vm.h
+%%COMPILER%%include/tvm/schedule.h
+%%COMPILER%%include/tvm/schedule_pass.h
+%%COMPILER%%include/tvm/target_info.h
+%%COMPILER%%include/tvm/tensor.h
+%%COMPILER%%include/tvm/tensor_intrin.h
 lib/libnnvm_compiler.so
 lib/libtvm.so
 lib/libtvm_runtime.so