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) ...@@ -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_OPENCL_SRCS src/runtime/opencl/*.cc)
file(GLOB RUNTIME_METAL_SRCS src/runtime/metal/*.mm) file(GLOB RUNTIME_METAL_SRCS src/runtime/metal/*.mm)
file(GLOB RUNTIME_RPC_SRCS src/runtime/rpc/*.cc) file(GLOB RUNTIME_RPC_SRCS src/runtime/rpc/*.cc)
file(GLOB RUNTIME_GRAPH_SRCS src/runtime/graph/*.cc)
if(USE_CUDA) if(USE_CUDA)
find_package(CUDA) find_package(CUDA)
...@@ -149,6 +150,11 @@ if(USE_RPC) ...@@ -149,6 +150,11 @@ if(USE_RPC)
list(APPEND RUNTIME_SRCS ${RUNTIME_RPC_SRCS}) list(APPEND RUNTIME_SRCS ${RUNTIME_RPC_SRCS})
endif(USE_RPC) 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) if(USE_LLVM)
find_package(LLVM CONFIG REQUIRED) find_package(LLVM CONFIG REQUIRED)
include_directories(${LLVM_INCLUDE_DIRS}) include_directories(${LLVM_INCLUDE_DIRS})
......
...@@ -17,10 +17,13 @@ ...@@ -17,10 +17,13 @@
#include "../src/runtime/dso_module.cc" #include "../src/runtime/dso_module.cc"
#include "../src/runtime/rpc/rpc_session.cc" #include "../src/runtime/rpc/rpc_session.cc"
#include "../src/runtime/rpc/rpc_event_impl.cc" #include "../src/runtime/rpc/rpc_event_impl.cc"
#include "../src/runtime/rpc/rpc_server_env.cc" #include "../src/runtime/rpc/rpc_server_env.cc"
#include "../src/runtime/rpc/rpc_module.cc" #include "../src/runtime/rpc/rpc_module.cc"
#include "../src/runtime/rpc/rpc_socket_impl.cc" #include "../src/runtime/rpc/rpc_socket_impl.cc"
#include "../src/runtime/thread_pool.cc" #include "../src/runtime/thread_pool.cc"
#include "../src/runtime/graph/graph_runtime.cc"
#ifdef TVM_OPENCL_RUNTIME #ifdef TVM_OPENCL_RUNTIME
#include "../src/runtime/opencl/opencl_device_api.cc" #include "../src/runtime/opencl/opencl_device_api.cc"
#include "../src/runtime/opencl/opencl_module.cc" #include "../src/runtime/opencl/opencl_module.cc"
......
...@@ -19,6 +19,8 @@ ...@@ -19,6 +19,8 @@
#include "../../src/runtime/rpc/rpc_server_env.cc" #include "../../src/runtime/rpc/rpc_server_env.cc"
#include "../../src/runtime/rpc/rpc_socket_impl.cc" #include "../../src/runtime/rpc/rpc_socket_impl.cc"
#include "../../src/runtime/rpc/rpc_module.cc" #include "../../src/runtime/rpc/rpc_module.cc"
// Graph runtime
#include "../../src/runtime/graph/graph_runtime.cc"
// Metal // Metal
#include "../../src/runtime/metal/metal_module.mm" #include "../../src/runtime/metal/metal_module.mm"
#include "../../src/runtime/metal/metal_device_api.mm" #include "../../src/runtime/metal/metal_device_api.mm"
......
...@@ -22,9 +22,9 @@ tvm.contrib.rpc ...@@ -22,9 +22,9 @@ tvm.contrib.rpc
.. automodule:: tvm.contrib.rpc .. automodule:: tvm.contrib.rpc
:members: :members:
tvm.contrib.graph tvm.contrib.graph_runtime
~~~~~~~~~~~~~~~~~ ~~~~~~~~~~~~~~~~~~~~~~~~~
.. automodule:: tvm.contrib.graph .. automodule:: tvm.contrib.graph_runtime
:members: :members:
tvm.contrib.util tvm.contrib.util
......
...@@ -17,6 +17,7 @@ ...@@ -17,6 +17,7 @@
#include "../src/runtime/rpc/rpc_session.cc" #include "../src/runtime/rpc/rpc_session.cc"
#include "../src/runtime/rpc/rpc_event_impl.cc" #include "../src/runtime/rpc/rpc_event_impl.cc"
#include "../src/runtime/rpc/rpc_server_env.cc" #include "../src/runtime/rpc/rpc_server_env.cc"
#include "../src/runtime/graph/graph_runtime.cc"
namespace tvm { namespace tvm {
namespace contrib { 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