From 6ffe9284da0d812413a1cc800e7c77bc7988a9da Mon Sep 17 00:00:00 2001
From: Ryan McCormick <rmccormick@nvidia.com>
Date: Thu, 13 Jun 2024 19:51:35 -0700
Subject: [PATCH] Add _10 suffix to NVINFER_LIBRARY and NVINFER_PLUGIN_LIBRARY
 for TRT 10

---
 CMakeLists.txt | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/CMakeLists.txt b/CMakeLists.txt
index 799e5ce..f7925c6 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -216,8 +216,9 @@ FOREACH(p ${TRITON_TENSORRT_LIB_PATHS})
     set(TRITON_TENSORRT_LDFLAGS ${TRITON_TENSORRT_LDFLAGS} "-L${p}")
 ENDFOREACH(p)
 
-find_library(NVINFER_LIBRARY NAMES nvinfer)
-find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin)
+# FIXME: TRT 10 appears to have differently named libraries with "_10" suffix
+find_library(NVINFER_LIBRARY NAMES nvinfer nvinfer_10)
+find_library(NVINFER_PLUGIN_LIBRARY NAMES nvinfer_plugin nvinfer_plugin_10)
 target_link_libraries(
   triton-tensorrt-backend
   PRIVATE