frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use CUTLASS GEMM for NT bmm (#85894)

Browse Source 
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
This commit is contained in:
Christian Puhrsch
2022-10-18 23:11:47 +00:00
committed by PyTorch MergeBot
parent 80790ecee4
commit f6c6048b10
10 changed files with 400 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
WORKSPACE
View File

@ -84,6 +84,12 @@ new_local_repository(
path = "third_party/eigen",
)
new_local_repository(
name = "cutlass",
build_file = "//third_party:cutlass.BUILD",
path = "third_party/cutlass",
)
new_local_repository(
name = "fbgemm",
build_file = "//third_party:fbgemm/BUILD.bazel",