We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 82298c5 commit 66861f4Copy full SHA for 66861f4
py/torch_tensorrt/dynamo/backend/backends.py
@@ -69,12 +69,13 @@ def aot_torch_tensorrt_aten_backend(
69
to_delete = {
70
key
71
for key in settings_aot_autograd["decompositions"]
72
- if "detach" in key._name or "transpose" in key._name
+ if "transpose" in key._name
73
}
74
75
for key in to_delete:
76
del settings_aot_autograd["decompositions"][key]
77
78
+ remove_detach(gm, settings)
79
return aot_autograd(
80
fw_compiler=_pretraced_backend_autograd,
81
decompositions=settings_aot_autograd["decompositions"],
0 commit comments