Skip to content

Commit 5906559

Browse files
arjkeshbencrabtreenavinsonijeniyatDewen Qi
committed
feature: Add support for TF2.7 (aws#2958)
Co-authored-by: Ben Crabtree <[email protected]> Co-authored-by: Navin Soni <[email protected]> Co-authored-by: Jeniya Tabassum <[email protected]> Co-authored-by: Dewen Qi <[email protected]> Co-authored-by: Payton Staub <[email protected]> Co-authored-by: qidewenwhen <[email protected]> Co-authored-by: Qingzi-Lan <[email protected]> Co-authored-by: Payton Staub <[email protected]> Co-authored-by: Miyoung <[email protected]> Co-authored-by: Shreya Pandit <[email protected]> Co-authored-by: Ahsan Khan <[email protected]> Co-authored-by: Mufaddal Rohawala <[email protected]>
1 parent 789afc7 commit 5906559

File tree

4 files changed

+73
-0
lines changed

4 files changed

+73
-0
lines changed

‎src/sagemaker/fw_utils.py

+2
Original file line numberDiff line numberDiff line change
@@ -75,6 +75,8 @@
7575
"2.6.0",
7676
"2.6.2",
7777
"2.6.3",
78+
"2.7",
79+
"2.7.1",
7880
"2.8",
7981
"2.8.0",
8082
],

‎src/sagemaker/image_uri_config/tensorflow.json

+65
Original file line numberDiff line numberDiff line change
@@ -280,6 +280,7 @@
280280
"2.4": "2.4.3",
281281
"2.5": "2.5.1",
282282
"2.6": "2.6.3",
283+
"2.7": "2.7.0",
283284
"2.8": "2.8.0"
284285
},
285286
"versions": {
@@ -1375,6 +1376,36 @@
13751376
},
13761377
"repository": "tensorflow-inference"
13771378
},
1379+
"2.7.0": {
1380+
"registries": {
1381+
"af-south-1": "626614931356",
1382+
"ap-east-1": "871362719292",
1383+
"ap-northeast-1": "763104351884",
1384+
"ap-northeast-2": "763104351884",
1385+
"ap-northeast-3": "364406365360",
1386+
"ap-south-1": "763104351884",
1387+
"ap-southeast-1": "763104351884",
1388+
"ap-southeast-2": "763104351884",
1389+
"ca-central-1": "763104351884",
1390+
"cn-north-1": "727897471807",
1391+
"cn-northwest-1": "727897471807",
1392+
"eu-central-1": "763104351884",
1393+
"eu-north-1": "763104351884",
1394+
"eu-south-1": "692866216735",
1395+
"eu-west-1": "763104351884",
1396+
"eu-west-2": "763104351884",
1397+
"eu-west-3": "763104351884",
1398+
"me-south-1": "217643126080",
1399+
"sa-east-1": "763104351884",
1400+
"us-east-1": "763104351884",
1401+
"us-east-2": "763104351884",
1402+
"us-gov-west-1": "442386744353",
1403+
"us-iso-east-1": "886529160074",
1404+
"us-west-1": "763104351884",
1405+
"us-west-2": "763104351884"
1406+
},
1407+
"repository": "tensorflow-inference"
1408+
},
13781409
"2.8.0": {
13791410
"registries": {
13801411
"af-south-1": "626614931356",
@@ -1432,6 +1463,7 @@
14321463
"2.4": "2.4.3",
14331464
"2.5": "2.5.1",
14341465
"2.6": "2.6.3",
1466+
"2.7": "2.7.1",
14351467
"2.8": "2.8.0"
14361468
},
14371469
"versions": {
@@ -2725,6 +2757,39 @@
27252757
},
27262758
"repository": "tensorflow-training"
27272759
},
2760+
"2.7.1": {
2761+
"py_versions": [
2762+
"py38"
2763+
],
2764+
"registries": {
2765+
"af-south-1": "626614931356",
2766+
"ap-east-1": "871362719292",
2767+
"ap-northeast-1": "763104351884",
2768+
"ap-northeast-2": "763104351884",
2769+
"ap-northeast-3": "364406365360",
2770+
"ap-south-1": "763104351884",
2771+
"ap-southeast-1": "763104351884",
2772+
"ap-southeast-2": "763104351884",
2773+
"ca-central-1": "763104351884",
2774+
"cn-north-1": "727897471807",
2775+
"cn-northwest-1": "727897471807",
2776+
"eu-central-1": "763104351884",
2777+
"eu-north-1": "763104351884",
2778+
"eu-south-1": "692866216735",
2779+
"eu-west-1": "763104351884",
2780+
"eu-west-2": "763104351884",
2781+
"eu-west-3": "763104351884",
2782+
"me-south-1": "217643126080",
2783+
"sa-east-1": "763104351884",
2784+
"us-east-1": "763104351884",
2785+
"us-east-2": "763104351884",
2786+
"us-gov-west-1": "442386744353",
2787+
"us-iso-east-1": "886529160074",
2788+
"us-west-1": "763104351884",
2789+
"us-west-2": "763104351884"
2790+
},
2791+
"repository": "tensorflow-training"
2792+
},
27282793
"2.8.0": {
27292794
"py_versions": [
27302795
"py39"

‎tests/integ/test_tf.py

+3
Original file line numberDiff line numberDiff line change
@@ -224,6 +224,9 @@ def test_mnist_distributed(
224224

225225
@pytest.mark.slow_test
226226
def test_mnist_async(sagemaker_session, cpu_instance_type, tf_full_version, tf_full_py_version):
227+
if tf_full_version == "2.7.0":
228+
tf_full_version = "2.7"
229+
227230
estimator = TensorFlow(
228231
entry_point=SCRIPT,
229232
source_dir=MNIST_RESOURCE_PATH,

‎tests/unit/test_fw_utils.py

+3
Original file line numberDiff line numberDiff line change
@@ -687,6 +687,8 @@ def test_validate_smdataparallel_args_not_raises():
687687
("ml.p3.16xlarge", "tensorflow", "2.6.2", "py38", smdataparallel_enabled),
688688
("ml.p3.16xlarge", "tensorflow", "2.6.3", "py38", smdataparallel_enabled),
689689
("ml.p3.16xlarge", "tensorflow", "2.6", "py38", smdataparallel_enabled),
690+
("ml.p3.16xlarge", "tensorflow", "2.7.1", "py38", smdataparallel_enabled),
691+
("ml.p3.16xlarge", "tensorflow", "2.7", "py38", smdataparallel_enabled),
690692
("ml.p3.16xlarge", "tensorflow", "2.8.0", "py39", smdataparallel_enabled),
691693
("ml.p3.16xlarge", "tensorflow", "2.8", "py39", smdataparallel_enabled),
692694
("ml.p3.16xlarge", "pytorch", "1.6.0", "py3", smdataparallel_enabled),
@@ -707,6 +709,7 @@ def test_validate_smdataparallel_args_not_raises():
707709
("ml.p3.16xlarge", "tensorflow", "2.6.0", "py38", smdataparallel_enabled_custom_mpi),
708710
("ml.p3.16xlarge", "tensorflow", "2.6.2", "py38", smdataparallel_enabled_custom_mpi),
709711
("ml.p3.16xlarge", "tensorflow", "2.6.3", "py38", smdataparallel_enabled_custom_mpi),
712+
("ml.p3.16xlarge", "tensorflow", "2.7.1", "py38", smdataparallel_enabled_custom_mpi),
710713
("ml.p3.16xlarge", "tensorflow", "2.8.0", "py39", smdataparallel_enabled_custom_mpi),
711714
("ml.p3.16xlarge", "pytorch", "1.8.0", "py3", smdataparallel_enabled_custom_mpi),
712715
("ml.p3.16xlarge", "pytorch", "1.9.1", "py38", smdataparallel_enabled_custom_mpi),

0 commit comments

Comments
 (0)