leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 29 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
include/cutlass/gemm/kernel/sm90_tile_scheduler_group.hpp
+4 -4
View File
@@ -66,18 +66,18 @@ public:
int32_t M_idx = 0; int32_t M_idx = 0;
int32_t N_idx = 0; int32_t N_idx = 0;
int32_t L_idx = 0; int32_t L_idx = 0;
bool is_valid_tile = false; int32_t is_valid_tile = 0;
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
bool bool
is_valid() const { is_valid() const {
return is_valid_tile; return is_valid_tile != 0;
} }
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
static WorkTileInfo static WorkTileInfo
invalid_work_tile() { invalid_work_tile() {
return {-1, -1, -1, false}; return {-1, -1, -1, 0};
} }
CUTLASS_HOST_DEVICE CUTLASS_HOST_DEVICE
@@ -304,7 +304,7 @@ public:
int32_t log_swizzle_size, int32_t log_swizzle_size,
RasterOrder raster_order) { 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 // Use a warp to "speculatively" check if the work tile maps to the next 32 groups
int lane_idx = canonical_lane_idx(); int lane_idx = canonical_lane_idx();