From d5a92567e8b08510709c4fa6bddb98717e536617 Mon Sep 17 00:00:00 2001
From: gs-olive <113141689+gs-olive@users.noreply.github.com>
Date: Wed, 1 Nov 2023 17:00:34 -0700
Subject: [PATCH] feat: Add `aten.unbind` decomposition for VIT

---
 py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
index 3d69b407f1..b95715b5ae 100644
--- 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,