pytorch
c5a48440 - Xformer SDP forward/backward kernel (#86157)

Commit
2 years ago
Xformer SDP forward/backward kernel (#86157) # Summary Include xformer kernel code and make header updates to successfully build. Need to update the kernel calling code and dispatch system to clean this up. Pull Request resolved: https://github.com/pytorch/pytorch/pull/86157 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading