pytorch / pytorch/pytorch

`torch.accelerator.is_available()` fails when privateuseone is used

Open
#166,696 2 comments 0 reactions 0 assignees View on GitHub
module: accelerator triaged
Dominant language
Python
Stars
103k
Forks
29.5k
PR merge metrics
PR metrics pending

Description

### 🐛 Describe the bug

Run `PYTORCH_TEST_WITH_DYNAMO=1 python test/test_rename_privateuse1_to_existing_device.py TestRenamePrivateuseoneToExistingBackend.test_external_module_register_with_existing_backend`

on 3733909afd07af889a9a245d0315853d17203da2

I skipped the test for now, since it's only exposed when dynamo interprets it.

https://gist.github.com/mlazos/50954c6c956e313d9a70fe9584f801b7

### Versions

3733909afd07af889a9a245d0315853d17203da2

cc @albanD @guangyey @EikanWang

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.