Skip to content

Commit b5b45a1

Browse files
committed
adding select_scatter decomp lowering ops in test
1 parent 3814267 commit b5b45a1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/py/dynamo/lowering/test_decompositions.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -434,6 +434,7 @@ def forward(self, x, src, dim, index):
434434
torch.ops.aten.slice.Tensor,
435435
torch.ops.aten.squeeze.dim,
436436
torch.ops.aten.cat.default,
437+
torch.ops.aten.reshape.default,
437438
}
438439
unexpected_ops = {torch.ops.aten.select_scatter.default}
439440

@@ -496,6 +497,7 @@ def forward(self, x, src, dim, index):
496497
expected_ops = {
497498
torch.ops.aten.slice.Tensor,
498499
torch.ops.aten.squeeze.dim,
500+
torch.ops.aten.unsqueeze.default,
499501
torch.ops.aten.cat.default,
500502
}
501503
unexpected_ops = {torch.ops.aten.select_scatter.default}

0 commit comments

Comments
 (0)