Skip to content

Commit f69c808

Browse files
committed
Fix EFA device id for DMA-BUF search
The Libfabric strings include the 0x prefix for the device id, which our simple string match didn't account for. Signed-off-by: Brian Barrett <[email protected]>
1 parent 047cf33 commit f69c808

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/nccl_ofi_net.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -604,9 +604,9 @@ int nccl_net_ofi_info_properties(nccl_net_ofi_plugin_t *plugin, struct fi_info *
604604
NCCL_OFI_TRACE(NCCL_INIT | NCCL_NET,
605605
"DMA-BUF disabled due to missing nic data");
606606
props->dmabuf_support = false;
607-
} else if (strcmp("efa0", nic_prov->nic->device_attr->device_id) == 0 ||
608-
strcmp("efa1", nic_prov->nic->device_attr->device_id) == 0 ||
609-
strcmp("efa2", nic_prov->nic->device_attr->device_id) == 0) {
607+
} else if (strcmp("0xefa0", nic_prov->nic->device_attr->device_id) == 0 ||
608+
strcmp("0xefa1", nic_prov->nic->device_attr->device_id) == 0 ||
609+
strcmp("0xefa2", nic_prov->nic->device_attr->device_id) == 0) {
610610
NCCL_OFI_TRACE(NCCL_INIT | NCCL_NET,
611611
"DMA-BUF disabled due to EFA device id %s",
612612
nic_prov->nic->device_attr->device_id);

0 commit comments

Comments
 (0)