Update

triton-inference-server · Jul 2, 2024 · 5458094 · 5458094
1 parent d53a49b
commit 5458094
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -142,8 +142,6 @@ if("$ENV{TRT_VERSION}" VERSION_LESS 10)
     ${SOURCE_DIR}/instance_state.cc
     ${SOURCE_DIR}/tensorrt_model_instance.cc
     ${SOURCE_DIR}/tensorrt_model_instance.h
-    ${SOURCE_DIR}/shape_tensor.cc
-    ${SOURCE_DIR}/shape_tensor.h
     ${SOURCE_DIR}/tensorrt_utils.cc
     ${SOURCE_DIR}/tensorrt_utils.h
     ${SOURCE_DIR}/filesystem.h
@@ -170,6 +168,8 @@ else()
     ${SOURCE_DIR}/instance_state.cc
     ${SOURCE_DIR}/tensorrt_model_instance.cc
     ${SOURCE_DIR}/tensorrt_model_instance.h
+    ${SOURCE_DIR}/shape_tensor.cc
+    ${SOURCE_DIR}/shape_tensor.h
     ${SOURCE_DIR}/tensorrt_utils.cc
     ${SOURCE_DIR}/tensorrt_utils.h
     ${SOURCE_DIR}/filesystem.h