Skip to content

Commit f69cd81

Browse files
committed
rdma: Add index to domain_rail
Add a rail_id to the domain (similar to endpoint and device) for future patches to clean up cq handling. Signed-off-by: Brian Barrett <[email protected]>
1 parent b623230 commit f69cd81

File tree

2 files changed

+4
-0
lines changed

2 files changed

+4
-0
lines changed

include/nccl_ofi_rdma.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -830,6 +830,8 @@ typedef struct nccl_net_ofi_rdma_device {
830830

831831

832832
typedef struct nccl_net_ofi_rdma_domain_rail {
833+
uint16_t rail_id;
834+
833835
/* Access domain handles */
834836
struct fid_domain *domain;
835837

src/nccl_ofi_rdma.cpp

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7457,6 +7457,8 @@ static nccl_net_ofi_domain_t *nccl_net_ofi_rdma_device_create_domain(nccl_net_of
74577457
nccl_net_ofi_rdma_device_rail_t *device_rail = rdma_device_get_rail(device, i);
74587458
nccl_net_ofi_rdma_domain_rail_t *domain_rail = rdma_domain_get_rail(domain, i);
74597459

7460+
domain_rail->rail_id = i;
7461+
74607462
ret = fi_domain(device_rail->fabric, device_rail->info,
74617463
&domain_rail->domain, NULL);
74627464
if (OFI_UNLIKELY(ret != 0)) {

0 commit comments

Comments
 (0)