chore: updates

peri044 · peri044 · commit 38642bb9ef8b · 2024-04-18T16:07:04.000-07:00
diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py
@@ -313,7 +313,7 @@ def run(
         )
         timing_cache = self._create_timing_cache(builder_config, existing_cache)
 
-        engine = self.builder.build_engine(self.ctx.net, builder_config)
+        engine = self.builder.build_serialized_network(self.ctx.net, builder_config)
         assert engine
 
         serialized_cache = (

Original file line number	Diff line number	Diff line change
`@@ -313,7 +313,7 @@ def run(`
`313`	`313`	`)`
`314`	`314`	`timing_cache = self._create_timing_cache(builder_config, existing_cache)`
`315`	`315`
`316`		`- engine = self.builder.build_engine(self.ctx.net, builder_config)`
	`316`	`+ engine = self.builder.build_serialized_network(self.ctx.net, builder_config)`
`317`	`317`	`assert engine`
`318`	`318`
`319`	`319`	`serialized_cache = (`