#13745:modify failed test1

tenstorrent · Nov 29, 2024 · c01e97d · c01e97d
1 parent 610a36c
commit c01e97d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/ttnn/unit_tests/test_reshape.py b/tests/ttnn/unit_tests/test_reshape.py
@@ -35,7 +35,7 @@ def test_reshape_sharded_rm(device, n, c, h, w):
         torch_input_tensor, layout=ttnn.ROW_MAJOR_LAYOUT, device=device, memory_config=sharded_mem_config
     )
 
-    tt_output_tensor = tt_input_tensor.reshape(n, c, h * 2, w // 2)
+    tt_output_tensor = tt_input_tensor.reshape_unsafe(n, c, h * 2, w // 2)
 
     sharded_mem_config = ttnn.create_sharded_memory_config(
         tt_output_tensor.shape,