zhuochenKIDD commented on a change in pull request #7616:
URL: https://github.com/apache/tvm/pull/7616#discussion_r595784664
##########
File path: cmake/modules/CUDA.cmake
##########
@@ -65,6 +65,17 @@ if(USE_CUDA)
list(APPEND RUNTIME_SRCS ${CONTRIB_THRUST_SRC})
endif(USE_THRUST)
+ if(USE_GRAPH_RUNTIME_CUGRAPH)
+ if(NOT USE_GRAPH_RUNTIME)
+ message(FATAL_ERROR "CUDA Graph is only supported by graph runtime,
should set USE_GRAPH_RUNTIME=ON")
+ endif()
+ if(CUDAToolkit_VERSION_MAJOR LESS "10")
+ message(FATAL_ERROR "CUDA Graph requires at least CUDA 10, got="
${CUDAToolkit_VERSION})
+ endif()
+ message(STATUS "Build with Graph runtime cuGraph support...")
Review comment:
Yes I removed all cuGraph or cu_graph name, use CUDA Graph instread
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
[email protected]
