Skip to content

Commit 6b96f09

Browse files
authored
chore: remove extra paths to find binaries (NVIDIA#4706)
Signed-off-by: Aurelien Chartier <[email protected]>
1 parent 5506f60 commit 6b96f09

File tree

1 file changed

+5
-11
lines changed

1 file changed

+5
-11
lines changed

triton_backend/inflight_batcher_llm/CMakeLists.txt

Lines changed: 5 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -151,23 +151,17 @@ find_package(Python3 COMPONENTS Interpreter Development)
151151
find_library(
152152
tensorrt_llm libtensorrt_llm.so REQUIRED
153153
PATHS ${Python3_SITEARCH}/tensorrt_llm/libs
154-
${TRTLLM_DIR}/cpp/build/tensorrt_llm
155-
${CMAKE_CURRENT_SOURCE_DIR}/../tensorrt_llm/cpp/build/tensorrt_llm)
154+
${TRTLLM_DIR}/cpp/build/tensorrt_llm)
156155

157156
find_library(
158157
nvinfer_plugin_tensorrt_llm libnvinfer_plugin_tensorrt_llm.so REQUIRED
159-
PATHS
160-
${Python3_SITEARCH}/tensorrt_llm/libs
161-
${TRTLLM_DIR}/cpp/build/tensorrt_llm/plugins
162-
${CMAKE_CURRENT_SOURCE_DIR}/../tensorrt_llm/cpp/build/tensorrt_llm/plugins)
158+
PATHS ${Python3_SITEARCH}/tensorrt_llm/libs
159+
${TRTLLM_DIR}/cpp/build/tensorrt_llm/plugins)
163160

164161
find_program(
165162
TRTLLM_EXECUTOR_WORKER executorWorker REQUIRED
166-
PATHS
167-
${Python3_SITEARCH}/tensorrt_llm/bin
168-
${TRTLLM_DIR}/cpp/build/tensorrt_llm/executor_worker
169-
${CMAKE_CURRENT_SOURCE_DIR}/../tensorrt_llm/cpp/build/tensorrt_llm/executor_worker
170-
)
163+
PATHS ${Python3_SITEARCH}/tensorrt_llm/bin
164+
${TRTLLM_DIR}/cpp/build/tensorrt_llm/executor_worker)
171165
install(
172166
PROGRAMS ${TRTLLM_EXECUTOR_WORKER}
173167
DESTINATION ${CMAKE_BINARY_DIR}

0 commit comments

Comments
 (0)