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

[EVT] Fix Row/Col broadcast with array arguments (#2120)

Browse Source 
* Use constexpr in if to prevent invalid comparison.

* Move constexpr check into else scope.
This commit is contained in:
Josh Fromm
2025-02-21 14:47:30 -08:00
committed by GitHub
parent afa1772203
commit eefa171318
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp
View File

@@ -1042,8 +1042,10 @@ struct Sm90RowBroadcast {
is_zero_ = params.null_default == ElementCompute(0);
}
// Dynamic non-batched scalar broadcast
else if (IsDynamicBroadcast && stride_N == bool(0) && stride_L == repeat_like(stride_L, 0) && !IsArrayOfPointers) {
is_zero_ = params.ptr_row[0] == ElementInput(0);
else if (IsDynamicBroadcast && stride_N == bool(0) && stride_L == repeat_like(stride_L, 0)) {
if constexpr (!IsArrayOfPointers) {
is_zero_ = params.ptr_row[0] == ElementInput(0);
}
}
}
@@ -1319,8 +1321,10 @@ struct Sm90ColBroadcast {
is_zero_ = params.null_default == ElementCompute(0);
}
// Dynamic non-batched scalar broadcast
else if (IsDynamicBroadcast && stride_M == bool(0) && stride_L == repeat_like(stride_L, 0) && !IsArrayOfPointers) {
is_zero_ = params.ptr_col[0] == ElementInput(0);
else if (IsDynamicBroadcast && stride_M == bool(0) && stride_L == repeat_like(stride_L, 0)) {
if constexpr (!IsArrayOfPointers) {
is_zero_ = params.ptr_col[0] == ElementInput(0);
}
}
}