MAINT: fix CMake script.

oddkiva · Dec 11, 2023 · ccf8ab3 · ccf8ab3
1 parent 6ce3fe4
commit ccf8ab3
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/cpp/examples/Shakti/TensorRT/CMakeLists.txt b/cpp/examples/Shakti/TensorRT/CMakeLists.txt
@@ -1,3 +1,7 @@
+if(NOT CMAKE_CUDA_COMPILER OR NOT TensorRT_FOUND)
+  return()
+endif()
+
 file(GLOB TRT_SOURCE_FILES FILES *.cpp)
 
 foreach(file ${TRT_SOURCE_FILES})

diff --git a/cpp/test/Shakti/Cuda/TensorRT/CMakeLists.txt b/cpp/test/Shakti/Cuda/TensorRT/CMakeLists.txt
@@ -1,3 +1,7 @@
+if(NOT CMAKE_CUDA_COMPILER OR NOT TensorRT_FOUND)
+  return()
+endif()
+
 file(GLOB test_tensorrt_SOURCE_FILES FILES test_*tensorrt*.cpp)
 
 foreach(file ${test_tensorrt_SOURCE_FILES})