verl/utils/megatron/memory.py

• MemoryBuffer · class · L18-L41 — class MemoryBuffer
• init · method · L20-L27 — def init(self, numel, numel_padded, dtype)
• zero · method · L29-L31 — def zero(self)
• get · method · L33-L41 — def get(self, shape, start_index)