template <typename OpTy, typename... Args>
OpTy create(Location location, Args &&... args) {
OperationState state(location, OpTy::getOperationName());
+ if (!state.name.getAbstractOperation())
+ llvm::report_fatal_error("Building op `" +
+ state.name.getStringRef().str() +
+ "` but it isn't registered in this MLIRContext");
OpTy::build(*this, state, std::forward<Args>(args)...);
auto *op = createOperation(state);
auto result = dyn_cast<OpTy>(op);
// Create the operation without using 'createOperation' as we don't want to
// insert it yet.
OperationState state(location, OpTy::getOperationName());
+ if (!state.name.getAbstractOperation())
+ llvm::report_fatal_error("Building op `" +
+ state.name.getStringRef().str() +
+ "` but it isn't registered in this MLIRContext");
OpTy::build(*this, state, std::forward<Args>(args)...);
Operation *op = Operation::create(state);
/// directly.
std::vector<AbstractOperation *> getRegisteredOperations();
+ /// Return true if this operation name is registered in this context.
+ bool isOperationRegistered(StringRef name);
+
// This is effectively private given that only MLIRContext.cpp can see the
// MLIRContextImpl type.
MLIRContextImpl &getImpl() { return *impl; }
static bool classof(Operation *op) {
if (auto *abstractOp = op->getAbstractOperation())
return TypeID::get<ConcreteType>() == abstractOp->typeID;
- return op->getName().getStringRef() == ConcreteType::getOperationName();
+ assert(op->getContext()->isOperationRegistered(
+ ConcreteType::getOperationName()) &&
+ "Casting attempt to an unregistered operation");
+ return false;
}
/// This is the hook used by the AsmParser to parse the custom form of this
return result;
}
+bool MLIRContext::isOperationRegistered(StringRef name) {
+ // Lock access to the context registry.
+ ScopedReaderLock registryLock(impl->contextMutex, impl->threadingIsEnabled);
+
+ return impl->registeredOperations.count(name);
+}
+
void Dialect::addOperation(AbstractOperation opInfo) {
assert((getNamespace().empty() ||
opInfo.name.split('.').first == getNamespace()) &&
static Dialect &lookupDialectForSymbol(MLIRContext *ctx, TypeID typeID) {
auto &impl = ctx->getImpl();
auto it = impl.registeredDialectSymbols.find(typeID);
- assert(it != impl.registeredDialectSymbols.end() &&
- "symbol is not registered.");
+ if (it == impl.registeredDialectSymbols.end())
+ llvm::report_fatal_error(
+ "Trying to create a type that was not registered in this MLIRContext.");
return *it->second;
}