From 3b28642801ab0f71b8c42bd66b463d653346dee3 Mon Sep 17 00:00:00 2001 From: reed-lau Date: Tue, 28 Sep 2021 19:04:09 +0800 Subject: [PATCH] fix wmma shape typo --- include/cutlass/arch/wmma_sm70.h | 2 +- include/cutlass/arch/wmma_sm72.h | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/include/cutlass/arch/wmma_sm70.h b/include/cutlass/arch/wmma_sm70.h index 73f8e897..3a234199 100644 --- a/include/cutlass/arch/wmma_sm70.h +++ b/include/cutlass/arch/wmma_sm70.h @@ -79,7 +79,7 @@ struct Wmma< platform::is_same, Shape>::value || platform::is_same, Shape>::value || platform::is_same, Shape>::value, - "Supported list of wmma operator shape for f16 multiplicands are: 16x16x16, 8x328x16, and 32x8x16"); + "Supported list of wmma operator shape for f16 multiplicands are: 16x16x16, 8x32x16, and 32x8x16"); // check supported wmma output data type for the given multiplicand data types static_assert( diff --git a/include/cutlass/arch/wmma_sm72.h b/include/cutlass/arch/wmma_sm72.h index c4bc1b49..2e321f0c 100644 --- a/include/cutlass/arch/wmma_sm72.h +++ b/include/cutlass/arch/wmma_sm72.h @@ -76,7 +76,7 @@ struct Wmma< platform::is_same, Shape>::value || platform::is_same, Shape>::value || platform::is_same, Shape>::value, - "Supported list of wmma operator shape for s8 multiplicands are: 16x16x16, 8x328x16, and 32x8x16"); + "Supported list of wmma operator shape for s8 multiplicands are: 16x16x16, 8x32x16, and 32x8x16"); // Wmma Fragment @@ -157,7 +157,7 @@ struct Wmma< platform::is_same, Shape>::value || platform::is_same, Shape>::value || platform::is_same, Shape>::value, - "Supported list of wmma operator shape for u8 multiplicands are: 16x16x16, 8x328x16, and 32x8x16"); + "Supported list of wmma operator shape for u8 multiplicands are: 16x16x16, 8x32x16, and 32x8x16"); // Wmma Fragment using FragmentA = nvcuda::wmma::fragment<