Skip to content

DEMO: testing torch device MPS with cirrus CI #71

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 4 commits into
base: array_api
Choose a base branch
from
Open

DEMO: testing torch device MPS with cirrus CI #71

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
96e0cce
TST: support torch device testing
tylerjereddy May 16, 2023
2aead3e
CI: try activating cirrus for my fork.
tylerjereddy May 16, 2023
e1e3480
CI: add torch
tylerjereddy May 16, 2023
4bb004d
CI: add array-api-compat
tylerjereddy May 16, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .cirrus.star
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@ def main(ctx):
# - commit message containing [wheel build]
######################################################################

if env.get("CIRRUS_REPO_FULL_NAME") != "scipy/scipy":
if env.get("CIRRUS_REPO_FULL_NAME") != "tylerjereddy/scipy":
return []

if env.get("CIRRUS_CRON", "") == "nightly":
Expand Down
4 changes: 3 additions & 1 deletion ci/cirrus_general_ci.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -143,4 +143,6 @@ macos_arm64_test_task:
python -m pip install click rich_click doit pydevtool
python -m pip install pytest pooch
export DYLD_LIBRARY_PATH=/usr/local/gfortran/lib:/opt/arm64-builds/lib
python dev.py test
python -m pip install torch array-api-compat
# spot check array API torch MPS device compliance
SCIPY_TORCH_DEVICE=mps python dev.py test -b numpy -b pytorch -s cluster
4 changes: 4 additions & 0 deletions scipy/conftest.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -132,6 +132,10 @@ def check_fpu_mode(request):

array_api_compatible = pytest.mark.parametrize("xp", array_api_backends)

if "pytorch" in array_api_available_backends:
torch_device_setting = os.environ.get("SCIPY_TORCH_DEVICE", "cpu")
torch.set_default_device(torch_device_setting)

skip_if_array_api = pytest.mark.skipif(
SCIPY_ARRAY_API,
reason="do not run with Array API on",
Expand Down