jitclass vs extension API:列表中可以使用什么?

Question

所以我有一段时间没有关注numba的发展，我发现了很多令人兴奋的事情，比如@jitclass和列表支持。

所以我想尝试一下，但是如果我尝试在nopython块中列出我的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 的示例进行操作，则拥有Interval列表没有任何问题。

我认为 jitclass 基本上是一种快速的方法，可以完成扩展 API 的冗长示例，我错了吗？如何在不必编写这些容易出错的长绑定函数的情况下实现它？

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>)

Answer 2

这实际上是在Numba 0.39(http://numba.pydata.org/numba-doc/0.39.0/release-notes.html#version-0-39-0)中添加的

List 已获得对包含引用计数类型(如 NumPy 数组和列表)的支持。请注意，列表仍然不能容纳异构类型。

所以现在它直接醒来了。