adding select_scatter decomp lowering ops in test

pytorch · Feb 26, 2024 · b5b45a1 · b5b45a1
1 parent 3814267
commit b5b45a1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/py/dynamo/lowering/test_decompositions.py b/tests/py/dynamo/lowering/test_decompositions.py
@@ -434,6 +434,7 @@ def forward(self, x, src, dim, index):
             torch.ops.aten.slice.Tensor,
             torch.ops.aten.squeeze.dim,
             torch.ops.aten.cat.default,
+            torch.ops.aten.reshape.default,
         }
         unexpected_ops = {torch.ops.aten.select_scatter.default}
 
@@ -496,6 +497,7 @@ def forward(self, x, src, dim, index):
         expected_ops = {
             torch.ops.aten.slice.Tensor,
             torch.ops.aten.squeeze.dim,
+            torch.ops.aten.unsqueeze.default,
             torch.ops.aten.cat.default,
         }
         unexpected_ops = {torch.ops.aten.select_scatter.default}