Module device_selection

Module device_selection 

Source
Expand description

ibverbs-specific device selection logic that pairs compute devices with the best available RDMA NICs based on PCI topology distance.

Functionsยง

create_cuda_to_ibv_mapping
Creates a mapping from CUDA PCI addresses to optimal RDMA devices.
resolve_ibv_device
Resolves RDMA device using auto-detection logic when needed.
select_optimal_ibv_device
Step 1: Parse device string into prefix and postfix Step 2: Get PCI address from compute device Step 3: Get PCI address for all RDMA NIC devices Step 4: Calculate PCI distances and return closest RDMA NIC device