-
Notifications
You must be signed in to change notification settings - Fork 12
Open
Description
When trying to assign a device like cuda:2 to a UPETCalculator this crashes as in the MetatomicCalculator it matches on the capabilities of the model which I think are cuda or cpu.
I think this should be fixed if the specific device e.g. cuda:2 is added to the model capabilites, if the UPETCalculator is instantiated with this GPU. (and then the normal checks can still be performed whether a GPU exists etc...).
For now, I just instantiate a model on the CPU and then manually move the components to the required GPU, but fixing this should make it easier for end-users.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels