使用 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運行我的代碼給出了堆棧跟蹤。