Skip to content

Commit 7d40287

Browse files
committed
Remove unnecessary #ifdef / #endif for launch_dependent_grids.
1 parent cf09543 commit 7d40287

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

include/cutlass/gemm/kernel/sm90_gemm_array_tma_warpspecialized_cooperative.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -994,11 +994,11 @@ class GemmUniversal<
994994

995995
// Get next work tile
996996
auto [next_work_tile_info, increment_pipe] = scheduler.fetch_next_work(work_tile_info, tile_scheduler_pipeline, tile_scheduler_pipe_consumer_state);
997-
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
997+
998998
if (!next_work_tile_info.is_valid()) {
999999
cutlass::arch::launch_dependent_grids();
10001000
}
1001-
#endif
1001+
10021002
work_tile_info = next_work_tile_info;
10031003
if (increment_pipe) {
10041004
++tile_scheduler_pipe_consumer_state;

include/cutlass/gemm/kernel/sm90_gemm_array_tma_warpspecialized_pingpong.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1038,11 +1038,11 @@ class GemmUniversal<
10381038

10391039
// Get next work tile
10401040
auto [next_work_tile_info, increment_pipe] = scheduler.fetch_next_work(work_tile_info, tile_scheduler_pipeline, tile_scheduler_pipe_consumer_state);
1041-
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
1041+
10421042
if (!next_work_tile_info.is_valid()) {
10431043
cutlass::arch::launch_dependent_grids();
10441044
}
1045-
#endif
1045+
10461046
work_tile_info = next_work_tile_info;
10471047
if (increment_pipe) {
10481048
++tile_scheduler_pipe_consumer_state;

0 commit comments

Comments
 (0)