cosdt / torch_backend

https://cosdt.github.io/torch_backend/
Other
3 stars 0 forks source link

Keep up to date #81

Open FFFrog opened 2 months ago

FFFrog commented 2 months ago
Type Description PR
Refactoring Re-implement pin_memory to be device-agnostic by leveraging the Accelerator concept. #126376
Refactoring ​​generalize custom_fwd&custom_bwd to be device-agnostic. #126531
Refactoring Refactor autocast C++ APIs to be device-agnostic. #124359
Refactoring refactor autocast python APIs. #124479
Refactoring refactor lazy init to device-agnostic. #118846
Refactoring Generalize host allocator to be device-agnostic. #123079
shink commented 2 months ago

https://github.com/pytorch/pytorch/pulls?q=is%3Apr+device+agnostic+is%3Aclosed+label%3AMerged