| /llvm-project-15.0.7/mlir/lib/Bindings/Python/ |
| H A D | IRModule.cpp | 37 void PyGlobals::loadDialectModule(llvm::StringRef dialectNamespace) { in loadDialectModule() argument
38 if (loadedDialectModulesCache.contains(dialectNamespace)) in loadDialectModule()
45 moduleName.append(dialectNamespace.data(), dialectNamespace.size()); in loadDialectModule()
60 loadedDialectModulesCache.insert(dialectNamespace); in loadDialectModule()
65 py::object &found = dialectClassMap[dialectNamespace]; in registerDialectImpl()
68 dialectNamespace + in registerDialectImpl()
89 loadDialectModule(dialectNamespace); in lookupDialectClass()
91 const auto foundIt = dialectClassMap.find(dialectNamespace); in lookupDialectClass()
100 dialectClassMap[dialectNamespace] = py::none(); in lookupDialectClass()
118 llvm::StringRef dialectNamespace = split.first; in lookupRawOpViewClass() local
[all …]
|
| H A D | Globals.h | 55 void loadDialectModule(llvm::StringRef dialectNamespace);
64 void registerDialectImpl(const std::string &dialectNamespace,
77 lookupDialectClass(const std::string &dialectNamespace);
|
| H A D | MainModule.cpp | 57 std::string dialectNamespace = in PYBIND11_MODULE() local
59 PyGlobals::get().registerDialectImpl(dialectNamespace, pyClass); in PYBIND11_MODULE()
|
| H A D | IRTypes.cpp | 634 [](std::string dialectNamespace, std::string typeData, in bindDerived()
637 toMlirStringRef(dialectNamespace), in bindDerived()
|
| H A D | IRAttributes.cpp | 332 [](std::string dialectNamespace, py::buffer buffer, PyType &type, in bindDerived()
337 context->get(), toMlirStringRef(dialectNamespace), bufferSize, in bindDerived()
|
| H A D | IRCore.cpp | 160 createCustomDialectWrapper(const std::string &dialectNamespace, in createCustomDialectWrapper() argument
162 auto dialectClass = PyGlobals::get().lookupDialectClass(dialectNamespace); in createCustomDialectWrapper()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | MLIRContext.cpp | 414 MLIRContext::getOrLoadDialect(StringRef dialectNamespace, TypeID dialectID, in getOrLoadDialect() argument
418 auto dialectIt = impl.loadedDialects.find(dialectNamespace); in getOrLoadDialect()
422 << "Load new dialect in Context " << dialectNamespace << "\n"); in getOrLoadDialect()
426 "Loading a dialect (" + dialectNamespace + in getOrLoadDialect()
432 impl.loadedDialects.insert({dialectNamespace, ctor()}).first->second; in getOrLoadDialect()
438 auto stringAttrsIt = impl.dialectReferencingStrAttrs.find(dialectNamespace); in getOrLoadDialect()
453 llvm::report_fatal_error("a dialect with namespace '" + dialectNamespace + in getOrLoadDialect()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Bufferization/IR/ |
| H A D | BufferizableOpInterface.h | 73 void allowDialect(StringRef dialectNamespace) { in allowDialect() argument
75 return op->getDialect()->getNamespace() == dialectNamespace; in allowDialect()
|
| /llvm-project-15.0.7/mlir/include/mlir/IR/ |
| H A D | MLIRContext.h | 220 Dialect *getOrLoadDialect(StringRef dialectNamespace, TypeID dialectID,
|
| H A D | BuiltinTypes.td | 618 "StringAttr":$dialectNamespace,
624 "StringAttr":$dialectNamespace, CArg<"StringRef", "{}">:$typeData
626 return $_get(dialectNamespace.getContext(), dialectNamespace, typeData);
|
| H A D | BuiltinAttributes.td | 725 let parameters = (ins "StringAttr":$dialectNamespace,
|
| /llvm-project-15.0.7/mlir/lib/CAPI/IR/ |
| H A D | BuiltinTypes.cpp | 374 MlirType mlirOpaqueTypeGet(MlirContext ctx, MlirStringRef dialectNamespace, in mlirOpaqueTypeGet() argument
377 OpaqueType::get(StringAttr::get(unwrap(ctx), unwrap(dialectNamespace)), in mlirOpaqueTypeGet()
|
| H A D | BuiltinAttributes.cpp | 172 MlirAttribute mlirOpaqueAttrGet(MlirContext ctx, MlirStringRef dialectNamespace, in mlirOpaqueAttrGet() argument
176 OpaqueAttr::get(StringAttr::get(unwrap(ctx), unwrap(dialectNamespace)), in mlirOpaqueAttrGet()
|
| /llvm-project-15.0.7/mlir/include/mlir-c/ |
| H A D | BuiltinTypes.h | 348 MlirStringRef dialectNamespace,
|
| H A D | BuiltinAttributes.h | 170 mlirOpaqueAttrGet(MlirContext ctx, MlirStringRef dialectNamespace,
|