Skip to content

Commit 299c56d

Browse files
masnesralfacebook-github-bot
authored andcommitted
Fix internal tests affected by D77980055
Summary: Looks like the change to how full_like is decomposed means these tests need some tweaks Differential Revision: D77981005
1 parent a4ac131 commit 299c56d

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

backends/arm/test/passes/test_decompose_cosine_similarity_pass.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,6 @@ def test_decompose_cosine_similarity_tosa_BI(module):
3434
"executorch_exir_dialects_edge__ops_aten_mul_Tensor": 5,
3535
"executorch_exir_dialects_edge__ops_aten_sum_dim_IntList": 3,
3636
"executorch_exir_dialects_edge__ops_aten_pow_Tensor_Scalar": 2,
37-
"executorch_exir_dialects_edge__ops_aten_full_like_default": 1,
3837
"executorch_exir_dialects_edge__ops_aten_maximum_default": 2,
3938
"executorch_exir_dialects_edge__ops_aten_reciprocal_default": 1,
4039
}

exir/tests/test_memory_planning.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -868,18 +868,19 @@ def forward(self, input, label):
868868

869869
ep.dump_executorch_program(True)
870870

871-
# 147 just so happens to be the index of the user_grad output arg of
871+
# 149 just so happens to be the index of the user_grad output arg of
872872
# convolution_backward.out. This is fairly fragile.
873873
# Check that the None output is not memory planned.
874+
self.assertEqual(len(ep.executorch_program.execution_plan[0].values), 151)
874875
self.assertEqual(
875876
ep.executorch_program.execution_plan[0]
876-
.values[147]
877+
.values[149]
877878
.val.data_buffer_idx, # pyright: ignore
878879
0,
879880
)
880881
self.assertEqual(
881882
ep.executorch_program.execution_plan[0]
882-
.values[147]
883+
.values[149]
883884
.val.allocation_info, # pyright: ignore
884885
None,
885886
)

0 commit comments

Comments
 (0)