Christian Puhrsch
|
943b20e7ae
|
Use tensor cores for NT bmm (#86856)
Copy of internal diff.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/86856
Approved by: https://github.com/drisspg
|
2022-11-02 21:51:40 +00:00 |
|
Christian Puhrsch
|
f6c6048b10
|
Use CUTLASS GEMM for NT bmm (#85894)
Copy of https://github.com/pytorch/pytorch/pull/85710
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85894
Approved by: https://github.com/drisspg
|
2022-10-18 23:11:47 +00:00 |
|
PyTorch MergeBot
|
d169f950da
|
Revert "Use CUTLASS GEMM for NT bmm [OSS-only] (#85894)"
This reverts commit ef58a132f2.
Reverted https://github.com/pytorch/pytorch/pull/85894 on behalf of https://github.com/DanilBaibak due to Break internal build
|
2022-10-13 15:28:09 +00:00 |
|
Christian Puhrsch
|
ef58a132f2
|
Use CUTLASS GEMM for NT bmm [OSS-only] (#85894)
OSS-only copy of https://github.com/pytorch/pytorch/pull/85710
Pull Request resolved: https://github.com/pytorch/pytorch/pull/85894
Approved by: https://github.com/drisspg
|
2022-10-12 20:03:28 +00:00 |
|