diff --git a/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h b/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h index 83157b60c590b..b27c9e81b3293 100644 --- a/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h +++ b/mlir/include/mlir/Dialect/Mesh/IR/MeshOps.h @@ -114,7 +114,7 @@ inline int64_t shardDimension(int64_t dimSize, int64_t shardCount) { return ShapedType::kDynamic; assert(dimSize % shardCount == 0); - return llvm::divideCeilSigned(dimSize, shardCount); + return dimSize / shardCount; } // Get the size of an unsharded dimension. diff --git a/mlir/lib/Conversion/LLVMCommon/MemRefBuilder.cpp b/mlir/lib/Conversion/LLVMCommon/MemRefBuilder.cpp index 1b2d0258130cb..19c3ba1f95020 100644 --- a/mlir/lib/Conversion/LLVMCommon/MemRefBuilder.cpp +++ b/mlir/lib/Conversion/LLVMCommon/MemRefBuilder.cpp @@ -365,7 +365,7 @@ void UnrankedMemRefDescriptor::computeSizes( Value two = createIndexAttrConstant(builder, loc, indexType, 2); Value indexSize = createIndexAttrConstant( builder, loc, indexType, - llvm::divideCeilSigned(typeConverter.getIndexTypeBitwidth(), 8)); + llvm::divideCeil(typeConverter.getIndexTypeBitwidth(), 8)); sizes.reserve(sizes.size() + values.size()); for (auto [desc, addressSpace] : llvm::zip(values, addressSpaces)) { @@ -378,8 +378,7 @@ void UnrankedMemRefDescriptor::computeSizes( // to data layout) into the unranked descriptor. Value pointerSize = createIndexAttrConstant( builder, loc, indexType, - llvm::divideCeilSigned(typeConverter.getPointerBitwidth(addressSpace), - 8)); + llvm::divideCeil(typeConverter.getPointerBitwidth(addressSpace), 8)); Value doublePointerSize = builder.create(loc, indexType, two, pointerSize); diff --git a/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp b/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp index 1933c1dfcfba4..054827d40f0f3 100644 --- a/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp +++ b/mlir/lib/Conversion/MemRefToLLVM/MemRefToLLVM.cpp @@ -971,7 +971,7 @@ struct MemorySpaceCastOpLowering resultUnderlyingDesc, resultElemPtrType); int64_t bytesToSkip = - 2 * llvm::divideCeilSigned( + 2 * llvm::divideCeil( getTypeConverter()->getPointerBitwidth(resultAddrSpace), 8); Value bytesToSkipConst = rewriter.create( loc, getIndexType(), rewriter.getIndexAttr(bytesToSkip));