| /llvm-project-15.0.7/mlir/lib/Dialect/Utils/ |
| H A D | StaticValueUtils.cpp | 22 void dispatchIndexOpFoldResult(OpFoldResult ofr, in dispatchIndexOpFoldResult() argument
26 auto v = ofr.dyn_cast<Value>(); in dispatchIndexOpFoldResult()
28 APInt apInt = ofr.get<Attribute>().cast<IntegerAttr>().getValue(); in dispatchIndexOpFoldResult()
40 for (OpFoldResult ofr : ofrs) in dispatchIndexOpFoldResults() local
41 dispatchIndexOpFoldResult(ofr, dynamicVec, staticVec, sentinel); in dispatchIndexOpFoldResults()
78 Optional<int64_t> getConstantIntValue(OpFoldResult ofr) { in getConstantIntValue() argument
80 if (auto val = ofr.dyn_cast<Value>()) { in getConstantIntValue()
87 Attribute attr = ofr.dyn_cast<Attribute>(); in getConstantIntValue()
94 bool isConstantIntValue(OpFoldResult ofr, int64_t value) { in isConstantIntValue() argument
95 auto val = getConstantIntValue(ofr); in isConstantIntValue()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Arithmetic/Utils/ |
| H A D | Utils.cpp | 31 for (OpFoldResult &ofr : values) { in canonicalizeSubViewPart()
32 if (ofr.is<Attribute>()) in canonicalizeSubViewPart()
36 ofr.dyn_cast<Value>().getDefiningOp<arith::ConstantIndexOp>()) in canonicalizeSubViewPart()
37 ofr = OpBuilder(cstOp).getIndexAttr(cstOp.value()); in canonicalizeSubViewPart()
54 OpFoldResult ofr) { in getValueOrCreateConstantIndexOp() argument
55 if (auto value = ofr.dyn_cast<Value>()) in getValueOrCreateConstantIndexOp()
57 auto attr = ofr.dyn_cast<Attribute>().dyn_cast<IntegerAttr>(); in getValueOrCreateConstantIndexOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Utils/ |
| H A D | StaticValueUtils.h | 30 void dispatchIndexOpFoldResult(OpFoldResult ofr,
57 Optional<int64_t> getConstantIntValue(OpFoldResult ofr);
60 bool isConstantIntValue(OpFoldResult ofr, int64_t value);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Transforms/ |
| H A D | SwapExtractSliceWithProducer.cpp | 30 if (llvm::any_of(sliceOp.getMixedStrides(), [](OpFoldResult ofr) { in replaceExtractSliceWithTiledProducer() argument
31 return !isConstantIntValue(ofr, 1); in replaceExtractSliceWithTiledProducer()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Hoisting.cpp | 60 auto getConstantIntValue = [](OpFoldResult ofr) -> llvm::Optional<int64_t> { in isEqualOffsetSizeOrStride() argument
61 Attribute attr = ofr.dyn_cast<Attribute>(); in isEqualOffsetSizeOrStride()
63 if (!attr && ofr.get<Value>().getDefiningOp<arith::ConstantOp>()) in isEqualOffsetSizeOrStride()
64 attr = ofr.get<Value>().getDefiningOp<arith::ConstantOp>().getValue(); in isEqualOffsetSizeOrStride()
|
| H A D | Tiling.cpp | 237 llvm::to_vector(llvm::make_filter_range(numThreads, [](OpFoldResult ofr) { in tileToForeachThreadOpImpl() argument
238 return !isConstantIntValue(ofr, 0); in tileToForeachThreadOpImpl()
241 llvm::to_vector(llvm::map_range(nonZeroNumThreads, [&](OpFoldResult ofr) { in tileToForeachThreadOpImpl() argument
243 return materializeOpFoldResult(ilocb, ofr); in tileToForeachThreadOpImpl()
|
| H A D | Transforms.cpp | 754 auto getIdxValue = [&](OpFoldResult ofr) { in matchAndRewrite() argument
755 if (auto val = ofr.dyn_cast<Value>()) in matchAndRewrite()
759 padOp.getLoc(), ofr.get<Attribute>().cast<IntegerAttr>().getInt()) in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | ViewLikeInterface.td | 421 return ::llvm::all_of(getMixedStrides(), [](::mlir::OpFoldResult ofr) {
422 return ::mlir::getConstantIntValue(ofr) == static_cast<int64_t>(1);
433 return ::llvm::all_of(getMixedOffsets(), [](::mlir::OpFoldResult ofr) {
434 return ::mlir::getConstantIntValue(ofr) == static_cast<int64_t>(0);
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Arithmetic/Utils/ |
| H A D | Utils.h | 81 OpFoldResult ofr);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 351 bool hasNonUnitStride = any_of(allStrides, [](OpFoldResult ofr) { in makeComposedExtractSliceOp() argument
352 return getConstantIntValue(ofr) != static_cast<int64_t>(1); in makeComposedExtractSliceOp()
404 if (llvm::any_of(padOp.getMixedLowPad(), [](OpFoldResult ofr) { in makeComposedPadHighOp() argument
405 return getConstantIntValue(ofr) != static_cast<int64_t>(0); in makeComposedPadHighOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | OpDefinition.h | 230 raw_ostream &operator<<(raw_ostream &os, OpFoldResult ofr);
241 inline raw_ostream &operator<<(raw_ostream &os, OpFoldResult ofr) {
242 if (Value value = ofr.dyn_cast<Value>())
245 ofr.dyn_cast<Attribute>().print(os);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 910 for (OpFoldResult ofr : getMixedSizes()) { in apply() local
911 if (auto attr = ofr.dyn_cast<Attribute>()) { in apply()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/ |
| H A D | TensorOps.td | 1022 return llvm::all_of(getMixedLowPad(), [](OpFoldResult ofr) {
1023 return getConstantIntValue(ofr) == static_cast<int64_t>(0);
1028 return llvm::all_of(getMixedHighPad(), [](OpFoldResult ofr) {
1029 return getConstantIntValue(ofr) == static_cast<int64_t>(0);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 1387 for (OpFoldResult ofr : op.getMixedOffsets()) in foldIdentityOffsetSizeAndStrideOpInterface() local
1388 if (getConstantIntValue(ofr) != static_cast<int64_t>(0)) in foldIdentityOffsetSizeAndStrideOpInterface()
1396 for (OpFoldResult ofr : op.getMixedStrides()) in foldIdentityOffsetSizeAndStrideOpInterface() local
1397 if (getConstantIntValue(ofr) != static_cast<int64_t>(1)) in foldIdentityOffsetSizeAndStrideOpInterface()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2488 if (llvm::any_of(mixedOffsets, [](OpFoldResult ofr) { in isTrivialSubViewOp() argument
2489 Optional<int64_t> intValue = getConstantIntValue(ofr); in isTrivialSubViewOp()
2495 if (llvm::any_of(mixedStrides, [](OpFoldResult ofr) { in isTrivialSubViewOp() argument
2496 Optional<int64_t> intValue = getConstantIntValue(ofr); in isTrivialSubViewOp()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Affine/IR/ |
| H A D | AffineOps.cpp | 719 for (OpFoldResult ofr : values) { in materializeConstants() local
720 if (auto value = ofr.dyn_cast<Value>()) { in materializeConstants()
729 b, b.getIndexAttr(ofr.get<Attribute>().cast<IntegerAttr>().getInt()), in materializeConstants()
|