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

Author: tqchen

CMakeLists.txt

@@ -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})

apps/android_rpc/app/src/main/jni/tvm_runtime.h

@@ -17,10 +17,13 @@
 #include "../src/runtime/dso_module.cc"
 #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/rpc/rpc_server_env.cc"
 #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"

apps/ios_rpc/tvmrpc/TVMRuntime.mm

@@ -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"

docs/api/python/contrib.rst

@@ -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

web/web_runtime.cc

@@ -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 {