Function createTestModule

unittests/Optimizer/DecompositionPatternsTest.cpp:81–183

Helper function to create a test module with a single gate operation

// Helper function to create a test module with a single gate operation
ModuleOp createTestModule(MLIRContext *context, StringRef gateSpecStr) {
  auto [gateName, numControls, isAdj] = parseGateSpec(gateSpecStr);

  // Limit the number of controls to 2
  numControls = std::min<size_t>(numControls, 2);

  size_t numQubits;
  if (gateName == "swap" || gateName == "exp_pauli") {
    assert(numControls == 0);
    // exp_pauli can have any number of qubits, we hardcode to 2 for the test.
    numQubits = 2;
  } else {
    numQubits = numControls + 1;
  }

  OpBuilder builder(context);
  auto module = ModuleOp::create(builder, builder.getUnknownLoc());
  builder.setInsertionPointToEnd(module.getBody());

  // Create function type: (qubits...) -> ()
  SmallVector<Type> inputTypes;
  auto refType = cudaq::quake::RefType::get(context);
  for (size_t i = 0; i < numQubits; ++i) {
    inputTypes.push_back(refType);
  }
  auto funcType = builder.getFunctionType(inputTypes, {});

  // Create function
  auto func = func::FuncOp::create(builder, builder.getUnknownLoc(),
                                   "test_func", funcType);
  auto *entry = func.addEntryBlock();
  builder.setInsertionPointToStart(entry);

  // Get operands (controls and target)
  SmallVector<Value> controls;
  for (size_t i = 0; i < numControls; ++i) {
    controls.push_back(entry->getArgument(i));
  }
  Value target = entry->getArgument(numControls);

  // Create the gate operation based on gate name
  Location loc = builder.getUnknownLoc();

  Value pi_2 = cudaq::opt::factory::createFloatConstant(loc, builder, M_PI_2,
                                                        builder.getF64Type());

  if (gateName == "h") {
    cudaq::quake::HOp::create(builder, loc, isAdj, controls, target);
  } else if (gateName == "s") {
    cudaq::quake::SOp::create(builder, loc, isAdj, controls, target);
  } else if (gateName == "t") {
    cudaq::quake::TOp::create(builder, loc, isAdj, controls, target);
  } else if (gateName == "x") {
    cudaq::quake::XOp::create(builder, loc, isAdj, controls, target);
  } else if (gateName == "y") {
    cudaq::quake::YOp::create(builder, loc, isAdj, controls, target);
  } else if (gateName == "z") {
    cudaq::quake::ZOp::create(builder, loc, isAdj, controls, target);