Skip to content
/ TorchDevice Public

Module contains class TorchDevice abstracting CUDA/MPS/CPU, intercepting, and redirecting calls to the available device to assist in porting code from CUDA to MPS.

License

Apache-2.0 license
4 stars 1 fork Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

unixwzrd/TorchDevice

Error
Looks like something went wrong!

About

Module contains class TorchDevice abstracting CUDA/MPS/CPU, intercepting, and redirecting calls to the available device to assist in porting code from CUDA to MPS.

Resources

Readme

License

Apache-2.0 license
Activity

Stars

4 stars

Watchers

1 watching

Forks

1 fork
Report repository

Releases 6

0.2.0 Interim dev release - very beta Latest
May 24, 2025
+ 5 releases

Sponsor this project

Packages

No packages published

Languages