From 762a15b304bf8a432a8b3c303a6ae3271cb5f522 Mon Sep 17 00:00:00 2001 From: Almeet Bhullar Date: Fri, 29 Dec 2023 03:30:19 +0000 Subject: [PATCH] #4171: Update move sweep test to create a dummy tensor so input and output to move are not in the same address --- tests/tt_eager/python_api_testing/sweep_tests/tt_lib_ops.py | 3 +++ 1 file changed, 3 insertions(+) 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 index ce413acf63b..521f444e1ef 100644 --- 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)