Skip to content

Commit fb1be2f

Browse files
committed
add asserts
1 parent 6a1daf0 commit fb1be2f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1762,6 +1762,14 @@ NVVM::IDArgPair CpAsyncBulkTensorReduceOp::getIntrinsicIDAndArgs(
17621762
size_t redKind = static_cast<size_t>(thisOp.getRedKind());
17631763
size_t mode = static_cast<size_t>(thisOp.getMode());
17641764
size_t dim = thisOp.getCoordinates().size();
1765+
1766+
assert(redKind < IDTable.size() &&
1767+
"Invalid redKind for CpAsyncBulkTensorReduceOp");
1768+
assert(mode < IDTable[redKind].size() &&
1769+
"Invalid mode for CpAsyncBulkTensorReduceOp");
1770+
assert(dim < IDTable[redKind][mode].size() &&
1771+
"Invalid dim for CpAsyncBulkTensorReduceOp");
1772+
17651773
llvm::Intrinsic::ID intrinsicID = IDTable[redKind][mode][dim];
17661774

17671775
assert(intrinsicID != noIntrinsic &&

0 commit comments

Comments
 (0)