Commit 8084a4da by Kirill Mavreshko Committed by Tianqi Chen

Fix fatal error <CL/opencl.h: No such file> when building with CMake (#1045)

parent cacca7b5
...@@ -154,7 +154,7 @@ endif(USE_CUDA) ...@@ -154,7 +154,7 @@ endif(USE_CUDA)
if(USE_OPENCL) if(USE_OPENCL)
find_package(OpenCL QUIET REQUIRED) find_package(OpenCL QUIET REQUIRED)
message(STATUS "Build with OpenCL support") message(STATUS "Build with OpenCL support")
include_directories(${OPENCL_INCLUDE_DIRS}) include_directories(${OpenCL_INCLUDE_DIRS})
list(APPEND TVM_RUNTIME_LINKER_LIBS ${OpenCL_LIBRARIES}) list(APPEND TVM_RUNTIME_LINKER_LIBS ${OpenCL_LIBRARIES})
list(APPEND RUNTIME_SRCS ${RUNTIME_OPENCL_SRCS}) list(APPEND RUNTIME_SRCS ${RUNTIME_OPENCL_SRCS})
add_definitions(-DTVM_OPENCL_RUNTIME=1) add_definitions(-DTVM_OPENCL_RUNTIME=1)
......
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