Numba jitclass 和继承

Question

我有一个类层次结构，我想通过使用 Numba jitclass 来加速我的代码。我已经为一些没有类继承的示例测试了@jitclass，它可以正常工作并加快代码速度。但是，如果我有类继承，则会在编译期间发生错误。下面是演示问题的示例代码。我将非常感谢任何 cmets 和建议。现在对我来说，Numba 似乎不支持类继承，但我在文档中没有找到任何关于它的信息。 代码示例：

import numpy as np
from numba import jitclass         
from numba import int32, float32

spec = [
    ('n', int32),               
    ('val', float32[:]),          
]

@jitclass(spec)
class Parent(object):
    def __init__(self, n):
        self.n = n
        self.val = np.zeros(n, dtype=np.float32)

spec = [
    ('incr', float32),          
]

@jitclass(spec)
class Child(Parent):
    def __init__(self, n):
        Parent.__init__(self, n)
        self.incr = 2.

def func(self):
    for i in xrange(0, self.n):
        self.val[i] += self.incr
    return self.val

par = Parent(10)
chl = Child(10)
print chl.func()

我得到的错误是：

TypeError: cannot subclass from a jitclass

Answer 1

目前（从 0.28.1 开始），Numba 不支持从 jitclass 继承/继承。文档中没有说明，但错误消息非常明确。我猜这个功能会在未来的某个时候添加，但现在它是一个限制。