fix: trt context memory sharing (#327)

pykeio · Nov 28, 2024 · 5d41c96 · 5d41c96
1 parent 7fa9734
commit 5d41c96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/execution_providers/tensorrt.rs b/src/execution_providers/tensorrt.rs
@@ -135,7 +135,7 @@ impl TensorRTExecutionProvider {
 
 	#[must_use]
 	pub fn with_context_memory_sharing(mut self, enable: bool) -> Self {
-		self.options.set("trt_enable_context_memory_sharing", if enable { "1" } else { "0" });
+		self.options.set("trt_context_memory_sharing_enable", if enable { "1" } else { "0" });
 		self
 	}