Skip to content

Commit fe6d6f5

Browse files
committed
hotfix: use correct hca selector
1 parent b5493bf commit fe6d6f5

File tree

1 file changed

+1
-3
lines changed

1 file changed

+1
-3
lines changed

checkpoint_engine/ps.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -303,9 +303,7 @@ def _get_rdma_devices() -> list[str]:
303303
return hca_list
304304
else:
305305
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-
]
306+
return [device for device in sorted(_ibv_get_device_list()) if hca is None or hca in device]
309307

310308

311309
def _get_my_rdma_device(local_rank: int, gpu_count: int, devices: list[str]) -> str:

0 commit comments

Comments
 (0)