python - jitclass 与扩展 API : what can be used in a list?

Question

所以我有一段时间没有关注 numba 的开发，但我发现了很多令人兴奋的东西，例如 @jitclass 和列表支持。

所以我想尝试一下，但是如果我尝试在 nopython block 中列出我的 jitclass 结构，我会收到错误:

reflected list(instance.jitclass.Interval#42f9788<lo:float64,hi:float64>): unsupported nested memory-managed object

使用简单的以下类

@jitclass([('lo', types.float64), ('hi', types.float64)])
class Interval(object):
    def __init__(self, lo, hi):
        self.lo = lo
        self.hi = hi

    @property
    def width(self):
        return self.hi - self.lo

以及最简单的对应代码:

@jit(nopython=True)
def f(my_list):
    return my_list

f([Interval(1,2)]*10)

但是，如果我按照使用冗长的扩展 API here 的示例进行操作， ，我没有遇到任何问题拥有间隔列表。

我认为 jitclass 基本上是完成扩展 API 的冗长示例的快速方法，我错了吗？如何才能实现这一目标而无需编写这些很长且容易出错的绑定(bind)函数？

Answer 1

这并不能完全回答您的问题，但如果您想要快速解决方法，我发现与列表不同，元组似乎可以工作:

f((Interval(1,2),)*10)

给出输出

(<numba.jitclass.boxing.Interval at 0x10d8d0d50>,
 <numba.jitclass.boxing.Interval at 0x10d8d0d70>,
 [... clipped ...]
 <numba.jitclass.boxing.Interval at 0x10d8d0e90>)