Fix wrong call to grayskull

tenstorrent · Sep 12, 2024 · b13c422 · b13c422
1 parent f0bf7df
commit b13c422
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/lowering/misc/test_fallback_incompatible_platform.py b/tests/lowering/misc/test_fallback_incompatible_platform.py
@@ -30,7 +30,7 @@ def test_floor(device, input_shape):
     # This test will fallback to torch.ops.aten.floor.default
     nodes = list(option._out_fx_graphs[0].nodes)
     target = [node.target for node in nodes]
-    if ttnn.device.is_grayskull:
+    if ttnn.device.is_grayskull(device):
         assert target.count(ttnn.floor) == 0
         assert target.count(torch.ops.aten.floor.default) == 1
     else: