You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
@etoyz Our latest torch_directml 0.2.4.dev240815 release adds support for device_count with PrivateUse1. Please run pip install torch-directml --upgrade to update to the latest build
Reproduce:
When the second parameter of
torch.nn.DataParallel()
is None,Then
torch._utils._get_all_device_indices()
will be called.Then
lambda m: list(range(m.device_count()))
will be called, and it can be proved that them
is<class 'torch_directml.PrivateUse1Module'>
torch_directml.PrivateUse1Module.device_count()
seems not to be implemented ???The text was updated successfully, but these errors were encountered: