leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 28 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 17:47:30 -05: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
include/cutlass/epilogue/fusion/sm90_visitor_load_tma_warpspecialized.hpp
+8 -4
View File
@@ -1042,8 +1042,10 @@ struct Sm90RowBroadcast {
is_zero_ = params.null_default == ElementCompute(0); is_zero_ = params.null_default == ElementCompute(0);
} }
// Dynamic non-batched scalar broadcast // Dynamic non-batched scalar broadcast
else if (IsDynamicBroadcast && stride_N == bool(0) && stride_L == repeat_like(stride_L, 0) && !IsArrayOfPointers) { else if (IsDynamicBroadcast && stride_N == bool(0) && stride_L == repeat_like(stride_L, 0)) {
is_zero_ = params.ptr_row[0] == ElementInput(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); is_zero_ = params.null_default == ElementCompute(0);
} }
// Dynamic non-batched scalar broadcast // Dynamic non-batched scalar broadcast
else if (IsDynamicBroadcast && stride_M == bool(0) && stride_L == repeat_like(stride_L, 0) && !IsArrayOfPointers) { else if (IsDynamicBroadcast && stride_M == bool(0) && stride_L == repeat_like(stride_L, 0)) {
is_zero_ = params.ptr_col[0] == ElementInput(0); if constexpr (!IsArrayOfPointers) {
is_zero_ = params.ptr_col[0] == ElementInput(0);
}
} }
} }