mirrors/llvm-project
0
0
Fork 0
mirror of https://github.com/llvm/llvm-project.git synced 2025-05-01 15:26:09 +00:00
Code Issues Projects Releases Wiki Activity

[mlir] Use llvm::is_contained (NFC)

Browse Source
This commit is contained in:
Kazu Hirata 2023-09-02 12:12:12 -07:00
parent d9ffd3219e
commit b95028ee2d
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/lib/Dialect/LLVMIR/IR/NVVMDialect.cpp
View File

@ -772,13 +772,13 @@ LogicalResult isAllowedSizeN(int sizeN, NVVM::WGMMATypes typeA) {
case mlir::NVVM::WGMMATypes::bf16: case mlir::NVVM::WGMMATypes::bf16:
case mlir::NVVM::WGMMATypes::e4m3: case mlir::NVVM::WGMMATypes::e4m3:
case mlir::NVVM::WGMMATypes::e5m2: case mlir::NVVM::WGMMATypes::e5m2:
if (llvm::any_of(allowedN, [&](int n) { return sizeN == n; })) if (llvm::is_contained(allowedN, sizeN))
return success(); return success();
break; break;
case mlir::NVVM::WGMMATypes::u8: case mlir::NVVM::WGMMATypes::u8:
case mlir::NVVM::WGMMATypes::s8: case mlir::NVVM::WGMMATypes::s8:
case mlir::NVVM::WGMMATypes::b1: case mlir::NVVM::WGMMATypes::b1:
if (llvm::any_of(allowedNshort, [&](int n) { return sizeN == n; })) if (llvm::is_contained(allowedNshort, sizeN))
return success(); return success();
} }
return failure(); return failure();

2
mlir/lib/IR/BuiltinAttributes.cpp
View File

@ -235,7 +235,7 @@ AffineMap StridedLayoutAttr::getAffineMap() const {
LogicalResult LogicalResult
StridedLayoutAttr::verify(function_ref<InFlightDiagnostic()> emitError, StridedLayoutAttr::verify(function_ref<InFlightDiagnostic()> emitError,
int64_t offset, ArrayRef<int64_t> strides) { int64_t offset, ArrayRef<int64_t> strides) {
if (llvm::any_of(strides, [&](int64_t stride) { return stride == 0; })) if (llvm::is_contained(strides, 0))
return emitError() << "strides must not be zero"; return emitError() << "strides must not be zero";
return success(); return success();