From a77925fac808c94556478cecee2925d0d988aaa5 Mon Sep 17 00:00:00 2001 From: Scott Roy Date: Sat, 8 Feb 2025 16:05:13 -0800 Subject: [PATCH] Fix broken tests from elide view copy Summary: Fixes test from T214758651 Differential Revision: D69344570 --- exir/tests/test_memory_planning.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/exir/tests/test_memory_planning.py b/exir/tests/test_memory_planning.py index 1e9a7e29f30..d5e5627dfa6 100644 --- a/exir/tests/test_memory_planning.py +++ b/exir/tests/test_memory_planning.py @@ -756,18 +756,18 @@ def forward(self, input, label): ep.dump_executorch_program(True) - # 155 just so happens to be the index of the user_grad output arg of + # 147 just so happens to be the index of the user_grad output arg of # convolution_backward.out. This is fairly fragile. # Check that the None output is not memory planned. self.assertEqual( ep.executorch_program.execution_plan[0] - .values[155] + .values[147] .val.data_buffer_idx, # pyright: ignore 0, ) self.assertEqual( ep.executorch_program.execution_plan[0] - .values[155] + .values[147] .val.allocation_info, # pyright: ignore None, )