From f0f697ae4ac66ef1806ce1ad2091fb8cb2afe3f2 Mon Sep 17 00:00:00 2001 From: Haocong WANG Date: Sun, 16 Apr 2023 10:57:34 +0800 Subject: [PATCH] Fix a typo (#676) [ROCm/composable_kernel commit: fc26d42a2e1f5f34c21f8f5e7aaa2fb381b2aceb] --- include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp index 2694aaf6f0..397ae1c1b9 100644 --- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp +++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp @@ -266,8 +266,8 @@ struct GridwiseGemm_k0mk1_k0nk1_mn_wmma // TODO: also check validity of all components (blockwise-copy, threadwise-copy, etc) constexpr long_index_t TwoGB = (long_index_t{1} << 31); - if(!(a_grid_desc_k0_m_k1.GetElementSpaceSize() * sizeof(ADataType) <= TwoGB && - b_grid_desc_k0_n_k1.GetElementSpaceSize() * sizeof(BDataType) <= TwoGB)) + if(!(a_grid_desc_k0_m_k1.GetElementSpaceSize() * sizeof(FloatA) <= TwoGB && + b_grid_desc_k0_n_k1.GetElementSpaceSize() * sizeof(FloatB) <= TwoGB)) { return false; }