Fix backend and devtools debug handle based tests (#6143)

Summary: Pull Request resolved: #6143 Fix the tests that started failing after D63913905 Reviewed By: cccclai Differential Revision: D64210644 fbshipit-source-id: d43165cd036d0552114ae8c79fed346c973ffcdb
pytorch · Oct 11, 2024 · f52f4d4 · f52f4d4
1 parent 4a4a90f
commit f52f4d4
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/exir/backend/test/test_backends.py b/exir/backend/test/test_backends.py
@@ -190,7 +190,7 @@ def forward(self, x):
             program=program,
             delegate=program.execution_plan[0].delegates[0],
             expected_id=BackendWithCompilerDemo.__name__,
-            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>1#",
+            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>2#",
         )
 
         # Check the delegate instruction
@@ -410,7 +410,7 @@ def forward(self, x):
             program=program,
             delegate=program.execution_plan[0].delegates[0],
             expected_id=BackendWithCompilerDemo.__name__,
-            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>1#",
+            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>2#",
         )
 
         # Check the delegate instruction

diff --git a/exir/backend/test/test_backends_lifted.py b/exir/backend/test/test_backends_lifted.py
@@ -210,7 +210,7 @@ def forward(self, x):
             program=program,
             delegate=program.execution_plan[0].delegates[0],
             expected_id=BackendWithCompilerDemo.__name__,
-            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>1#",
+            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>2#",
         )
 
         # Check the delegate instruction
@@ -414,7 +414,7 @@ def forward(self, x):
             program=program,
             delegate=program.execution_plan[0].delegates[0],
             expected_id=BackendWithCompilerDemo.__name__,
-            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>1#",
+            expected_processed=b"1version:0#op:demo::aten.sin.default, numel:1, dtype:torch.float32<debug_handle>2#",
         )
 
         # Check the delegate instruction