feat: Add aten.unbind decomposition for VIT (#2430)

pytorch · Nov 2, 2023 · 7c462ae · 7c462ae
1 parent 37b5cde
commit 7c462ae
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -153,6 +153,7 @@
     aten.transpose.int,
     aten.tril.default,
     aten.triu.default,
+    aten.unbind,
     aten.unfold,
     aten.unfold_backward,
     aten.unfold_copy,