debug

example/01_gemm/gemm_wmma_fp16.cpp

@@ -40,7 +40,7 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmWmma_CShuffle
            64,          // MPerBlock
            128,         // NPerBlock
            64,          // KPerBlock
-           8,           // K1
+           4,           // K1
            16,          // MPerWmma
            16,          // NPerWmma
            2,           // M-Repeat // M-PerWmma / M-Repeat = M-Wave
@@ -49,15 +49,15 @@ using DeviceGemmInstance = ck::tensor_operation::device::DeviceGemmWmma_CShuffle
            S<1, 0, 2>,
            S<1, 0, 2>,
            2,
-           8,
-           8,
+           1,
+           1,
            true,
            S<4, 32, 1>,
            S<1, 0, 2>,
            S<1, 0, 2>,
            2,
-           8,
-           8,
+           1,
+           1,
            true,
            1,           // C shuffle (M Repeat) Per store
            1,           // C shuffle (N Repeat) Per store

include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp

@@ -781,6 +781,8 @@ struct GridwiseGemm_Wmma
         // GEMM
         constexpr auto KPack = math::integer_least_multiple(K1, WmmaK);
 
+	static_assert(KPerBlock % KPack == 0, "");
+
         auto blockwise_gemm =
             BlockwiseGemmWMMA<BlockSize,
                               ADataType,