pytorch
f865e23a - [MPS] Introduce MPSUnaryGradCachedGraph & MPSBinaryGradCachedGraph (#95289)

Commit
1 year ago
[MPS] Introduce MPSUnaryGradCachedGraph & MPSBinaryGradCachedGraph (#95289) This PR introduces `MPSUnaryGradCachedGraph` & `MPSBinaryGradCachedGraph` to replace duplicate CachedGraph creation in backward functions. Pull Request resolved: https://github.com/pytorch/pytorch/pull/95289 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading