mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary:
- make init to be device-agnostic and move it to AcceleratorHooksInterface
- refactoring context related to device initialization
Original pull request: https://github.com/pytorch/pytorch/pull/136519
Test Plan: contbuild & OSS CI, see
|
||
|---|---|---|
| .. | ||
| __init__.py | ||
| lazy_ir.py | ||
| lazy_ts_lowering.py | ||
| native_functions.py | ||
| register_dispatch_key.py | ||
| ufunc.py | ||