Skip to content

Commit 867dfbf

Browse files
committed
[GPU] Update MemDescSubsliceOp verification to adjust splitOffset increment logic for improved handling of CTA dimensions
1 parent dc2dfbc commit 867dfbf

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

lib/Dialect/TritonGPU/IR/Ops.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1020,7 +1020,7 @@ LogicalResult MemDescSubsliceOp::verify() {
10201020
// Splitting at `dimSize` is valid as long as all points in [0, dimSize)
10211021
// stay within the same CTA.
10221022
for (int splitOffset = 0; splitOffset < dstTy.getDimSize(dim);
1023-
++splitOffset) {
1023+
splitOffset *= 2) {
10241024
namedOffsets[dim] = {kDim, splitOffset};
10251025
for (auto [inDim, val] : llInv.apply(namedOffsets)) {
10261026
if (inDim == kBlock && val != 0) {

0 commit comments

Comments
 (0)