79 lines · cpp
1//===- IntegerSet.cpp - MLIR Integer Set class ----------------------------===//2//3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.4// See https://llvm.org/LICENSE.txt for license information.5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception6//7//===----------------------------------------------------------------------===//8 9#include "mlir/IR/IntegerSet.h"10#include "IntegerSetDetail.h"11 12using namespace mlir;13using namespace mlir::detail;14 15unsigned IntegerSet::getNumDims() const { return set->dimCount; }16unsigned IntegerSet::getNumSymbols() const { return set->symbolCount; }17unsigned IntegerSet::getNumInputs() const {18 return set->dimCount + set->symbolCount;19}20 21unsigned IntegerSet::getNumConstraints() const {22 return set->constraints.size();23}24 25unsigned IntegerSet::getNumEqualities() const {26 unsigned numEqualities = 0;27 for (unsigned i = 0, e = getNumConstraints(); i < e; i++)28 if (isEq(i))29 ++numEqualities;30 return numEqualities;31}32 33unsigned IntegerSet::getNumInequalities() const {34 return getNumConstraints() - getNumEqualities();35}36 37bool IntegerSet::isEmptyIntegerSet() const {38 return *this == getEmptySet(set->dimCount, set->symbolCount, getContext());39}40 41ArrayRef<AffineExpr> IntegerSet::getConstraints() const {42 return set->constraints;43}44 45AffineExpr IntegerSet::getConstraint(unsigned idx) const {46 return getConstraints()[idx];47}48 49/// Returns the equality bits, which specify whether each of the constraints50/// is an equality or inequality.51ArrayRef<bool> IntegerSet::getEqFlags() const { return set->eqFlags; }52 53/// Returns true if the idx^th constraint is an equality, false if it is an54/// inequality.55bool IntegerSet::isEq(unsigned idx) const { return getEqFlags()[idx]; }56 57MLIRContext *IntegerSet::getContext() const {58 return getConstraint(0).getContext();59}60 61/// Walk all of the AffineExpr's in this set. Each node in an expression62/// tree is visited in postorder.63void IntegerSet::walkExprs(function_ref<void(AffineExpr)> callback) const {64 for (auto expr : getConstraints())65 expr.walk(callback);66}67 68IntegerSet IntegerSet::replaceDimsAndSymbols(69 ArrayRef<AffineExpr> dimReplacements, ArrayRef<AffineExpr> symReplacements,70 unsigned numResultDims, unsigned numResultSyms) {71 SmallVector<AffineExpr, 8> constraints;72 constraints.reserve(getNumConstraints());73 for (auto cst : getConstraints())74 constraints.push_back(75 cst.replaceDimsAndSymbols(dimReplacements, symReplacements));76 77 return get(numResultDims, numResultSyms, constraints, getEqFlags());78}79