maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Add a utility function to OperationName for extracting the dialect name. 

PiperOrigin-RevId: 251333376
This commit is contained in:
River Riddle 2019-06-03 16:32:25 -07:00 committed by Mehdi Amini
parent 2cef9f07d2
commit 42c19e8272
3 changed files with 15 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mlir/include/mlir/IR/OperationSupport.h
View File

@ -179,6 +179,9 @@ public:
OperationName(AbstractOperation *op) : representation(op) {}
OperationName(StringRef name, MLIRContext *context);
/// Return the name of the dialect this operation is registered to.
StringRef getDialect() const;
/// Return the name of this operation. This always succeeds.
StringRef getStringRef() const;

5
mlir/lib/Analysis/Verifier.cpp
View File

@ -274,8 +274,7 @@ LogicalResult FuncVerifier::verifyOperation(Operation &op) {
return success();
// Otherwise, verify that the parent dialect allows un-registered operations.
auto opName = op.getName().getStringRef();
auto dialectPrefix = opName.split('.').first;
auto dialectPrefix = op.getName().getDialect();
// Check for an existing answer for the operation dialect.
auto it = dialectAllowsUnknownOps.find(dialectPrefix);
@ -291,7 +290,7 @@ LogicalResult FuncVerifier::verifyOperation(Operation &op) {
}
if (!it->second) {
return failure("unregistered operation '" + opName +
return failure("unregistered operation '" + op.getName().getStringRef() +
"' found in dialect ('" + dialectPrefix +
"') that does not allow unknown operations",
op);

22
mlir/lib/IR/Operation.cpp
View File

@ -38,6 +38,11 @@ OperationName::OperationName(StringRef name, MLIRContext *context) {
representation = Identifier::get(name, context);
}
/// Return the name of the dialect this operation is registered to.
StringRef OperationName::getDialect() const {
return getStringRef().split('.').first;
}
/// Return the name of this operation. This always succeeds.
StringRef OperationName::getStringRef() const {
if (auto *op = representation.dyn_cast<const AbstractOperation *>())
@ -275,10 +280,8 @@ Dialect *Operation::getDialect() {
return &abstractOp->dialect;
// If this operation hasn't been registered or doesn't have abstract
// operation, fall back to a dialect which matches the prefix.
auto opName = getName().getStringRef();
auto dialectPrefix = opName.split('.').first;
return getContext()->getRegisteredDialect(dialectPrefix);
// operation, try looking up the dialect name in the context.
return getContext()->getRegisteredDialect(getName().getDialect());
}
Region *Operation::getContainingRegion() const {
@ -528,16 +531,9 @@ LogicalResult Operation::fold(ArrayRef<Attribute> operands,
return success();
// Otherwise, fall back on the dialect hook to handle it.
Dialect *dialect;
if (abstractOp) {
dialect = &abstractOp->dialect;
} else {
// If this operation hasn't been registered, lookup the parent dialect.
auto opName = getName().getStringRef();
auto dialectPrefix = opName.split('.').first;
if (!(dialect = getContext()->getRegisteredDialect(dialectPrefix)))
Dialect *dialect = getDialect();
if (!dialect)
return failure();
}
SmallVector<Attribute, 8> constants;
if (failed(dialect->constantFoldHook(this, operands, constants)))