mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
e7a42702f9
Pull Request resolved: https://github.com/pytorch/pytorch/pull/126531 Approved by: https://github.com/jgong5, https://github.com/gujinghui, https://github.com/albanD, https://github.com/EikanWang ghstack dependencies: #126527
10 lines
181 B
Python
10 lines
181 B
Python
from .autocast_mode import (
|
|
_enter_autocast,
|
|
_exit_autocast,
|
|
autocast,
|
|
custom_bwd,
|
|
custom_fwd,
|
|
is_autocast_available,
|
|
)
|
|
from .grad_scaler import GradScaler
|