python - 在 Python 中访问共享内存的段错误

Question

我在尝试访问共享内存时遇到段错误，我将其归结为以下示例。 f1 和 f2 是相同的，除了我将共享内存对象存储在 f2 的一个额外变量中，但是 f2 有效，f1 段错误。为什么会这样？

(以相反的顺序调用这两个函数不会改变行为。)

from multiprocessing import shared_memory
import numpy as np

segment_name = "segment"
data_shape = (5,5,5)


def get_view(shm):
    shape = (1,) + data_shape
    arr_one = np.ndarray(shape, dtype=np.float64, buffer=shm.buf)
    arr_two = np.ndarray(data_shape, dtype=np.float64, buffer=arr_one[0])
    return arr_two


def f1():
    arr = get_view(shared_memory.SharedMemory(name=segment_name))
    print(arr[0][0][0])


def f2():
    shm = shared_memory.SharedMemory(name=segment_name)
    arr = get_view(shm)
    print(arr[0][0][0])


if __name__ == '__main__':
    shape = (1,) + data_shape
    dummy = np.ndarray(shape, dtype=np.float64)
    shared_memory.SharedMemory(name=segment_name, create=True, size=dummy.nbytes)
    f2()
    f1()

Answer 1

事实证明，额外的引用是必要的。如果 arr 不指向共享内存对象，它的 del 方法将被立即调用并取消链接其引用由 numpy 数组保存的底层缓冲区。