pytorch
de86146c - rocblas alt impl during backward pass only (#71881)

Commit
2 years ago
rocblas alt impl during backward pass only (#71881) In preparation of adopting future rocblas library options, it is necessary to track when the backward pass of training is executing. The scope-based helper class `BackwardPassGuard` is provided to toggle state. Pull Request resolved: https://github.com/pytorch/pytorch/pull/71881 Approved by: https://github.com/albanD
Author
Committer
Parents
Loading