update makefile

asappresearch · May 13, 2021 · d6daad6 · d6daad6
1 parent ed64427
commit d6daad6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/sru/csrc/CMakeLists.txt b/sru/csrc/CMakeLists.txt
@@ -21,7 +21,9 @@ add_executable(example_app main_test_cpp.cpp)
 target_link_libraries(example_app "${TORCH_LIBRARIES}")
 if (UNIX AND NOT APPLE)
     target_link_libraries(example_app -Wl,--no-as-needed sru_cpu)
+    target_link_libraries(example_app -Wl,--no-as-needed sru_cuda)
 else()
     target_link_libraries(example_app -Wl,-all_load sru_cpu)
+    target_link_libraries(example_app -Wl,-all_load sru_cuda)
 endif()
 target_compile_features(example_app PRIVATE cxx_std_14)