Function mlirTypeFromAnnotation

python/cudaq/kernel/utils.py:380–593

Return the MLIR Type corresponding to the given kernel function argument type annotation. Throws an exception if the programmer did not annotate function argument types.

(annotation,
                           ctx,
                           raiseError=False,
                           cudaqAliases=None)

def mlirTypeFromAnnotation(annotation,
                           ctx,
                           raiseError=False,
                           cudaqAliases=None):
    """
    Return the MLIR Type corresponding to the given kernel function argument
    type annotation.  Throws an exception if the programmer did not annotate
    function argument types.
    """
    _cudaq_names = cudaqAliases if cudaqAliases else {'cudaq'}

    localEmitFatalError = emitFatalError
    if raiseError:
        # Client calling this will handle errors
        def emitFatalErrorOverride(msg):
            raise RuntimeError(msg)

        localEmitFatalError = emitFatalErrorOverride

    if annotation == None:
        localEmitFatalError(
            'cudaq.kernel functions must have argument type annotations.')

    with ctx:

        if hasattr(annotation, 'attr') and hasattr(annotation.value, 'id'):
            if annotation.value.id in _cudaq_names:
                if annotation.attr in ['qview', 'qvector']:
                    return quake.VeqType.get()
                if annotation.attr in ['State']:
                    return cc.PointerType.get(cc.StateType.get())
                if annotation.attr == 'qubit':
                    return quake.RefType.get()
                if annotation.attr == 'pauli_word':
                    return cc.CharspanType.get()
                if annotation.attr == 'measure_handle':
                    return cc.MeasureHandleType.get()

            if annotation.value.id in ['numpy', 'np']:
                if annotation.attr in ['array', 'ndarray']:
                    return cc.StdvecType.get(F64Type.get())
                if annotation.attr == 'complex128':
                    return ComplexType.get(F64Type.get())
                if annotation.attr == 'complex64':
                    return ComplexType.get(F32Type.get())
                if annotation.attr == 'float64':
                    return F64Type.get()
                if annotation.attr == 'float32':
                    return F32Type.get()
                if annotation.attr == 'int64':
                    return IntegerType.get_signless(64)
                if annotation.attr == 'int32':
                    return IntegerType.get_signless(32)
                if annotation.attr == 'int16':
                    return IntegerType.get_signless(16)
                if annotation.attr == 'int8':
                    return IntegerType.get_signless(8)