我想加快一个类似树形数据结构的简单实现的python代码。首先想到的是将其与numba同步。我遇到的问题是如何告诉numba Node
的类型,因为它的成员之一又是Node
类型。这是一个最小的示例:
spec=[("value", nb.float64), ("parent", Node.class_type.instance_type)]
@nb.jitclass(spec)
class Node:
def __init__(self, value:float, parent:"Node") -> None:
self.value:float = value
self.parent:Node = parent
该示例显然不起作用,因为在定义spec
时,尚不知道如何处理类Node
。
在numba中甚至可能吗?