Skip to content

Commit 905454d

Browse files
authored
Merge pull request #11794 from luotao1/fix_tensorrt_compiler_bug
fix tensorrt compiler bug
2 parents 2ecc562 + 9211824 commit 905454d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

paddle/contrib/inference/CMakeLists.txt

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,10 @@ if(APPLE)
1818
endif(APPLE)
1919

2020

21-
set(inference_deps paddle_inference_api paddle_fluid_api paddle_inference_tensorrt_subgraph_engine)
21+
set(inference_deps paddle_inference_api paddle_fluid_api)
22+
if(WITH_GPU AND TENSORRT_FOUND)
23+
set(inference_deps ${inference_deps} paddle_inference_tensorrt_subgraph_engine)
24+
endif()
2225

2326
function(inference_api_test TARGET_NAME)
2427
if (WITH_TESTING)

0 commit comments

Comments
 (0)