Skip to content

Commit 414ba84

Browse files
committed
require both tensorrt and cuda
1 parent 83dc33f commit 414ba84

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

semantic_inference/CMakeLists.txt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,7 @@ find_package(CUDAToolkit)
2121
find_package(TensorRT COMPONENTS onnxparser infer_plugin)
2222

2323
set(ENABLE_TENSORRT_CXX_VALUE 1)
24-
if(SEMANTIC_INFERENCE_USE_TRT
25-
AND NOT CUDAToolkit_FOUND
26-
AND NOT TensorRT_FOUND
27-
)
24+
if(SEMANTIC_INFERENCE_USE_TRT AND (NOT CUDAToolkit_FOUND OR NOT TensorRT_FOUND))
2825
message(WARNING "TensorRT is required for inference to run!"
2926
" Follow the install instructions (if possible) to enable inference"
3027
)

0 commit comments

Comments
 (0)