Skip to content

Commit d0dfb9c

Browse files
mawad-amdclaude
andcommitted
Fix stale import: iris.allocators -> iris.host.memory.allocators
test_rocr_behaviors.py still referenced the old flat path for VMemAllocator. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 4efb825 commit d0dfb9c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

‎tests/unittests/test_rocr_behaviors.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -197,7 +197,7 @@ def test_vmem_allocator_uses_cumulative_access():
197197
"""
198198
import torch.distributed as dist
199199

200-
from iris.allocators.vmem_allocator import VMemAllocator
200+
from iris.host.memory.allocators.vmem_allocator import VMemAllocator
201201

202202
rank = dist.get_rank()
203203
world_size = dist.get_world_size()

0 commit comments

Comments
 (0)