Skip to content
Snippets Groups Projects
Commit f8b40dc8 authored by hashimsharif's avatar hashimsharif
Browse files

Adding build support of OpenMP for CPU tensor runtime

parent 86d33f5a
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
llvm/projects/hpvm-tensor-rt/CMakeLists.txt
+ 3
− 1
View file @ f8b40dc8
...@@ -108,8 +108,10 @@ cuda_add_cublas_to_target(tensor_runtime_online) ...@@ -108,8 +108,10 @@ cuda_add_cublas_to_target(tensor_runtime_online)
target_link_libraries(tensor_runtime_online ${LINK_LIBS}) target_link_libraries(tensor_runtime_online ${LINK_LIBS})
# Adding new rule for building a cuDNN runtime library # Adding new rule for building a cuDNN runtime library
find_package(OpenMP REQUIRED)
cuda_add_library(tensor_cpu_runtime tensor_runtime/src/tensor_cpu_runtime.cc) cuda_add_library(tensor_cpu_runtime tensor_runtime/src/tensor_cpu_runtime.cc)
target_link_libraries(tensor_cpu_runtime) target_compile_options(tensor_cpu_runtime PRIVATE ${OpenMP_CXX_FLAGS})
target_link_libraries(tensor_cpu_runtime PRIVATE ${OpenMP_CXX_FLAGS})
### TODO: Remove unsued CMake rules after careful consideration ### TODO: Remove unsued CMake rules after careful consideration
......
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