pytorch
b291c121 - Create native function for determining which implementation of SDP to call (#89029)

Commit
2 years ago
Create native function for determining which implementation of SDP to call (#89029) # Summary Creates a callable native function that can determine which implementation of scaled dot product will get called. This allows to bump re-order the runtime dispatch of SDP to enable autograd. Pull Request resolved: https://github.com/pytorch/pytorch/pull/89029 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading