python - NumbaPro JIT类的构造函数

我正在寻找一种将JIT用于python类构造函数的方法，如下所示：

import numpy as np
from numbapro import jit, autojit
from time import time

class Test(object):
    @jit((float, float, float), target="cpu")
    def __init__(self, x, y, z):
        self._x = x
        self._y = y
        self._z = z

    @autojit
    def runTest(self):
        N = 1000000
        self._z = 0
        for i in xrange(N):
            self._z = self._z + np.sin(i)

        return self._z

if __name__ == '__main__':
    a = Test(4,5,6)

    start_time = time()
    z = a.runTest()
    end_time = time() # Get the CPU end time
    print("Math Time: {0} s".format(end_time - start_time))
    print z

但是，似乎我必须为self指定类型，但我不知道该如何实现。
也许有人知道如何解决这个问题？

提前致谢
和我

最佳答案

从版本0.12：here is the Github issue的numba删除了对“ jitting”类的支持。曾经有support for classes in Numba，称为“扩展类型”，但如今所有相关示例均不起作用，给出：TypeError: 'NotImplementedType' object is not callable如果在当前版本（0.17）中执行。

因此，我强烈建议您将runTest功能移至其自己的功能，并使用a tuple或a numpy array将数据传递给该功能，因为支持这些机制。否则，您将不得不使用旧版本的numba。

class Test(object):

    def __init__(self, x, y, z):
        self.x = x
        self.y = y
        self.z = z


@jit(float(float))
def runTest(z):
    N = 1000000
    z = 0
    for i in xrange(N):
        z = z + np.sin(i)
    return z


if __name__ == '__main__':
    a = Test(4,5,6)

    start_time = time()
    z = runTest(a.z)
    end_time = time() # Get the CPU end time
    print("Math Time: {0} s".format(end_time - start_time))
    print z

关于python - NumbaPro JIT类的构造函数，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/29080082/