feat: Decomposition for _unsafe_index (#2386)

gs-olive · web-flow · commit 7a1cdb3d0941 · 2023-10-10T15:20:51.000-07:00
diff --git a/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py b/py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py
@@ -158,6 +158,7 @@
     aten.unfold,
     aten.unfold_backward,
     aten.unfold_copy,
+    aten._unsafe_index,
     aten.upsample_bilinear2d,
     aten.upsample_bilinear2d.vec,
     aten.upsample_nearest2d_backward,
