Function getKernelFuncOp

python/utils/OpaqueArguments.h:48–69

/// @brief Search the given Module for the function with provided name.
template <bool noThrow = false>
mlir::func::FuncOp getKernelFuncOp(mlir::ModuleOp mod,
                                   const std::string &name) {
  std::string lookupName = name;
  if (mod->hasAttr(cudaq::runtime::pythonUniqueAttrName)) {
    mlir::StringRef uniqName = cast<mlir::StringAttr>(
        mod->getAttr(cudaq::runtime::pythonUniqueAttrName));
    if (uniqName.substr(0, name.size()) == name) {
      // FIXME: The path to this call used the wrong name.
      lookupName = uniqName;
    }
  }
  lookupName = cudaq::runtime::cudaqGenPrefixName + lookupName;
  if (auto result = mod.lookupSymbol<mlir::func::FuncOp>(lookupName))
    return result;
  if constexpr (noThrow) {
    return {};
  } else /*constexpr*/ {
    mod.dump();
    throw std::runtime_error("Could not find " + lookupName +
                             " function in current module.");
  }
}

template <bool noThrow = false>
mlir::func::FuncOp getKernelFuncOp(MlirModule module,