如何解决如何进行嵌套 jitclass 数组实例化
我正在尝试使用 numba 来加速我的代码,并且需要在另一个 jitclass 中使用一个 jitclass 数组。但是,我似乎找不到将数组传递给 jitclass 的方法。
from numba import jitclass,int32
my_inner_spec = [
('a',int32)
]
@jitclass(my_inner_spec)
class MyInner:
def __init__(self,a):
self.a = a
my_outer_spec = [('inner',MyInner.class_type.instance_type[:])]
@jitclass(my_outer_spec)
class MyOuter:
def __init__(self,inner):
self.inner= inner
MyOuter([MyInner(0)])
这会产生这个错误:
TypingError Traceback (most recent call last)
<ipython-input-106-3f644c4985b0> in <module>()
18 self.inner= inner
19
---> 20 MyOuter([MyInner(0)])
...
...
TypingError: Failed in nopython mode pipeline (step: nopython frontend)
Internal error at <numba.typeinfer.CallConstraint object at 0x7fbdde552550>.
Failed in nopython mode pipeline (step: nopython mode backend)
Cannot cast reflected list(instance.jitclass.MyInner#7fbdde6c8650<a:int32>) to array(instance.jitclass.MyInner#7fbdde6c8650<a:int32>,1d,A): %".19" = load {i8*,i8*},{i8*,i8*}* %"inner"
File "<ipython-input-106-3f644c4985b0>",line 18:
def __init__(self,inner):
self.inner= inner
^
[1] During: lowering "(self).inner = inner" at <ipython-input-106-3f644c4985b0> (18)
[2] During: resolving callee type: jitclass.MyOuter#7fbdde6c8810<inner:array(instance.jitclass.MyInner#7fbdde6c8650<a:int32>,A)>
[3] During: typing of call at <string> (3)
Enable logging at debug level for details.
File "<string>",line 3:
<source missing,REPL/exec in use?>
解决方法
正如 docs 所说,typed.List 是一项实验性功能,从普通 Python 列表推断类型似乎不起作用。
以下内容有点冗长,但确实有效:
my_outer_spec = [('inner',nb.types.ListType(MyInner.class_type.instance_type))]
@nb.experimental.jitclass(my_outer_spec)
class MyOuter:
def __init__(self,inner):
self.inner = inner
lst = nb.typed.List()
lst.append(MyInner(0))
MyOuter(lst)
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。