Skip to content

Commit 6ddbf02

Browse files
committed
rdma: Add INIT specifier to forced rail count prints
They were previously NET-only, which are not enabled by default. Signed-off-by: Eric Raut <eraut@amazon.com>
1 parent 2c8fdb6 commit 6ddbf02

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/nccl_ofi_rdma.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7031,10 +7031,10 @@ nccl_net_ofi_rdma_device_t::nccl_net_ofi_rdma_device_t(nccl_net_ofi_plugin_t *pl
70317031
length = target_length;
70327032
info_list = new_list;
70337033
}
7034-
NCCL_OFI_INFO(NCCL_NET, "Created device with %zu rails (originally found %zu rails)",
7034+
NCCL_OFI_INFO(NCCL_INIT | NCCL_NET, "Created device with %zu rails (originally found %zu rails)",
70357035
length, original_length);
70367036
} else {
7037-
NCCL_OFI_INFO(NCCL_NET, "Created device with %zu rails", length);
7037+
NCCL_OFI_INFO(NCCL_INIT | NCCL_NET, "Created device with %zu rails", length);
70387038
}
70397039

70407040
/* Set NIC information */

0 commit comments

Comments
 (0)