Skip to content

Commit 7fa036f

Browse files
committed
fix: Add additional decompositions for SD
1 parent b1665ef commit 7fa036f

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

py/torch_tensorrt/dynamo/lowering/_decompositions.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,12 @@
171171
aten.zero_,
172172
aten.zeros,
173173
aten.zeros_like,
174+
# Non-default convenience decompositions
175+
aten.clamp_min,
176+
aten.clamp_max,
177+
aten.linalg_vector_norm,
178+
aten.full,
179+
aten.repeat,
174180
}
175181
torch_disabled_decompositions: Set[torch._ops.OpOverload] = {}
176182

0 commit comments

Comments
 (0)