File tree Expand file tree Collapse file tree 1 file changed +8
-0
lines changed
mlir/lib/Dialect/LLVMIR/IR Expand file tree Collapse file tree 1 file changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -1762,6 +1762,14 @@ NVVM::IDArgPair CpAsyncBulkTensorReduceOp::getIntrinsicIDAndArgs(
1762
1762
size_t redKind = static_cast <size_t >(thisOp.getRedKind ());
1763
1763
size_t mode = static_cast <size_t >(thisOp.getMode ());
1764
1764
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
+
1765
1773
llvm::Intrinsic::ID intrinsicID = IDTable[redKind][mode][dim];
1766
1774
1767
1775
assert (intrinsicID != noIntrinsic &&
You can’t perform that action at this time.
0 commit comments