Home
last modified time | relevance | path

Searched refs:AffineExprKind (Results 1 – 12 of 12) sorted by relevance

/external/llvm-project/mlir/lib/IR/
DAffineExpr.cpp23 AffineExprKind AffineExpr::getKind() const { return expr->kind; } in getKind()
43 AffineExpr mlir::getAffineBinaryOpExpr(AffineExprKind kind, AffineExpr lhs, in getAffineBinaryOpExpr()
45 if (kind == AffineExprKind::Add) in getAffineBinaryOpExpr()
47 if (kind == AffineExprKind::Mul) in getAffineBinaryOpExpr()
49 if (kind == AffineExprKind::FloorDiv) in getAffineBinaryOpExpr()
51 if (kind == AffineExprKind::CeilDiv) in getAffineBinaryOpExpr()
53 if (kind == AffineExprKind::Mod) in getAffineBinaryOpExpr()
65 case AffineExprKind::Constant: in replaceDimsAndSymbols()
67 case AffineExprKind::DimId: { in replaceDimsAndSymbols()
73 case AffineExprKind::SymbolId: { in replaceDimsAndSymbols()
[all …]
DAffineExprDetail.h30 AffineExprKind kind;
38 return static_cast<AffineExprKind>(std::get<0>(key)) == kind &&
45 result->kind = static_cast<AffineExprKind>(std::get<0>(key)); in construct()
61 return kind == static_cast<AffineExprKind>(key.first) &&
68 result->kind = static_cast<AffineExprKind>(key.first); in construct()
86 result->kind = AffineExprKind::Constant; in construct()
DAffineMap.cpp43 case AffineExprKind::Add: in constantFoldImpl()
46 case AffineExprKind::Mul: in constantFoldImpl()
49 case AffineExprKind::Mod: in constantFoldImpl()
52 case AffineExprKind::FloorDiv: in constantFoldImpl()
55 case AffineExprKind::CeilDiv: in constantFoldImpl()
58 case AffineExprKind::Constant: in constantFoldImpl()
60 case AffineExprKind::DimId: in constantFoldImpl()
65 case AffineExprKind::SymbolId: in constantFoldImpl()
DBuiltinTypes.cpp527 if (bin.getKind() == AffineExprKind::CeilDiv || in extractStrides()
528 bin.getKind() == AffineExprKind::FloorDiv || in extractStrides()
529 bin.getKind() == AffineExprKind::Mod) in extractStrides()
532 if (bin.getKind() == AffineExprKind::Mul) { in extractStrides()
550 if (bin.getKind() == AffineExprKind::Add) { in extractStrides()
DAsmPrinter.cpp1988 case AffineExprKind::SymbolId: { in printAffineExprInternal()
1996 case AffineExprKind::DimId: { in printAffineExprInternal()
2004 case AffineExprKind::Constant: in printAffineExprInternal()
2007 case AffineExprKind::Add: in printAffineExprInternal()
2010 case AffineExprKind::Mul: in printAffineExprInternal()
2013 case AffineExprKind::FloorDiv: in printAffineExprInternal()
2016 case AffineExprKind::CeilDiv: in printAffineExprInternal()
2019 case AffineExprKind::Mod: in printAffineExprInternal()
2029 if (binOp.getKind() != AffineExprKind::Add) { in printAffineExprInternal()
2035 if (rhsConst && binOp.getKind() == AffineExprKind::Mul && in printAffineExprInternal()
[all …]
/external/llvm-project/mlir/include/mlir/IR/
DAffineExprVisitor.h77 case AffineExprKind::Add: { in walkPostOrder()
82 case AffineExprKind::Mul: { in walkPostOrder()
87 case AffineExprKind::Mod: { in walkPostOrder()
92 case AffineExprKind::FloorDiv: { in walkPostOrder()
97 case AffineExprKind::CeilDiv: { in walkPostOrder()
102 case AffineExprKind::Constant: in walkPostOrder()
105 case AffineExprKind::DimId: in walkPostOrder()
108 case AffineExprKind::SymbolId: in walkPostOrder()
119 case AffineExprKind::Add: { in visit()
123 case AffineExprKind::Mul: { in visit()
[all …]
DAffineExpr.h38 enum class AffineExprKind { enum
90 AffineExprKind getKind() const;
233 AffineExpr getAffineBinaryOpExpr(AffineExprKind kind, AffineExpr lhs,
250 return getKind() <= AffineExprKind::LAST_AFFINE_BINARY_OP; in isa()
252 return getKind() == AffineExprKind::DimId; in isa()
254 return getKind() == AffineExprKind::SymbolId; in isa()
256 return getKind() == AffineExprKind::Constant; in isa()
/external/llvm-project/mlir/lib/CAPI/IR/
DAffineExpr.cpp96 return unwrap(affineExpr).getKind() == mlir::AffineExprKind::Add; in mlirAffineExprIsAAdd()
100 return wrap(getAffineBinaryOpExpr(mlir::AffineExprKind::Add, unwrap(lhs), in mlirAffineAddExprGet()
109 return unwrap(affineExpr).getKind() == mlir::AffineExprKind::Mul; in mlirAffineExprIsAMul()
113 return wrap(getAffineBinaryOpExpr(mlir::AffineExprKind::Mul, unwrap(lhs), in mlirAffineMulExprGet()
122 return unwrap(affineExpr).getKind() == mlir::AffineExprKind::Mod; in mlirAffineExprIsAMod()
126 return wrap(getAffineBinaryOpExpr(mlir::AffineExprKind::Mod, unwrap(lhs), in mlirAffineModExprGet()
135 return unwrap(affineExpr).getKind() == mlir::AffineExprKind::FloorDiv; in mlirAffineExprIsAFloorDiv()
140 return wrap(getAffineBinaryOpExpr(mlir::AffineExprKind::FloorDiv, unwrap(lhs), in mlirAffineFloorDivExprGet()
149 return unwrap(affineExpr).getKind() == mlir::AffineExprKind::CeilDiv; in mlirAffineExprIsACeilDiv()
154 return wrap(getAffineBinaryOpExpr(mlir::AffineExprKind::CeilDiv, unwrap(lhs), in mlirAffineCeilDivExprGet()
/external/llvm-project/mlir/lib/Dialect/SDBM/
DSDBMExpr.cpp45 return AffineExprMatcher(AffineExprKind::Add, *this, other); in operator +()
48 return AffineExprMatcher(AffineExprKind::Mul, *this, other); in operator *()
51 return AffineExprMatcher(AffineExprKind::FloorDiv, *this, other); in floorDiv()
54 return AffineExprMatcher(AffineExprKind::CeilDiv, *this, other); in ceilDiv()
57 return AffineExprMatcher(AffineExprKind::Mod, *this, other); in operator %()
65 AffineExprMatcher(AffineExprKind k, AffineExprMatcher a, AffineExprMatcher b);
66 AffineExprKind kind; // only used to match in binary op cases.
89 : kind(AffineExprKind::Constant), storage(new AffineExprMatcherStorage()) {} in AffineExprMatcher()
101 if (kind > AffineExprKind::LAST_AFFINE_BINARY_OP) { in match()
131 AffineExprMatcher::AffineExprMatcher(AffineExprKind k, AffineExprMatcher a, in AffineExprMatcher()
[all …]
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DBufferize.cpp92 case AffineExprKind::DimId: { in allocateBuffersForResults()
DTransforms.cpp360 if (bin.getKind() == mlir::AffineExprKind::Add) in substWithMin()
DTiling.cpp103 if (expr.getKind() == mlir::AffineExprKind::Mul) in visitAffineBinaryOpExpr()