Skip to content

Commit f6e1d5a

Browse files
committed
Remove unnecessary #ifdef #endif for general gemm.
1 parent a3a8e0a commit f6e1d5a

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_cooperative.hpp

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -804,15 +804,14 @@ class GemmUniversal<
804804
// Update starting mainloop pipeline state for the next tile
805805
mainloop_pipe_consumer_state.advance(work_k_tile_count);
806806
}
807-
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
807+
808808
if (scheduler.is_last_tile(work_tile_info)) {
809809
// Hint on an early release of global memory resources.
810810
// The timing of calling this function only influences performance,
811811
// not functional correctness.
812812
cutlass::arch::launch_dependent_grids();
813813
814814
}
815-
#endif
816815
817816
// Index of warp group within consumer warp groups
818817
int consumer_warp_group_idx = canonical_warp_group_idx() - NumLoadWarpGroups;

include/cutlass/gemm/kernel/sm90_gemm_tma_warpspecialized_pingpong.hpp

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -800,7 +800,6 @@ class GemmUniversal<
800800
else if (warp_group_role == WarpGroupRole::Consumer0 || warp_group_role == WarpGroupRole::Consumer1) {
801801
cutlass::arch::warpgroup_reg_alloc<MmaRegisterRequirement>();
802802
803-
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
804803
// It is possible to have work tiles start off invalid,
805804
// so we have to check that first.
806805
if (not work_tile_info.is_valid()) {
@@ -811,7 +810,6 @@ class GemmUniversal<
811810
812811
return;
813812
}
814-
#endif
815813
816814
if constexpr (IsSchedDynamicPersistent) {
817815
// Consumer0's initial tile is static. It starts consuming the 2nd tile.
@@ -868,15 +866,13 @@ class GemmUniversal<
868866
// Update starting mainloop pipeline state for the next tile
869867
mainloop_pipe_consumer_state.advance(k_tile_count * NumMmaWarpGroups);
870868

871-
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
872869
if (scheduler.is_last_tile(work_tile_info, NumMmaWarpGroups)) {
873870
// Hint on an early release of global memory resources.
874871
// The timing of calling this function only influences performance,
875872
// not functional correctness.
876873
cutlass::arch::launch_dependent_grids();
877874

878875
}
879-
#endif
880876

881877
// Order two Math WG's Epilogue one after the other
882878
math_wg_order_barrier.wait();

0 commit comments

Comments
 (0)