CUTLASS 3.1 (#915)

Co-authored-by: Aniket Shivam <ashivam@nvidia.com>
This commit is contained in:
ANIKET SHIVAM
2023-04-14 23:19:34 -04:00
committed by GitHub
co-authored by Aniket Shivam
parent 9b8166e3f0
commit d572cc1aab
482 changed files with 37175 additions and 16410 deletions
+1 -1
View File
@@ -56,7 +56,7 @@
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
///////////////////////////////////////////// Integer wmma.mma ////////////////////////////////////////////////
///////////////////////////////////////////////////////////////////////////////////////////////////////////////
// TODO: FIXME SM75 should SM72, but the compilation breaks as SM72 shows up and runs on VOLTA
// TODO: SM75 should be SM72, but the compilation breaks as SM72 shows up and runs on VOLTA
TEST(SM75_warp_wmma_row_col_s8, 16x16x16_16x16x16_16x16x16) {
// Threadblock and warp with just one native WMMA operation (most basic unit test)
using WarpShape = cutlass::gemm::GemmShape<16, 16, 16>;