leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Remove unnecessary #ifdef / #endif for launch_dependent_grids.

Browse Source
This commit is contained in:
HydraQYH
2025-11-18 08:45:40 +08:00
parent b6ad6db219
commit 929e1e0259
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/cutlass/gemm/kernel/sm90_gemm_array_tma_warpspecialized_cooperative.hpp
View File

@@ -998,11 +998,11 @@ public:
// Get next work tile
auto [next_work_tile_info, increment_pipe] = scheduler.fetch_next_work(work_tile_info, tile_scheduler_pipeline, tile_scheduler_pipe_consumer_state);
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
if (!next_work_tile_info.is_valid()) {
cutlass::arch::launch_dependent_grids();
}
#endif
work_tile_info = next_work_tile_info;
if (increment_pipe) {
++tile_scheduler_pipe_consumer_state;

4
include/cutlass/gemm/kernel/sm90_gemm_array_tma_warpspecialized_pingpong.hpp
View File

@@ -1042,11 +1042,11 @@ public:
// Get next work tile
auto [next_work_tile_info, increment_pipe] = scheduler.fetch_next_work(work_tile_info, tile_scheduler_pipeline, tile_scheduler_pipe_consumer_state);
#ifdef CUTLASS_ENABLE_GDC_FOR_SM90
if (!next_work_tile_info.is_valid()) {
cutlass::arch::launch_dependent_grids();
}
#endif
work_tile_info = next_work_tile_info;
if (increment_pipe) {
++tile_scheduler_pipe_consumer_state;