Skip to content

Commit 9109fa1

Browse files
author
Mike Schneider
committed
adding PT 1.13.1 for smdataparallel.
1 parent d598c07 commit 9109fa1

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/unit/test_fw_utils.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -912,6 +912,8 @@ def test_validate_smdataparallel_args_not_raises():
912912
("ml.p3.16xlarge", "pytorch", "1.12.0", "py38", smdataparallel_enabled),
913913
("ml.p3.16xlarge", "pytorch", "1.12.1", "py38", smdataparallel_enabled),
914914
("ml.p3.16xlarge", "pytorch", "1.12", "py38", smdataparallel_enabled),
915+
("ml.p3.16xlarge", "pytorch", "1.13", "py39", smdataparallel_enabled),
916+
("ml.p3.16xlarge", "pytorch", "1.13.1", "py39", smdataparallel_enabled),
915917
("ml.p3.16xlarge", "tensorflow", "2.4.1", "py3", smdataparallel_enabled_custom_mpi),
916918
("ml.p3.16xlarge", "tensorflow", "2.4.1", "py37", smdataparallel_enabled_custom_mpi),
917919
("ml.p3.16xlarge", "tensorflow", "2.4.3", "py3", smdataparallel_enabled_custom_mpi),

0 commit comments

Comments
 (0)