From 89f6bf27393a7fccc9500b30c8da6079b6d8b13a Mon Sep 17 00:00:00 2001 From: Jiazhen Han Date: Fri, 2 May 2025 12:41:18 -0700 Subject: [PATCH] Fix group scale gemm when K==128 (#2275) Co-authored-by: Jiazhen Han --- ...90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/include/cutlass/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp b/include/cutlass/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp index a931e988..5b8f1059 100644 --- a/include/cutlass/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp +++ b/include/cutlass/gemm/collective/sm90_mma_tma_gmma_ss_warpspecialized_fp8_blockwise_scaling.hpp @@ -1001,7 +1001,7 @@ struct CollectiveMma< // Advance smem_pipe_read and smem_pipe_release ++smem_pipe_release; } - { + if (k_tile_count == 1) { pipeline.consumer_wait(smem_pipe_read, barrier_token); //