#4171: Update move sweep test to create a dummy tensor so input and o…

…utput to move are not in the same address
tenstorrent · Jan 3, 2024 · 762a15b · 762a15b
1 parent b039fa3
commit 762a15b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/tt_eager/python_api_testing/sweep_tests/tt_lib_ops.py b/tests/tt_eager/python_api_testing/sweep_tests/tt_lib_ops.py
@@ -109,7 +109,10 @@ def move(
     output_mem_config,
     **kwargs,
 ):
+    dummy_tensor = setup_tt_tensor(x, device, layout[0], input_mem_config[0], dtype[0])
     t0 = setup_tt_tensor(x, device, layout[0], input_mem_config[0], dtype[0])
+    # Free up dummy tensor from memory to make available to move
+    dummy_tensor.deallocate()
     t1 = ttl.tensor.move(t0, output_mem_config=output_mem_config)
 
     return tt2torch_tensor(t1)