resolve comments

pytorch · Oct 15, 2024 · 5ae553b · 5ae553b
1 parent 0dde8d5
commit 5ae553b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py
@@ -633,7 +633,8 @@ def run(
 
         self._construct_trt_network_def()
 
-        self._save_weight_mapping()
+        if not self.compilation_settings.immutable_weights:
+            self._save_weight_mapping()
 
         build_engine_start_time = datetime.now()
         _LOGGER.info("Not found cached TRT engines. Start building engine.")
@@ -652,7 +653,7 @@ def run(
         assert serialized_engine
 
         _LOGGER.info(
-            f"Build weight-stripped TRT engine elapsed time: {datetime.now() - build_engine_start_time}"
+            f"Build TRT engine elapsed time: {datetime.now() - build_engine_start_time}"
         )
         _LOGGER.info(f"TRT Engine uses: {serialized_engine.nbytes} bytes of Memory")