Skip to content

Implement power-of-two VMem allocator #441

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Draft
Copilot wants to merge 3 commits into
base: main
Choose a base branch
from
Draft

Implement power-of-two VMem allocator #441

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
92e25d7
Initial plan
Copilot Mar 9, 2026
52dd8a8
Implement power-of-two VMem allocator (VMemPow2Allocator)
Copilot Mar 9, 2026
17edd0b
Address review feedback: fix all 12 issues in VMemPow2Allocator
Copilot Mar 9, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion iris/allocators/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -8,5 +8,6 @@
from .base import BaseAllocator
from .torch_allocator import TorchAllocator
from .vmem_allocator import VMemAllocator
from .vmem_pow2_allocator import VMemPow2Allocator

__all__ = ["BaseAllocator", "TorchAllocator", "VMemAllocator"]
__all__ = ["BaseAllocator", "TorchAllocator", "VMemAllocator", "VMemPow2Allocator"]
10 changes: 6 additions & 4 deletions iris/allocators/vmem_allocator.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -107,14 +107,16 @@ def get_allocation_segments(self):
Get list of allocation segments for segmented DMA-BUF export.

Returns:
List of (offset, size, va) tuples for each allocation in order.
Each tuple describes one physically-backed segment that needs
to be exported/imported separately.
List of ``(offset, size, va, generation)`` tuples for each
allocation in order. Each tuple describes one physically-backed
segment that needs to be exported/imported separately.
*generation* is always ``0`` for this allocator since it never
remaps a VA with fresh physical memory.
"""
segments = []
for offset, size in self.allocation_order:
va = self.base_va + offset
segments.append((offset, size, va))
segments.append((offset, size, va, 0))
return segments

def get_minimum_allocation_size(self) -> int:
Expand Down
Loading