chore: Improve error propagation for torch compile (#2106)

gs-olive · web-flow · commit 230cf6a927b9 · 2023-07-17T12:49:12.000-07:00
diff --git a/py/torch_tensorrt/dynamo/backend/backends.py b/py/torch_tensorrt/dynamo/backend/backends.py
@@ -88,13 +88,14 @@ def _pretraced_backend(
             )
             return gm.forward
         else:
-            raise AssertionError(
+            logger.critical(
                 "Halting compilation on build failure since "
                 + "pass_through_build_failures was specified as True. "
                 + "To return the default Torch implementation and avoid "
                 + "halting compilation on engine build failures, "
                 + "specify pass_through_build_failures=False."
             )
+            raise
 
 
 def _compile_module(

Original file line number	Diff line number	Diff line change
`@@ -88,13 +88,14 @@ def _pretraced_backend(`
`88`	`88`	`)`
`89`	`89`	`return gm.forward`
`90`	`90`	`else:`
`91`		`- raise AssertionError(`
	`91`	`+ logger.critical(`
`92`	`92`	`"Halting compilation on build failure since "`
`93`	`93`	`+ "pass_through_build_failures was specified as True. "`
`94`	`94`	`+ "To return the default Torch implementation and avoid "`
`95`	`95`	`+ "halting compilation on engine build failures, "`
`96`	`96`	`+ "specify pass_through_build_failures=False."`
`97`	`97`	`)`
	`98`	`+ raise`
`98`	`99`
`99`	`100`
`100`	`101`	`def _compile_module(`