Numba 错误地解释了生成器函数

Question

尽管Numba 据称支持生成器函数。具体来说，我正在尝试实现 Heap 的算法来生成数组的排列，因为 Numba 不能与 itertools 包一起使用（基本上只是从 Wikipedia)：

import numpy as np
from numba import njit


@njit()
def permutations(arr):

    c = np.zeros(arr.shape[0], dtype=np.int64)
    yield arr
    i = 0
    while i < arr.shape[0]:
        if c[i] < i:
            if i % 2 == 0:
                temp = arr[1]
                arr[1] = arr[i]
                arr[i] = temp
            else:
                temp = arr[c[i]]
                arr[c[i]] = arr[i]
                arr[i] = temp
            yield arr
            c[i] += 1
            i = 0
        else:
            c[i] = 0
            i += 1


def main():

    arr = np.array(['A', 'B', 'C', 'D'])
    for permutation in permutations(arr):
        print(permutation)


if __name__ == '__main__':
    main()

如果没有 njit()，输出是正确的：

['A' 'B' 'C' 'D']
['B' 'A' 'C' 'D']
['B' 'C' 'A' 'D']
['C' 'B' 'A' 'D']
['C' 'A' 'B' 'D']
['A' 'C' 'B' 'D']
['D' 'C' 'B' 'A']
['C' 'D' 'B' 'A']
['C' 'B' 'D' 'A']
['B' 'C' 'D' 'A']
['B' 'D' 'C' 'A']
['D' 'B' 'C' 'A']
['D' 'A' 'C' 'B']
['A' 'D' 'C' 'B']
['A' 'C' 'D' 'B']
['C' 'A' 'D' 'B']
['C' 'D' 'A' 'B']
['D' 'C' 'A' 'B']
['D' 'C' 'B' 'A']
['C' 'D' 'B' 'A']
['C' 'B' 'D' 'A']
['B' 'C' 'D' 'A']
['B' 'D' 'C' 'A']
['D' 'B' 'C' 'A']

但是，使用njit()，我明白了：

['A' 'B' 'C' 'D']
['B' 'A' 'C' 'D']

我需要做一些特殊的事情来 JIT 生成器函数吗？我是否错误地使用了生成器的输出来理解 Numba 的解释方式？这是生成器函数的 Numba 实现中的错误吗？