Skip to content

Commit 9b47216

Browse files
authored
Update libraries for SMDDP collectives validation (#170)
1 parent 1020f3f commit 9b47216

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/sagemaker_training/smdataparallel.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,6 @@
4242
SMDDP_LIB_PATHS = [
4343
"/opt/conda/lib/libsmddp.so",
4444
"/opt/conda/lib/libsmddpcoll.so",
45-
"/opt/conda/lib/liboutofband.so",
46-
"/opt/conda/lib/libgloo.so",
4745
]
4846

4947

@@ -476,9 +474,9 @@ def _validate_smddp_coll_libs_present():
476474
deep-learning-containers package.
477475
478476
libsmddp.so is preloaded for training when SMDDP Collectives are used.
479-
The other libraries libsmddpcoll.so, liboutofband.so and libgloo.so are
480-
essential for successful training. Thus, if the libraries are absent,
481-
we want to avoid training failure by disabling SMDDP Collectives.
477+
The other library libsmddpcoll.so is also essential for successful training.
478+
Thus, if the libraries are absent, we want to avoid training failure
479+
by disabling SMDDP Collectives.
482480
"""
483481
for path in SMDDP_LIB_PATHS:
484482
if not os.path.exists(path):

0 commit comments

Comments
 (0)