From 17ffd56dfee6132a4eb4aec663c9170cf0d2a73d Mon Sep 17 00:00:00 2001 From: HydraQYH Date: Fri, 21 Nov 2025 11:17:01 +0800 Subject: [PATCH] Remove unnecessary #ifdef #endif for general gemm. --- .../gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp | 3 +-- .../gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp | 4 ---- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp b/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp index 5c04259b..e9908fad 100644 --- a/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp +++ b/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp @@ -804,7 +804,7 @@ public: // Update starting mainloop pipeline state for the next tile mainloop_pipe_consumer_state.advance(work_k_tile_count); } - #ifdef CUTLASS_ENABLE_GDC_FOR_SM90 + if (scheduler.is_last_tile(work_tile_info)) { // Hint on an early release of global memory resources. // The timing of calling this function only influences performance, @@ -812,7 +812,6 @@ public: cutlass::arch::launch_dependent_grids(); } - #endif // Index of warp group within consumer warp groups int consumer_warp_group_idx = canonical_warp_group_idx() - NumLoadWarpGroups; diff --git a/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp b/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp index 073f3a50..f3d99d4e 100644 --- a/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp +++ b/include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp @@ -800,7 +800,6 @@ public: else if (warp_group_role == WarpGroupRole::Consumer0 || warp_group_role == WarpGroupRole::Consumer1) { cutlass::arch::warpgroup_reg_alloc(); - #ifdef CUTLASS_ENABLE_GDC_FOR_SM90 // It is possible to have work tiles start off invalid, // so we have to check that first. if (not work_tile_info.is_valid()) { @@ -811,7 +810,6 @@ public: return; } - #endif if constexpr (IsSchedDynamicPersistent) { // Consumer0's initial tile is static. It starts consuming the 2nd tile. @@ -868,7 +866,6 @@ public: // Update starting mainloop pipeline state for the next tile mainloop_pipe_consumer_state.advance(k_tile_count * NumMmaWarpGroups); - #ifdef CUTLASS_ENABLE_GDC_FOR_SM90 if (scheduler.is_last_tile(work_tile_info, NumMmaWarpGroups)) { // Hint on an early release of global memory resources. // The timing of calling this function only influences performance, @@ -876,7 +873,6 @@ public: cutlass::arch::launch_dependent_grids(); } - #endif // Order two Math WG's Epilogue one after the other math_wg_order_barrier.wait();