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

fix blackwell grouped groupwise hang (#2267)

Browse Source
This commit is contained in:
Haicheng Wu
2025-04-29 11:54:20 -04:00
committed by GitHub
parent e94e888df3
commit 697126019e
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
include/cutlass/gemm/kernel/sm90_tile_scheduler_group.hpp
View File

@ -66,18 +66,18 @@ public:
int32_t M_idx = 0;
int32_t N_idx = 0;
int32_t L_idx = 0;
bool is_valid_tile = false;
int32_t is_valid_tile = 0;
CUTLASS_HOST_DEVICE
bool
is_valid() const {
return is_valid_tile;
return is_valid_tile != 0;
}
CUTLASS_HOST_DEVICE
static WorkTileInfo
invalid_work_tile() {
return {-1, -1, -1, false};
return {-1, -1, -1, 0};
}
CUTLASS_HOST_DEVICE
@ -304,7 +304,7 @@ public:
int32_t log_swizzle_size,
RasterOrder raster_order) {
bool valid_tile = true;
int32_t valid_tile = 1;
// Use a warp to "speculatively" check if the work tile maps to the next 32 groups
int lane_idx = canonical_lane_idx();