#0: refactoring to use pytest device so it works in methodology

tenstorrent · Dec 11, 2024 · 2c9a75b · 2c9a75b
1 parent d508002
commit 2c9a75b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/ttnn/unit_tests/test_reshape.py b/tests/ttnn/unit_tests/test_reshape.py
@@ -314,8 +314,7 @@ def test_reshape_tile_with_padding(input_shape, output_shape, layout, device):
 
 
 # issue 15851
-def test_issue_15851():
-    device = ttnn.open_device(device_id=0)
+def test_issue_15851(device):
     x = ttnn.from_torch(torch.ones(1024), layout=ttnn.TILE_LAYOUT).to(device)
     y = ttnn.reshape(x, (32, 32))