execution.py 文件源码

def compileToNative(loopFunction, inputs):
    numbaSizeType = numba.from_dtype(numpy.dtype(sizeType))[:]

    sig = []
    for param in loopFunction.parameters:
        if isinstance(param, Countdown):
            sig.append(numbaSizeType)

        elif isinstance(param, (SizeArray, OutSizeArray)):
            sig.append(numbaSizeType)

        elif isinstance(param, (DataArray, OutDataArray)):
            sig.append(numba.from_dtype(numpy.dtype(param.dataType))[:])

        else:
            assert False, "unexpected type: {0}".format(param)

    sig = tuple(sig)
    return numba.jit([sig], nopython=True)(loopFunction.fcn)