From d7354628596ca5d159f2808ca623f440ce31af8c Mon Sep 17 00:00:00 2001 From: Tianqi Chen <tqchen@users.noreply.github.com> Date: Mon, 13 Nov 2017 21:25:10 -0800 Subject: [PATCH] [APP] improve parameter pack (#645) --- apps/howto_deploy/tvm_runtime_pack.cc | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/apps/howto_deploy/tvm_runtime_pack.cc b/apps/howto_deploy/tvm_runtime_pack.cc index e5c65b66b..9a090d863 100644 --- a/apps/howto_deploy/tvm_runtime_pack.cc +++ b/apps/howto_deploy/tvm_runtime_pack.cc @@ -44,10 +44,19 @@ // #include "../../src/runtime/rpc/rpc_event_impl.cc" // #include "../../src/runtime/rpc/rpc_server_env.cc" +// These macros enables the device API when uncommented. +#define TVM_CUDA_RUNTIME 1 +#define TVM_METAL_RUNTIME 1 +#define TVM_OPENCL_RUNTIME 1 + // Uncomment the following lines to enable Metal // #include "../../src/runtime/metal/metal_device_api.mm" // #include "../../src/runtime/metal/metal_module.mm" +// Uncomment the following lines to enable CUDA +// #include "../../src/runtime/cuda/cuda_device_api.cc" +// #include "../../src/runtime/cuda/cuda_runtime.cc" + // Uncomment the following lines to enable OpenCL // #include "../../src/runtime/opencl/opencl_device_api.cc" // #include "../../src/runtime/opencl/opencl_module.cc" -- GitLab