vstrt/vs_tensorrt.cpp: share optimization profile when using dynamic …

…shapes
AmusementClub · May 18, 2024 · ba84703 · ba84703
1 parent de31928
commit ba84703
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/vstrt/vs_tensorrt.cpp b/vstrt/vs_tensorrt.cpp
@@ -445,6 +445,8 @@ static void VS_CC vsTrtCreate(
             is_dynamic
         );
 
+        // https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-1000-ea/developer-guide/index.html#perform-inference
+#if NV_TENSORRT_MAJOR < 10
         // duplicates ICudaEngine instances
         //
         // According to
@@ -458,6 +460,7 @@ static void VS_CC vsTrtCreate(
                 return set_error(std::get<ErrorMessage>(maybe_engine));
             }
         }
+#endif // NV_TENSORRT_MAJOR < 10
 
         if (std::holds_alternative<InferenceInstance>(maybe_instance)) {
             auto instance = std::move(std::get<InferenceInstance>(maybe_instance));