-
Notifications
You must be signed in to change notification settings - Fork 76
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Improve device support and add support for Apple Silicon chipset (mps
)
#34
Improve device support and add support for Apple Silicon chipset (mps
)
#34
Conversation
Co-authored-by: Steffen Schneider <[email protected]>
Co-authored-by: Steffen Schneider <[email protected]>
Co-authored-by: Steffen Schneider <[email protected]>
663e50d
to
e1e6848
Compare
Co-authored-by: Steffen Schneider <[email protected]>
Co-authored-by: Steffen Schneider <[email protected]>
Co-authored-by: Steffen Schneider <[email protected]>
11c6acf
to
42965ea
Compare
Based on the branch from this pull request, I've also extended the device support to more generic GPU devices with Is the dev team also interested in making |
…BRA into rodrigo/move-model-to-device
mps
)
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's add more tests to verify the functionality.
…BRA into rodrigo/move-model-to-device
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
lgtm - didn't directly test myself though :)
This pull request improves device support for sklearn API models, enabling transfer between CPU, CUDA and Apple Silicon GPU environments.
Fix https://github.com/AdaptiveMotorControlLab/CEBRA-dev/pull/631