Change path

Author: fpetrini15

CMakeLists.txt

@@ -334,8 +334,13 @@ endif() # TRITON_ENABLE_ONNXRUNTIME_OPENVINO
 if(TRITON_ONNXRUNTIME_DOCKER_BUILD)
   set(_GEN_FLAGS "")
   if(${RHEL_BUILD} AND "${TRITON_BUILD_TENSORRT_HOME}" STREQUAL "")
-    set(TRITON_BUILD_TENSORRT_HOME "/usr/local/cuda/targets/${CMAKE_SYSTEM_PROCESSOR}-linux/")
+    if(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "aarch64")
+      set(TRITON_BUILD_TENSORRT_HOME "/usr/local/cuda/targets/sbsa-linux/")
+    else()
+      set(TRITON_BUILD_TENSORRT_HOME "/usr/local/cuda/targets/x86_64-linux/")
+    endif(${CMAKE_SYSTEM_PROCESSOR} STREQUAL "aarch64")
   endif(${RHEL_BUILD} AND "${TRITON_BUILD_TENSORRT_HOME}" STREQUAL "")
+  message(${TRITON_BUILD_TENSORRT_HOME})
   if(NOT ${TRITON_BUILD_TARGET_PLATFORM} STREQUAL "")
     set(_GEN_FLAGS ${_GEN_FLAGS} "--target-platform=${TRITON_BUILD_TARGET_PLATFORM}")
   endif() # TRITON_BUILD_TARGET_PLATFORM