cython中函数内部的数组可以有多大的限制？

Question

这段代码编译和运行得很好：

cdef enum:
    SIZE = 1000000
cdef int ar[SIZE]
cdef int i

for i in range(SIZE):
    ar[i] = i
print(ar[5])

但是这段代码：

cdef enum:
    SIZE = 1000000
    
def test():
    cdef int ar[SIZE]
    cdef int i

    for i in range(SIZE):
        ar[i] = i
    print(ar[5])
    
test()

使python内核崩溃（我正在使用jupyter Magic运行它）。

函数内部的数组大小是否有限制？如果有，是否有办法取消该限制？

Answer 1

在第一种情况下，数组是一个全局变量，它是静态定义的普通C数组，在堆上分配。在第二种情况下，在堆栈上分配局部变量数组。问题是堆栈有一个固定的最大大小（通常类似于2个MiB，但在不同的平台上可能会有很大的变化）。如果您试图写入超出堆栈限制的分配内容，如果幸运的话，您会遇到崩溃或很好的堆栈溢出错误。请注意，函数调用和局部变量临时占用堆栈中的一些空间。虽然堆栈的大小可以调整，但最好的解决方案是对相对较大的数组或大小可变的数组使用动态分配。希望在Cython中可以使用malloc和free（请参阅文档），但要小心不要忘记free（也不要两次释放数组）。另一种解决方案是创建一个Numpy阵列，然后使用内存视图（这成本更高，但可以防止任何可能的泄漏）。