From 4e93a15dd682391dc4ebc8cfcbe4f69aa4911fe9 Mon Sep 17 00:00:00 2001 From: Yuri Victorovich Date: Tue, 17 Dec 2019 06:07:09 +0000 Subject: [PATCH] misc/tvm: Add the COMPILER option enabling NN model compilation C++ API. --- misc/tvm/Makefile | 13 +++-- misc/tvm/pkg-plist | 121 ++++++++++++++++++++++++++++++++++++++++++++- 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 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