This repository was archived by the owner on Feb 7, 2025. It is now read-only.
This repository was archived by the owner on Feb 7, 2025. It is now read-only.
Keep up to date #81
Description
| 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 |