From 19c30545022283aa06e99c7898f2588e72f75389 Mon Sep 17 00:00:00 2001
From: Sai Kiran Polisetty <spolisetty@nvidia.com>
Date: Tue, 2 Jul 2024 22:51:47 +0530
Subject: [PATCH] Update

---
 CMakeLists.txt | 17 ++++++-----------
 1 file changed, 6 insertions(+), 11 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index c729fcc..8314a3b 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -122,17 +122,6 @@ message(STATUS "TRT_VERSION envvar is $ENV{TRT_VERSION}")
 #
 if("$ENV{TRT_VERSION}" VERSION_LESS 10)
   set(SOURCE_DIR "src_trt8")
-else()
-  set(SOURCE_DIR "src")
-endif()
-
-
-#
-# Shared library implementing the Triton Backend API
-#
-configure_file(${SOURCE_DIR}/libtriton_tensorrt.ldscript libtriton_tensorrt.ldscript COPYONLY)
-
-if("$ENV{TRT_VERSION}" VERSION_LESS 10)
   add_library(
     triton-tensorrt-backend SHARED
     ${SOURCE_DIR}/tensorrt.cc
@@ -159,6 +148,7 @@ if("$ENV{TRT_VERSION}" VERSION_LESS 10)
     ${SOURCE_DIR}/io_binding_info.h
   )
 else()
+  set(SOURCE_DIR "src")
   add_library(
     triton-tensorrt-backend SHARED
     ${SOURCE_DIR}/tensorrt.cc
@@ -188,6 +178,11 @@ else()
   )
 endif()
 
+#
+# Shared library implementing the Triton Backend API
+#
+configure_file(${SOURCE_DIR}/libtriton_tensorrt.ldscript libtriton_tensorrt.ldscript COPYONLY)
+
 add_library(
   TritonTensorRTBackend::triton-tensorrt-backend ALIAS triton-tensorrt-backend
 )