Home
last modified time | relevance | path

Searched refs:AffineConstantExpr (Results 1 – 25 of 33) sorted by relevance

12

/llvm-project-15.0.7/mlir/lib/IR/
H A DAffineExpr.cpp504 AffineConstantExpr::AffineConstantExpr(AffineExpr::ImplType *ptr) in AffineConstantExpr() function in AffineConstantExpr
506 int64_t AffineConstantExpr::getValue() const { in getValue()
525 auto lhsConst = lhs.dyn_cast<AffineConstantExpr>(); in simplifyAdd()
534 if (lhs.isa<AffineConstantExpr>() || in simplifyAdd()
558 AffineConstantExpr rLhsConstExpr; in simplifyAdd()
570 AffineConstantExpr rRhsConstExpr; in simplifyAdd()
1030 rhs.isa<AffineConstantExpr>()))) { in getSemiAffineExprFromFlatForm()
1033 if (rhs.isa<AffineConstantExpr>()) { in getSemiAffineExprFromFlatForm()
1121 if (!expr.getRHS().isa<AffineConstantExpr>()) { in visitMulExpr()
1172 if (!expr.getRHS().isa<AffineConstantExpr>()) { in visitModExpr()
[all …]
H A DAffineMap.cpp60 return expr.cast<AffineConstantExpr>().getValue(); in constantFoldImpl()
127 if (auto constExpr = expr.dyn_cast<AffineConstantExpr>()) { in isMinorIdentityWithBroadcasting()
176 if (auto constExpr = expr.dyn_cast<AffineConstantExpr>()) { in isPermutationOfMinorIdentityWithBroadcasting()
272 return getNumResults() == 1 && getResult(0).isa<AffineConstantExpr>(); in isSingleConstant()
277 return expr.isa<AffineConstantExpr>(); in isConstant()
283 return getResult(0).cast<AffineConstantExpr>().getValue(); in getSingleConstantResult()
290 result.emplace_back(expr.cast<AffineConstantExpr>().getValue()); in getConstantResults()
474 res.push_back(e.cast<AffineConstantExpr>().getValue()); in compose()
498 auto constExpr = expr.dyn_cast<AffineConstantExpr>(); in isProjectedPermutation()
685 if (auto constExpr = map.getResult(i).dyn_cast<AffineConstantExpr>()) { in inverseAndBroadcastProjectedPermutation()
H A DBuiltinTypes.cpp844 if (auto cst = offsetExpr.dyn_cast<AffineConstantExpr>()) in getStridesAndOffset()
849 if (auto c = e.dyn_cast<AffineConstantExpr>()) in getStridesAndOffset()
962 if (auto cst = m.getResult(0).dyn_cast<AffineConstantExpr>()) in canonicalizeStridedLayout()
H A DAsmPrinter.cpp2368 os << expr.cast<AffineConstantExpr>().getValue(); in printAffineExprInternal()
2397 auto rhsConst = rhsExpr.dyn_cast<AffineConstantExpr>(); in printAffineExprInternal()
2426 if (auto rrhs = rrhsExpr.dyn_cast<AffineConstantExpr>()) { in printAffineExprInternal()
2460 if (auto rhsConst = rhsExpr.dyn_cast<AffineConstantExpr>()) { in printAffineExprInternal()
/llvm-project-15.0.7/mlir/test/python/ir/
H A Daffine_expr.py73 a2 = AffineConstantExpr.get(42)
158 c2 = AffineConstantExpr.get(2)
186 c2 = AffineConstantExpr.get(2)
227 c2 = AffineConstantExpr.get(2)
255 c2 = AffineConstantExpr.get(2)
300 c42 = AffineConstantExpr.get(42)
309 c2 = AffineConstantExpr.get(2)
348 c2 = AffineConstantExpr.get(2)
355 print(AffineConstantExpr.isinstance(d1))
357 print(AffineConstantExpr.isinstance(c2))
H A Daffine_map.py34 c2 = AffineConstantExpr.get(2)
215 replace0 = map1.replace(s0, AffineConstantExpr.get(42), 3, 3)
216 replace1 = map1.replace(s1, AffineConstantExpr.get(42), 3, 3)
217 replace3 = map1.replace(s2, AffineConstantExpr.get(42), 3, 2)
H A Dinteger_set.py34 c42 = AffineConstantExpr.get(42)
104 c42 = AffineConstantExpr.get(42)
H A Dattributes.py151 c2 = AffineConstantExpr.get(2)
/llvm-project-15.0.7/mlir/include/mlir/IR/
H A DAffineExprVisitor.h106 expr.cast<AffineConstantExpr>()); in walkPostOrder()
143 expr.cast<AffineConstantExpr>()); in visit()
180 RetTy visitConstantExpr(AffineConstantExpr expr) { return RetTy(); } in visitConstantExpr()
282 void visitConstantExpr(AffineConstantExpr expr);
H A DAffineExpr.h232 class AffineConstantExpr : public AffineExpr {
235 /* implicit */ AffineConstantExpr(AffineExpr::ImplType *ptr = nullptr);
277 if (std::is_same<U, AffineConstantExpr>::value) in isa()
H A DAffineMap.h556 } else if (auto constExpr = expr.dyn_cast<AffineConstantExpr>()) { in applyPermutationMap()
/llvm-project-15.0.7/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Daffine.py162 return AffineConstantExpr(py_value)
192 class AffineConstantExpr(AffineExprDef): class
200 return _ir.AffineConstantExpr.get(self.value)
/llvm-project-15.0.7/mlir/lib/CAPI/IR/
H A DAffineExpr.cpp101 return unwrap(affineExpr).isa<AffineConstantExpr>(); in mlirAffineExprIsAConstant()
109 return unwrap(affineExpr).cast<AffineConstantExpr>().getValue(); in mlirAffineConstantExprGetValue()
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DVectorInterfaces.td125 return expr.template isa<::mlir::AffineConstantExpr>() &&
126 expr.template dyn_cast<::mlir::AffineConstantExpr>().getValue() == 0;
269 if (dim.isa<AffineConstantExpr>())
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Analysis/
H A DLoopAnalysis.cpp94 if (auto constExpr = resultExpr.dyn_cast<AffineConstantExpr>()) { in getConstantTripCount()
123 if (auto constExpr = resultExpr.dyn_cast<AffineConstantExpr>()) { in getLargestDivisorOfTripCount()
H A DUtils.cpp169 lbMap.getResult(0).isa<AffineConstantExpr>()) in isSliceMaximalFastCheck()
202 if (!srcLbResult.isa<AffineConstantExpr>() || in isSliceMaximalFastCheck()
203 !srcUbResult.isa<AffineConstantExpr>() || in isSliceMaximalFastCheck()
204 !dstLbResult.isa<AffineConstantExpr>() || in isSliceMaximalFastCheck()
205 !dstUbResult.isa<AffineConstantExpr>()) in isSliceMaximalFastCheck()
983 auto cExpr = loopSpanExpr.dyn_cast<AffineConstantExpr>(); in getConstDifference()
/llvm-project-15.0.7/mlir/lib/Dialect/SCF/Transforms/
H A DLoopSpecialization.cpp46 if (auto constantIndex = expr.dyn_cast<AffineConstantExpr>()) in specializeParallelLoopForUnrolling()
82 if (auto constantIndex = expr.dyn_cast<AffineConstantExpr>()) in specializeForLoopForUnrolling()
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Utils/
H A DUtils.cpp75 auto rhsConst = expr.getRHS().dyn_cast<AffineConstantExpr>(); in visitModExpr()
113 auto rhsConst = expr.getRHS().dyn_cast<AffineConstantExpr>(); in visitFloorDivExpr()
155 auto rhsConst = expr.getRHS().dyn_cast<AffineConstantExpr>(); in visitCeilDivExpr()
187 Value visitConstantExpr(AffineConstantExpr expr) { in visitConstantExpr()
470 auto c1 = bin.getLHS().dyn_cast<AffineConstantExpr>(); in substWithMin()
471 auto c2 = bin.getRHS().dyn_cast<AffineConstantExpr>(); in substWithMin()
497 std::get<1>(tuple).template dyn_cast<AffineConstantExpr>(); in normalizeAffineParallel()
1466 if (binaryExpr.getRHS().isa<AffineConstantExpr>()) in getTileSizePos()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp212 LogicalResult visitConstantExpr(AffineConstantExpr expr) { return failure(); } in visitConstantExpr()
240 mulExpr = getAffineExprOfType<AffineConstantExpr>(lhsExpr, rhsExpr); in isDimExprOrMulExpr()
554 bool visitConstantExpr(AffineConstantExpr constExpr) { return false; } in visitConstantExpr()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorTransferPermutationMapRewritePatterns.cpp215 auto dimExpr = expr.dyn_cast<AffineConstantExpr>(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DLoops.cpp286 if (auto val = expr.dyn_cast<AffineConstantExpr>()) { in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Conversion/VectorToGPU/
H A DVectorToGPU.cpp401 if (map.getResult(0).isa<AffineConstantExpr>()) { in convertTransferReadOp()
402 assert(map.getResult(0).cast<AffineConstantExpr>().getValue() == 0); in convertTransferReadOp()
/llvm-project-15.0.7/mlir/python/mlir/_mlir_libs/_mlir/
H A Dir.pyi21 "AffineConstantExpr",
243 class AffineConstantExpr(AffineExpr):
246 def get(*args, **kwargs) -> AffineConstantExpr: ...
/llvm-project-15.0.7/mlir/lib/Conversion/SCFToGPU/
H A DSCFToGPU.cpp324 if (auto constExpr = result.dyn_cast<AffineConstantExpr>()) { in deriveStaticUpperBound()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/
H A DUtils.cpp71 assert(expr.getRHS().cast<AffineConstantExpr>().getValue() > 0 && in visitAffineBinaryOpExpr()
325 if (auto constExpr = result.dyn_cast<AffineConstantExpr>()) in getConstantUpperBoundForIndex()

12