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