chore: trt 10 fixes

peri044 · peri044 · commit 8faf7ecbd5fe · 2024-03-20T09:57:54.000-07:00
diff --git a/py/torch_tensorrt/dynamo/_compiler.py b/py/torch_tensorrt/dynamo/_compiler.py
@@ -634,7 +634,7 @@ def convert_module_to_trt_engine(
     import io
 
     with io.BytesIO() as engine_bytes:
-        engine_bytes.write(interpreter_result.engine.serialize())
+        engine_bytes.write(interpreter_result.engine)
         engine_bytearray = engine_bytes.getvalue()
 
     return engine_bytearray
diff --git a/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py b/py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py
@@ -172,7 +172,7 @@ def run(
 
         if version.parse(trt.__version__) >= version.parse("8.2"):
             builder_config.profiling_verbosity = (
-                trt.ProfilingVerbosity.VERBOSE
+                trt.ProfilingVerbosity.DETAILED
                 if self.compilation_settings.debug
                 else trt.ProfilingVerbosity.LAYER_NAMES_ONLY
             )
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/elementwise/base.py b/py/torch_tensorrt/dynamo/conversion/impl/elementwise/base.py
@@ -147,17 +147,6 @@ def convert_binary_elementwise(
             ctx, rhs_val, trt_promoted_type, name, target, source_ir
         )
 
-    # Check the limitation in the doc string.
-    if ctx.net.has_implicit_batch_dimension:
-        if is_lhs_trt_tensor and not is_rhs_trt_tensor:
-            assert len(lhs_val.shape) >= len(
-                rhs_val.shape
-            ), f"{lhs_val.shape} >= {rhs_val.shape}"
-        elif not is_lhs_trt_tensor and is_rhs_trt_tensor:
-            assert len(rhs_val.shape) >= len(
-                lhs_val.shape
-            ), f"{rhs_val.shape} >= {lhs_val.shape}"
-
     lhs_val, rhs_val = broadcast(
         ctx.net, lhs_val, rhs_val, f"{name}_lhs", f"{name}_rhs"
     )

Original file line number	Diff line number	Diff line change
`@@ -172,7 +172,7 @@ def run(`
`172`	`172`
`173`	`173`	`if version.parse(trt.__version__) >= version.parse("8.2"):`
`174`	`174`	`builder_config.profiling_verbosity = (`
`175`		`- trt.ProfilingVerbosity.VERBOSE`
	`175`	`+ trt.ProfilingVerbosity.DETAILED`
`176`	`176`	`if self.compilation_settings.debug`
`177`	`177`	`else trt.ProfilingVerbosity.LAYER_NAMES_ONLY`
`178`	`178`	`)`