dan_the_3rd
|
9b8166e3f0
|
fMHA: Add backward pass (#844)
* fMHA: Add backward pass
* Better checks for strides/alignments
* Remove fb-internal URL
* torch.Tensor.untyped_storage requires pytorch 2.0+
* minor changes
* make test
---------
Co-authored-by: danthe3rd <danthe3rd>
Co-authored-by: Haicheng Wu <haichengw@nvidia.com>
|
2023-04-06 20:44:58 -04:00 |
|