使用 numba 重载时 Python 内核死掉

Question

我试图通过使用 numba 的重载（此处的文档）在 numba jitted 函数中使用外部定义的函数，但是当我运行它时 Python 内核死了。 这是我重现的最小示例：

from numpy.polynomial.legendre import leggauss as lg  # My externally defined function
import numba as nb
from numba.extending import overload

@overload(lg)
def implimentlg(n):
   if not isinstance(n, (int, nb.types.Integer, nb.int32)):
      raise nb.errors.TypingError("must be int")

   def impli(n):
      return lg(n)
   return impli

@nb.jit(nopython=True)
def tmp():
   return lg(15)

tmp()

我猜我的问题与我的implimentlg函数中的输入有关，但是没有堆栈跟踪我就不知所措了。

Answer 1

问题是我使用neovim包“nvim-ipy”在iPython终端中运行我的python代码。 直接使用python3 myFile.py运行我的代码给出了堆栈跟踪。