Skip to content

Commit 66861f4

Browse files
committed
using detach lowering pass
1 parent 82298c5 commit 66861f4

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: py/torch_tensorrt/dynamo/backend/backends.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -69,12 +69,13 @@ def aot_torch_tensorrt_aten_backend(
6969
to_delete = {
7070
key
7171
for key in settings_aot_autograd["decompositions"]
72-
if "detach" in key._name or "transpose" in key._name
72+
if "transpose" in key._name
7373
}
7474

7575
for key in to_delete:
7676
del settings_aot_autograd["decompositions"][key]
7777

78+
remove_detach(gm, settings)
7879
return aot_autograd(
7980
fw_compiler=_pretraced_backend_autograd,
8081
decompositions=settings_aot_autograd["decompositions"],

0 commit comments

Comments
 (0)