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,