Commit b18143e5 by Tianqi Chen Committed by GitHub

[DOC][BUILD] Fix cmake and docs (#485)

parent 145b3d0f
......@@ -87,6 +87,7 @@ file(GLOB RUNTIME_CUDA_SRCS src/runtime/cuda/*.cc)
file(GLOB RUNTIME_OPENCL_SRCS src/runtime/opencl/*.cc)
file(GLOB RUNTIME_METAL_SRCS src/runtime/metal/*.mm)
file(GLOB RUNTIME_RPC_SRCS src/runtime/rpc/*.cc)
file(GLOB RUNTIME_GRAPH_SRCS src/runtime/graph/*.cc)
if(USE_CUDA)
find_package(CUDA)
......@@ -149,6 +150,11 @@ if(USE_RPC)
list(APPEND RUNTIME_SRCS ${RUNTIME_RPC_SRCS})
endif(USE_RPC)
if(USE_GRAPH_RUNTIME)
message(STATUS "Build with Graph runtime support...")
list(APPEND RUNTIME_SRCS ${RUNTIME_GRAPH_SRCS})
endif(USE_GRAPH_RUNTIME)
if(USE_LLVM)
find_package(LLVM CONFIG REQUIRED)
include_directories(${LLVM_INCLUDE_DIRS})
......
......@@ -21,6 +21,9 @@
#include "../src/runtime/rpc/rpc_module.cc"
#include "../src/runtime/rpc/rpc_socket_impl.cc"
#include "../src/runtime/thread_pool.cc"
#include "../src/runtime/graph/graph_runtime.cc"
#ifdef TVM_OPENCL_RUNTIME
#include "../src/runtime/opencl/opencl_device_api.cc"
#include "../src/runtime/opencl/opencl_module.cc"
......
......@@ -19,6 +19,8 @@
#include "../../src/runtime/rpc/rpc_server_env.cc"
#include "../../src/runtime/rpc/rpc_socket_impl.cc"
#include "../../src/runtime/rpc/rpc_module.cc"
// Graph runtime
#include "../../src/runtime/graph/graph_runtime.cc"
// Metal
#include "../../src/runtime/metal/metal_module.mm"
#include "../../src/runtime/metal/metal_device_api.mm"
......
......@@ -22,9 +22,9 @@ tvm.contrib.rpc
.. automodule:: tvm.contrib.rpc
:members:
tvm.contrib.graph
~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.graph
tvm.contrib.graph_runtime
~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.graph_runtime
:members:
tvm.contrib.util
......
......@@ -17,6 +17,7 @@
#include "../src/runtime/rpc/rpc_session.cc"
#include "../src/runtime/rpc/rpc_event_impl.cc"
#include "../src/runtime/rpc/rpc_server_env.cc"
#include "../src/runtime/graph/graph_runtime.cc"
namespace tvm {
namespace contrib {
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment