Skip to content

Commit eedd1ce

Browse files
[Backend] Assert that num threads is always power of 2 (NFC) (#7251) (#4899)
(cherry picked from commit 235496e) Co-authored-by: Jeff Niu <[email protected]>
1 parent d872350 commit eedd1ce

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

lib/Conversion/TritonGPUToLLVM/Utility.cpp

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -296,10 +296,9 @@ Value getThreadId(OpBuilder &rewriter, Location loc) {
296296
tid = rewriter.create<arith::SubIOp>(loc, tid, b.i32_val(*startId));
297297
}
298298

299-
if (llvm::isPowerOf2_32(upperBound)) {
300-
// help LLVM's known bits analysis:
301-
tid = b.and_(tid, b.i32_val(upperBound - 1));
302-
}
299+
assert(llvm::isPowerOf2_32(upperBound));
300+
// help LLVM's known bits analysis:
301+
tid = b.and_(tid, b.i32_val(upperBound - 1));
303302

304303
return tid;
305304
}

0 commit comments

Comments
 (0)