Function jitCode

runtime/cudaq/builder/kernel_builder.cpp:947–1125

}

std::tuple<bool, ExecutionEngine *>
jitCode(ImplicitLocOpBuilder &builder, ExecutionEngine *jit,
        std::unordered_map<ExecutionEngine *, std::size_t> &jitHash,
        std::string kernelName, std::vector<std::string> extraLibPaths,
        StateVectorStorage &stateVectorStorage) {

  // Start of by getting the current ModuleOp
  auto *block = builder.getBlock();
  auto *context = builder.getContext();
  auto *function = block->getParentOp();
  auto currentModule = function->getParentOfType<ModuleOp>();

  // Create a unique hash from that ModuleOp
  auto hash = llvm::hash_code{0};
  currentModule.walk([&hash](Operation *op) {
    hash = llvm::hash_combine(hash, OperationEquivalence::computeHash(op));
  });
  auto moduleHash = static_cast<size_t>(hash);

  if (jit) {
    // Have we added more instructions since the last time we jit the code? If
    // so, we need to delete this JIT engine and create a new one.
    if (moduleHash == jitHash[jit])
      return std::make_tuple(false, jit);
    else {
      // need to redo the jit, remove the old one
      jitHash.erase(jit);
    }
  }

  CUDAQ_INFO("kernel_builder running jitCode.");

  auto module = currentModule.clone();
  auto ctx = module.getContext();
  SmallVector<mlir::NamedAttribute> names;
  names.emplace_back(mlir::StringAttr::get(ctx, kernelName),
                     mlir::StringAttr::get(ctx, "BuilderKernel.EntryPoint"));
  auto mapAttr = mlir::DictionaryAttr::get(ctx, names);
  module->setAttr(cudaq::runtime::mangledNameMap, mapAttr);

  // Tag as an entrypoint if it is one
  tagEntryPoint(builder, module, StringRef{});

  {
    PassManager pm(context);
    pm.addInstrumentation(std::make_unique<cudaq::TracePassInstrumentation>());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createUnwindLowering());
    cudaq::opt::addAggressiveInlining(pm);
    pm.addPass(createCanonicalizerPass());
    pm.addPass(cudaq::opt::createApplySpecialization());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createClassicalMemToReg());
    pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
    pm.addPass(cudaq::opt::createExpandMeasurementsPass());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createLoopNormalize());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createLoopUnroll());
    pm.addNestedPass<func::FuncOp>(createCanonicalizerPass());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createQuakeAddDeallocs());
    pm.addNestedPass<func::FuncOp>(cudaq::opt::createQuakeAddMetadata());