Source code for unilab.ipc.shared_buffer

"""Base class for device-adaptive shared memory buffers."""

import torch



[docs]
class SharedBufferBase:
    """Device-adaptive shared memory buffer base class."""


[docs]
    def __init__(self, capacity: int, device: str, defer_gpu: bool = False):
        del defer_gpu
        self.capacity = capacity
        self.device = device
        self.ptr = torch.zeros(1, dtype=torch.int64).share_memory_()