Skip to content
Merged
Changes from 4 commits
Commits
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
8 changes: 6 additions & 2 deletions cuda_core/cuda/core/experimental/_memory.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -1208,8 +1208,12 @@ class VirtualMemoryResource(MemoryResource):
)
if self.config.location_type == "host":
self.device = None
if platform.system() == "Windows":
raise NotImplementedError("VirtualMemoryResource is not supported on Windows")

if device and not device.properties.virtual_memory_management_supported:
raise NotImplementedError("VirtualMemoryResource requires CUDA VMM API support")

if not device and self.config.location_type == "device":
raise ValueError("VirtualMemoryResource requires a device for device memory allocations")

# Validate RDMA support if requested
if self.config.gpu_direct_rdma and self.device is not None:
Expand Down