File tree 2 files changed +7
-15
lines changed
py/torch_tensorrt/dynamo/backend
2 files changed +7
-15
lines changed Original file line number Diff line number Diff line change @@ -123,19 +123,12 @@ def create_backend(
123
123
Returns:
124
124
Backend for torch.compile
125
125
"""
126
- if debug :
127
- logger .setLevel (logging .DEBUG )
128
-
129
- settings = CompilationSettings (
126
+ return partial (
127
+ torch_tensorrt_backend ,
130
128
debug = debug ,
131
129
precision = precision ,
132
130
workspace_size = workspace_size ,
133
131
min_block_size = min_block_size ,
134
132
torch_executed_ops = torch_executed_ops ,
135
133
pass_through_build_failures = pass_through_build_failures ,
136
134
)
137
-
138
- return partial (
139
- torch_tensorrt_backend ,
140
- settings = settings ,
141
- )
Original file line number Diff line number Diff line change @@ -77,13 +77,12 @@ def _pretraced_backend(
77
77
)
78
78
return trt_compiled
79
79
except :
80
- logger .error (
81
- "FX2TRT conversion failed on the subgraph. See trace above. "
82
- + "Returning GraphModule forward instead." ,
83
- exc_info = True ,
84
- )
85
-
86
80
if not settings .pass_through_build_failures :
81
+ logger .error (
82
+ "FX2TRT conversion failed on the subgraph. See trace above. "
83
+ + "Returning GraphModule forward instead." ,
84
+ exc_info = True ,
85
+ )
87
86
return gm .forward
88
87
else :
89
88
raise AssertionError (
You can’t perform that action at this time.
0 commit comments