We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b5493bf commit fe6d6f5Copy full SHA for fe6d6f5
checkpoint_engine/ps.py
@@ -303,9 +303,7 @@ def _get_rdma_devices() -> list[str]:
303
return hca_list
304
else:
305
hca = hca_list[0]
306
- return [
307
- device for device in sorted(_ibv_get_device_list()) if hca is not None and hca in device
308
- ]
+ return [device for device in sorted(_ibv_get_device_list()) if hca is None or hca in device]
309
310
311
def _get_my_rdma_device(local_rank: int, gpu_count: int, devices: list[str]) -> str:
0 commit comments