Skip to content

Commit b426c21

Browse files
sirutBuasaibencrabtree
sirutBuasai
authored and
bencrabtree
committed
feat: Update SM Python SDK for PT 2.2.0 SM DLC (aws#4481)
* update pt2.2 sm training dlc pysdk * update pt2.2 sm inference dlc pysdk and region list
1 parent 45a471f commit b426c21

File tree

4 files changed

+104
-3
lines changed

4 files changed

+104
-3
lines changed

‎src/sagemaker/fw_utils.py

+1
Original file line numberDiff line numberDiff line change
@@ -158,6 +158,7 @@
158158
"2.0.0",
159159
"2.0.1",
160160
"2.1.0",
161+
"2.2.0",
161162
]
162163

163164

‎src/sagemaker/image_uri_config/pytorch.json

+99-2
Original file line numberDiff line numberDiff line change
@@ -848,6 +848,7 @@
848848
"ap-northeast-2": "763104351884",
849849
"ap-northeast-3": "364406365360",
850850
"ap-south-1": "763104351884",
851+
"ap-south-2": "772153158452",
851852
"ap-southeast-1": "763104351884",
852853
"ap-southeast-2": "763104351884",
853854
"ap-southeast-3": "907027046896",
@@ -856,11 +857,13 @@
856857
"cn-north-1": "727897471807",
857858
"cn-northwest-1": "727897471807",
858859
"eu-central-1": "763104351884",
860+
"eu-central-2": "380420809688",
859861
"eu-north-1": "763104351884",
860862
"eu-west-1": "763104351884",
861863
"eu-west-2": "763104351884",
862864
"eu-west-3": "763104351884",
863865
"eu-south-1": "692866216735",
866+
"eu-south-2": "503227376785",
864867
"me-south-1": "217643126080",
865868
"sa-east-1": "763104351884",
866869
"us-east-1": "763104351884",
@@ -887,6 +890,7 @@
887890
"ap-northeast-2": "763104351884",
888891
"ap-northeast-3": "364406365360",
889892
"ap-south-1": "763104351884",
893+
"ap-south-2": "772153158452",
890894
"ap-southeast-1": "763104351884",
891895
"ap-southeast-2": "763104351884",
892896
"ap-southeast-3": "907027046896",
@@ -895,11 +899,13 @@
895899
"cn-north-1": "727897471807",
896900
"cn-northwest-1": "727897471807",
897901
"eu-central-1": "763104351884",
902+
"eu-central-2": "380420809688",
898903
"eu-north-1": "763104351884",
899904
"eu-west-1": "763104351884",
900905
"eu-west-2": "763104351884",
901906
"eu-west-3": "763104351884",
902907
"eu-south-1": "692866216735",
908+
"eu-south-2": "503227376785",
903909
"me-south-1": "217643126080",
904910
"sa-east-1": "763104351884",
905911
"us-east-1": "763104351884",
@@ -926,6 +932,7 @@
926932
"ap-northeast-2": "763104351884",
927933
"ap-northeast-3": "364406365360",
928934
"ap-south-1": "763104351884",
935+
"ap-south-2": "772153158452",
929936
"ap-southeast-1": "763104351884",
930937
"ap-southeast-2": "763104351884",
931938
"ap-southeast-3": "907027046896",
@@ -934,11 +941,13 @@
934941
"cn-north-1": "727897471807",
935942
"cn-northwest-1": "727897471807",
936943
"eu-central-1": "763104351884",
944+
"eu-central-2": "380420809688",
937945
"eu-north-1": "763104351884",
938946
"eu-west-1": "763104351884",
939947
"eu-west-2": "763104351884",
940948
"eu-west-3": "763104351884",
941949
"eu-south-1": "692866216735",
950+
"eu-south-2": "503227376785",
942951
"me-south-1": "217643126080",
943952
"sa-east-1": "763104351884",
944953
"us-east-1": "763104351884",
@@ -965,6 +974,7 @@
965974
"ap-northeast-2": "763104351884",
966975
"ap-northeast-3": "364406365360",
967976
"ap-south-1": "763104351884",
977+
"ap-south-2": "772153158452",
968978
"ap-southeast-1": "763104351884",
969979
"ap-southeast-2": "763104351884",
970980
"ap-southeast-3": "907027046896",
@@ -973,11 +983,55 @@
973983
"cn-north-1": "727897471807",
974984
"cn-northwest-1": "727897471807",
975985
"eu-central-1": "763104351884",
986+
"eu-central-2": "380420809688",
976987
"eu-north-1": "763104351884",
977988
"eu-west-1": "763104351884",
978989
"eu-west-2": "763104351884",
979990
"eu-west-3": "763104351884",
980991
"eu-south-1": "692866216735",
992+
"eu-south-2": "503227376785",
993+
"me-south-1": "217643126080",
994+
"sa-east-1": "763104351884",
995+
"us-east-1": "763104351884",
996+
"us-east-2": "763104351884",
997+
"us-gov-east-1": "446045086412",
998+
"us-gov-west-1": "442386744353",
999+
"us-iso-east-1": "886529160074",
1000+
"us-isob-east-1": "094389454867",
1001+
"us-west-1": "763104351884",
1002+
"us-west-2": "763104351884",
1003+
"ca-west-1": "204538143572"
1004+
},
1005+
"repository": "pytorch-inference"
1006+
},
1007+
"2.2.0": {
1008+
"py_versions": [
1009+
"py310"
1010+
],
1011+
"registries": {
1012+
"af-south-1": "626614931356",
1013+
"il-central-1": "780543022126",
1014+
"ap-east-1": "871362719292",
1015+
"ap-northeast-1": "763104351884",
1016+
"ap-northeast-2": "763104351884",
1017+
"ap-northeast-3": "364406365360",
1018+
"ap-south-1": "763104351884",
1019+
"ap-south-2": "772153158452",
1020+
"ap-southeast-1": "763104351884",
1021+
"ap-southeast-2": "763104351884",
1022+
"ap-southeast-3": "907027046896",
1023+
"ap-southeast-4": "457447274322",
1024+
"ca-central-1": "763104351884",
1025+
"cn-north-1": "727897471807",
1026+
"cn-northwest-1": "727897471807",
1027+
"eu-central-1": "763104351884",
1028+
"eu-central-2": "380420809688",
1029+
"eu-north-1": "763104351884",
1030+
"eu-west-1": "763104351884",
1031+
"eu-west-2": "763104351884",
1032+
"eu-west-3": "763104351884",
1033+
"eu-south-1": "692866216735",
1034+
"eu-south-2": "503227376785",
9811035
"me-south-1": "217643126080",
9821036
"sa-east-1": "763104351884",
9831037
"us-east-1": "763104351884",
@@ -1190,7 +1244,8 @@
11901244
"1.12": "1.12.1",
11911245
"1.13": "1.13.1",
11921246
"2.0": "2.0.1",
1193-
"2.1": "2.1.0"
1247+
"2.1": "2.1.0",
1248+
"2.2": "2.2.0"
11941249
},
11951250
"versions": {
11961251
"0.4.0": {
@@ -2113,7 +2168,49 @@
21132168
"ca-west-1": "204538143572"
21142169
},
21152170
"repository": "pytorch-training"
2171+
},
2172+
"2.2.0": {
2173+
"py_versions": [
2174+
"py310"
2175+
],
2176+
"registries": {
2177+
"af-south-1": "626614931356",
2178+
"il-central-1": "780543022126",
2179+
"ap-east-1": "871362719292",
2180+
"ap-northeast-1": "763104351884",
2181+
"ap-northeast-2": "763104351884",
2182+
"ap-northeast-3": "364406365360",
2183+
"ap-south-1": "763104351884",
2184+
"ap-south-2": "772153158452",
2185+
"ap-southeast-1": "763104351884",
2186+
"ap-southeast-2": "763104351884",
2187+
"ap-southeast-3": "907027046896",
2188+
"ap-southeast-4": "457447274322",
2189+
"ca-central-1": "763104351884",
2190+
"cn-north-1": "727897471807",
2191+
"cn-northwest-1": "727897471807",
2192+
"eu-central-1": "763104351884",
2193+
"eu-central-2": "380420809688",
2194+
"eu-north-1": "763104351884",
2195+
"eu-west-1": "763104351884",
2196+
"eu-west-2": "763104351884",
2197+
"eu-west-3": "763104351884",
2198+
"eu-south-1": "692866216735",
2199+
"eu-south-2": "503227376785",
2200+
"me-south-1": "217643126080",
2201+
"sa-east-1": "763104351884",
2202+
"us-east-1": "763104351884",
2203+
"us-east-2": "763104351884",
2204+
"us-gov-east-1": "446045086412",
2205+
"us-gov-west-1": "442386744353",
2206+
"us-iso-east-1": "886529160074",
2207+
"us-isob-east-1": "094389454867",
2208+
"us-west-1": "763104351884",
2209+
"us-west-2": "763104351884",
2210+
"ca-west-1": "204538143572"
2211+
},
2212+
"repository": "pytorch-training"
21162213
}
21172214
}
21182215
}
2119-
}
2216+
}

‎tests/unit/test_fw_utils.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -998,6 +998,7 @@ def test_validate_pytorchddp_not_raises():
998998
"2.0.0",
999999
"2.0.1",
10001000
"2.1.0",
1001+
"2.2.0",
10011002
]
10021003
for framework_version in pytorchddp_supported_fw_versions:
10031004
fw_utils.validate_pytorch_distribution(
@@ -1060,7 +1061,7 @@ def test_validate_torch_distributed_not_raises():
10601061

10611062
# Case 3: Distribution is torch_distributed enabled, supported framework and instances
10621063
torch_distributed_enabled = {"torch_distributed": {"enabled": True}}
1063-
torch_distributed_gpu_supported_fw_versions = ["1.13.1", "2.0.0", "2.0.1", "2.1.0"]
1064+
torch_distributed_gpu_supported_fw_versions = ["1.13.1", "2.0.0", "2.0.1", "2.1.0", "2.2.0"]
10641065
for framework_version in torch_distributed_gpu_supported_fw_versions:
10651066
fw_utils.validate_torch_distributed_distribution(
10661067
instance_type="ml.p3.8xlarge",

‎tests/unit/test_utils.py

+2
Original file line numberDiff line numberDiff line change
@@ -384,6 +384,8 @@ def test_set_nested_value():
384384

385385

386386
def test_get_short_version():
387+
assert sagemaker.utils.get_short_version("2.2.0") == "2.2"
388+
assert sagemaker.utils.get_short_version("2.2") == "2.2"
387389
assert sagemaker.utils.get_short_version("2.1.0") == "2.1"
388390
assert sagemaker.utils.get_short_version("2.1") == "2.1"
389391
assert sagemaker.utils.get_short_version("2.0.1") == "2.0"

0 commit comments

Comments
 (0)