Skip to content
This repository was archived by the owner on Aug 15, 2025. It is now read-only.

Commit f8e5f34

Browse files
authored
Fix torch.compile validation from mps
1 parent f8b4485 commit f8e5f34

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

test/smoke_test/smoke_test.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ def test_linalg(device="cpu") -> None:
243243

244244
def smoke_test_compile() -> None:
245245
supported_dtypes = [torch.float16, torch.float32, torch.float64]
246-
dv = "cuda" if target_os == "linux" else "mps"
246+
dv = "cuda" if target_os == "linux" else "cpu"
247247

248248
def foo(x: torch.Tensor) -> torch.Tensor:
249249
return torch.sin(x) + torch.cos(x)

0 commit comments

Comments
 (0)