🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/164399

• 📄 Preview Python docs built from this PR
• 📄 Preview C++ docs built from this PR
• ❓ Need help or want to give feedback on the CI? Visit the bot commands wiki

Note: Links to docs will display an error until the docs builds have been completed.

✅ You can merge normally! (1 Unrelated Failure)

As of commit 1512848 with merge base b9e73e6 ():

This comment was automatically generated by Dr. CI and updates every 15 minutes.

cc @albanD

We need to allow None to be passed in as the default for autoselection probably.

what does torch.accelerator.current_accelerator() return on a device w/ no accelerator is it not cpu? That being said in retrospect there should have been no default arg for this func

what does torch.accelerator.current_accelerator() return on a device w/ no accelerator is it not cpu? That being said in retrospect there should have been no default arg for this func

Hi @drisspg, torch.accelerator.current_accelerator() will return None for CPU-only builds, since CPU is not considered an accelerator (see https://github.com/pytorch/pytorch/blob/main/docs/source/torch.rst#accelerators).

When an accelerator is available, torch.accelerator.current_accelerator() returns a torch.device without an explicit device index (e.g., torch.device("cuda")).

@pytorchbot merge

Merge started

Your change will be merged once all checks pass (ETA 0-4 Hours).

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team