Skip to content

Commit 4809dea

Browse files
masnesralfacebook-github-bot
authored andcommitted
Fix internal tests affected by D77980055 (#12554)
Summary: Pull Request resolved: #12554 Looks like the change to how full_like is decomposed means these tests need some tweaks Reviewed By: digantdesai Differential Revision: D77981005
1 parent 988435d commit 4809dea

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

backends/arm/test/passes/test_decompose_cosine_similarity_pass.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ 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,
37+
"executorch_exir_dialects_edge__ops_aten_full_default": 1,
3838
"executorch_exir_dialects_edge__ops_aten_maximum_default": 2,
3939
"executorch_exir_dialects_edge__ops_aten_reciprocal_default": 1,
4040
}

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)