Skip to content

Commit

Permalink
wip
Browse files Browse the repository at this point in the history
  • Loading branch information
baptistecolle committed Dec 4, 2024
1 parent db8c3ca commit cbe667c
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions .github/workflows/debug-metadata.yml
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,4 @@ jobs:
-v /mnt/hf_cache:/mnt/cache/ \
-e PJRT_DEVICE=TPU \
us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/xla@sha256:8f1dcd5b03f993e4da5c20d17c77aff6a5f22d5455f8eb042d2e4b16ac460526 \
bash -c '
python -c "import torch_xla.core.xla_model as xm; assert xm.xla_device().type == 'xla', 'XLA device not available'"
'
bash -c "python -c 'import torch_xla.core.xla_model as xm; assert xm.xla_device().type == \"xla\", \"XLA device not available\"'"

0 comments on commit cbe667c

Please sign in to comment.