up

metascroy · metascroy · commit fe8410b02b65 · 2025-03-04T10:07:30.000-08:00
diff --git a/setup.py b/setup.py
@@ -109,6 +109,7 @@ def __init__(self):
         )
         if self.build_experimental_mps:
             assert self._is_macos(), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires MacOS"
+            assert self._is_arm64(), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires arm64"
             assert (
                 torch.mps.is_available()
             ), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires MPS be available"

Original file line number	Diff line number	Diff line change
`@@ -109,6 +109,7 @@ def __init__(self):`
`109`	`109`	`)`
`110`	`110`	`if self.build_experimental_mps:`
`111`	`111`	`assert self._is_macos(), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires MacOS"`
	`112`	`+ assert self._is_arm64(), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires arm64"`
`112`	`113`	`assert (`
`113`	`114`	`torch.mps.is_available()`
`114`	`115`	`), "TORCHAO_BUILD_EXPERIMENTAL_MPS requires MPS be available"`