Function __generalCustomOperation

python/cudaq/kernel/kernel_builder.py:179–230

Utility function for adding a generic quantum operation to the MLIR representation for the PyKernel. A controlled version can be invoked by passing additional arguments to the operation. For an N-qubit operation, the last N arguments are treated as `targets` and excess argument

(self, opName, *args)

def __generalCustomOperation(self, opName, *args):
    """
    Utility function for adding a generic quantum operation to the MLIR
    representation for the PyKernel.

    A controlled version can be invoked by passing additional arguments to the
    operation. For an N-qubit operation, the last N arguments are treated as
    `targets` and excess arguments as `controls`.
    """

    global globalRegisteredOperations
    unitary = globalRegisteredOperations[opName]

    numTargets = int(np.log2(np.sqrt(unitary.size)))

    qubits = []
    self.clearCache()
    with self.insertPoint, self.loc:
        for arg in args:
            if isinstance(arg, QuakeValue):
                qubits.append(arg.mlirValue)
            else:
                emitFatalError(f"invalid argument type passed to {opName}.")

        targets = []
        controls = []

        if numTargets == len(qubits):
            targets = qubits
        elif numTargets < len(qubits):
            numControls = len(qubits) - numTargets
            targets = qubits[-numTargets:]
            controls = qubits[:numControls]
        else:
            emitFatalError(
                f"too few arguments passed to {opName}, expected ({numTargets})"
            )

        globalName = f'{nvqppPrefix}{opName}_generator_{numTargets}.rodata'
        currentST = SymbolTable(self.module.operation)
        if not globalName in currentST:
            with InsertionPoint(self.module.body):
                gen_vector_of_complex_constant(self.loc, self.module,
                                               globalName, unitary.tolist())

        quake.CustomUnitaryConstantOp([],
                                      matrix=FlatSymbolRefAttr.get(globalName),
                                      parameters=[],
                                      controls=controls,
                                      targets=targets,
                                      is_adj=False)
        return


class PyKernel(object):