MLIR
20.0.0git
|
#include "mlir-c/IntegerSet.h"
#include "mlir-c/AffineExpr.h"
#include "mlir/CAPI/AffineExpr.h"
#include "mlir/CAPI/IR.h"
#include "mlir/CAPI/IntegerSet.h"
#include "mlir/CAPI/Utils.h"
#include "mlir/IR/IntegerSet.h"
Go to the source code of this file.
Functions | |
MlirContext | mlirIntegerSetGetContext (MlirIntegerSet set) |
Gets the context in which the given integer set lives. More... | |
bool | mlirIntegerSetEqual (MlirIntegerSet s1, MlirIntegerSet s2) |
Checks if two integer set objects are equal. More... | |
void | mlirIntegerSetPrint (MlirIntegerSet set, MlirStringCallback callback, void *userData) |
Prints an integer set by sending chunks of the string representation and forwarding userData to callback`. More... | |
void | mlirIntegerSetDump (MlirIntegerSet set) |
Prints an integer set to the standard error stream. More... | |
MlirIntegerSet | mlirIntegerSetEmptyGet (MlirContext context, intptr_t numDims, intptr_t numSymbols) |
Gets or creates a new canonically empty integer set with the give number of dimensions and symbols in the given context. More... | |
MlirIntegerSet | mlirIntegerSetGet (MlirContext context, intptr_t numDims, intptr_t numSymbols, intptr_t numConstraints, const MlirAffineExpr *constraints, const bool *eqFlags) |
Gets or creates a new integer set in the given context. More... | |
MlirIntegerSet | mlirIntegerSetReplaceGet (MlirIntegerSet set, const MlirAffineExpr *dimReplacements, const MlirAffineExpr *symbolReplacements, intptr_t numResultDims, intptr_t numResultSymbols) |
Gets or creates a new integer set in which the values and dimensions of the given set are replaced with the given affine expressions. More... | |
bool | mlirIntegerSetIsCanonicalEmpty (MlirIntegerSet set) |
Checks whether the given set is a canonical empty set, e.g., the set returned by mlirIntegerSetEmptyGet. More... | |
intptr_t | mlirIntegerSetGetNumDims (MlirIntegerSet set) |
Returns the number of dimensions in the given set. More... | |
intptr_t | mlirIntegerSetGetNumSymbols (MlirIntegerSet set) |
Returns the number of symbols in the given set. More... | |
intptr_t | mlirIntegerSetGetNumInputs (MlirIntegerSet set) |
Returns the number of inputs (dimensions + symbols) in the given set. More... | |
intptr_t | mlirIntegerSetGetNumConstraints (MlirIntegerSet set) |
Returns the number of constraints (equalities + inequalities) in the given set. More... | |
intptr_t | mlirIntegerSetGetNumEqualities (MlirIntegerSet set) |
Returns the number of equalities in the given set. More... | |
intptr_t | mlirIntegerSetGetNumInequalities (MlirIntegerSet set) |
Returns the number of inequalities in the given set. More... | |
MlirAffineExpr | mlirIntegerSetGetConstraint (MlirIntegerSet set, intptr_t pos) |
Returns pos -th constraint of the set. More... | |
bool | mlirIntegerSetIsConstraintEq (MlirIntegerSet set, intptr_t pos) |
Returns true of the pos -th constraint of the set is an equality constraint, false otherwise. More... | |
void mlirIntegerSetDump | ( | MlirIntegerSet | set | ) |
Prints an integer set to the standard error stream.
Definition at line 33 of file IntegerSet.cpp.
References unwrap().
MlirIntegerSet mlirIntegerSetEmptyGet | ( | MlirContext | context, |
intptr_t | numDims, | ||
intptr_t | numSymbols | ||
) |
Gets or creates a new canonically empty integer set with the give number of dimensions and symbols in the given context.
Definition at line 35 of file IntegerSet.cpp.
References mlir::IntegerSet::getEmptySet(), unwrap(), and wrap().
bool mlirIntegerSetEqual | ( | MlirIntegerSet | s1, |
MlirIntegerSet | s2 | ||
) |
Checks if two integer set objects are equal.
This is a "shallow" comparison of two objects. Only the sets with some small number of constraints are uniqued and compare equal here. Set objects that represent the same integer set with different constraints may be considered non-equal by this check. Set difference followed by an (expensive) emptiness check should be used to check equivalence of the underlying integer sets.
Definition at line 23 of file IntegerSet.cpp.
References unwrap().
Referenced by mlir::python::PyIntegerSet::operator==().
MlirIntegerSet mlirIntegerSetGet | ( | MlirContext | context, |
intptr_t | numDims, | ||
intptr_t | numSymbols, | ||
intptr_t | numConstraints, | ||
const MlirAffineExpr * | constraints, | ||
const bool * | eqFlags | ||
) |
Gets or creates a new integer set in the given context.
The set is defined by a list of affine constraints, with the given number of input dimensions and symbols, which are treated as either equalities (eqFlags is 1) or inequalities (eqFlags is 0). Both constraints
and eqFlags
are expected to point to at least numConstraint
consecutive values.
Definition at line 42 of file IntegerSet.cpp.
References mlir::IntegerSet::get(), unwrapList(), and wrap().
MlirAffineExpr mlirIntegerSetGetConstraint | ( | MlirIntegerSet | set, |
intptr_t | pos | ||
) |
Returns pos
-th constraint of the set.
Definition at line 97 of file IntegerSet.cpp.
MlirContext mlirIntegerSetGetContext | ( | MlirIntegerSet | set | ) |
Gets the context in which the given integer set lives.
Definition at line 19 of file IntegerSet.cpp.
References getContext(), unwrap(), and wrap().
Referenced by mlir::python::PyIntegerSet::createFromCapsule().
intptr_t mlirIntegerSetGetNumConstraints | ( | MlirIntegerSet | set | ) |
Returns the number of constraints (equalities + inequalities) in the given set.
Definition at line 85 of file IntegerSet.cpp.
References unwrap().
intptr_t mlirIntegerSetGetNumDims | ( | MlirIntegerSet | set | ) |
Returns the number of dimensions in the given set.
Definition at line 73 of file IntegerSet.cpp.
References unwrap().
intptr_t mlirIntegerSetGetNumEqualities | ( | MlirIntegerSet | set | ) |
Returns the number of equalities in the given set.
Definition at line 89 of file IntegerSet.cpp.
References unwrap().
intptr_t mlirIntegerSetGetNumInequalities | ( | MlirIntegerSet | set | ) |
Returns the number of inequalities in the given set.
Definition at line 93 of file IntegerSet.cpp.
References unwrap().
intptr_t mlirIntegerSetGetNumInputs | ( | MlirIntegerSet | set | ) |
Returns the number of inputs (dimensions + symbols) in the given set.
Definition at line 81 of file IntegerSet.cpp.
References unwrap().
intptr_t mlirIntegerSetGetNumSymbols | ( | MlirIntegerSet | set | ) |
Returns the number of symbols in the given set.
Definition at line 77 of file IntegerSet.cpp.
References unwrap().
bool mlirIntegerSetIsCanonicalEmpty | ( | MlirIntegerSet | set | ) |
Checks whether the given set is a canonical empty set, e.g., the set returned by mlirIntegerSetEmptyGet.
Definition at line 69 of file IntegerSet.cpp.
References unwrap().
bool mlirIntegerSetIsConstraintEq | ( | MlirIntegerSet | set, |
intptr_t | pos | ||
) |
Returns true
of the pos
-th constraint of the set is an equality constraint, false
otherwise.
Definition at line 101 of file IntegerSet.cpp.
References unwrap().
void mlirIntegerSetPrint | ( | MlirIntegerSet | set, |
MlirStringCallback | callback, | ||
void * | userData | ||
) |
Prints an integer set by sending chunks of the string representation and forwarding userData to
callback`.
Note that the callback may be called several times with consecutive chunks of the string.
Definition at line 27 of file IntegerSet.cpp.
References mlir::Diagnostic::print(), and unwrap().
MlirIntegerSet mlirIntegerSetReplaceGet | ( | MlirIntegerSet | set, |
const MlirAffineExpr * | dimReplacements, | ||
const MlirAffineExpr * | symbolReplacements, | ||
intptr_t | numResultDims, | ||
intptr_t | numResultSymbols | ||
) |
Gets or creates a new integer set in which the values and dimensions of the given set are replaced with the given affine expressions.
dimReplacements
and symbolReplacements
are expected to point to at least as many consecutive expressions as the given set has dimensions and symbols, respectively. The new set will have numResultDims
and numResultSymbols
dimensions and symbols, respectively.
Definition at line 56 of file IntegerSet.cpp.
References unwrap(), unwrapList(), and wrap().