From 44f144f1b7a5deb34ab6171fc4ea8bf7bcb1c20f Mon Sep 17 00:00:00 2001 From: Qingzi-Lan Date: Wed, 23 Aug 2023 13:46:16 -0700 Subject: [PATCH 1/3] feature: Neo service GA on TLV --- .../image-classification-neo.json | 3 ++- .../image_uri_config/inferentia-mxnet.json | 6 +++-- .../image_uri_config/inferentia-pytorch.json | 9 ++++--- .../inferentia-tensorflow.json | 6 +++-- src/sagemaker/image_uri_config/neo-mxnet.json | 3 ++- .../image_uri_config/neo-pytorch.json | 24 ++++++++++++------- .../image_uri_config/neo-tensorflow.json | 6 +++-- .../image_uri_config/xgboost-neo.json | 3 ++- tests/unit/sagemaker/image_uris/test_neo.py | 3 ++- 9 files changed, 42 insertions(+), 21 deletions(-) diff --git a/src/sagemaker/image_uri_config/image-classification-neo.json b/src/sagemaker/image_uri_config/image-classification-neo.json index 371bd445a9..66c2da8481 100644 --- a/src/sagemaker/image_uri_config/image-classification-neo.json +++ b/src/sagemaker/image_uri_config/image-classification-neo.json @@ -28,7 +28,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "image-classification-neo" } diff --git a/src/sagemaker/image_uri_config/inferentia-mxnet.json b/src/sagemaker/image_uri_config/inferentia-mxnet.json index d53c21f7e2..5a371430ce 100644 --- a/src/sagemaker/image_uri_config/inferentia-mxnet.json +++ b/src/sagemaker/image_uri_config/inferentia-mxnet.json @@ -30,7 +30,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-mxnet" }, @@ -62,7 +63,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-mxnet" } diff --git a/src/sagemaker/image_uri_config/inferentia-pytorch.json b/src/sagemaker/image_uri_config/inferentia-pytorch.json index b3d8c80c21..479ee29218 100644 --- a/src/sagemaker/image_uri_config/inferentia-pytorch.json +++ b/src/sagemaker/image_uri_config/inferentia-pytorch.json @@ -30,7 +30,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-pytorch" }, @@ -62,7 +63,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-pytorch" }, @@ -94,7 +96,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-pytorch" } diff --git a/src/sagemaker/image_uri_config/inferentia-tensorflow.json b/src/sagemaker/image_uri_config/inferentia-tensorflow.json index ce1d2603b6..ca2bbcdb8b 100644 --- a/src/sagemaker/image_uri_config/inferentia-tensorflow.json +++ b/src/sagemaker/image_uri_config/inferentia-tensorflow.json @@ -30,7 +30,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-tensorflow" }, @@ -62,7 +63,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-neo-tensorflow" } diff --git a/src/sagemaker/image_uri_config/neo-mxnet.json b/src/sagemaker/image_uri_config/neo-mxnet.json index 268abde298..730379a81b 100644 --- a/src/sagemaker/image_uri_config/neo-mxnet.json +++ b/src/sagemaker/image_uri_config/neo-mxnet.json @@ -44,7 +44,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-mxnet" } diff --git a/src/sagemaker/image_uri_config/neo-pytorch.json b/src/sagemaker/image_uri_config/neo-pytorch.json index 16e772f40b..b6a3e633af 100644 --- a/src/sagemaker/image_uri_config/neo-pytorch.json +++ b/src/sagemaker/image_uri_config/neo-pytorch.json @@ -48,7 +48,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -80,7 +81,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -112,7 +114,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -144,7 +147,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -176,7 +180,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -208,7 +213,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -240,7 +246,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" }, @@ -272,7 +279,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-pytorch" } diff --git a/src/sagemaker/image_uri_config/neo-tensorflow.json b/src/sagemaker/image_uri_config/neo-tensorflow.json index f7fb0fc5c3..4fd6c6c58b 100644 --- a/src/sagemaker/image_uri_config/neo-tensorflow.json +++ b/src/sagemaker/image_uri_config/neo-tensorflow.json @@ -50,7 +50,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-tensorflow" }, @@ -82,7 +83,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "sagemaker-inference-tensorflow" } diff --git a/src/sagemaker/image_uri_config/xgboost-neo.json b/src/sagemaker/image_uri_config/xgboost-neo.json index 8c44ff381c..96ac466bdf 100644 --- a/src/sagemaker/image_uri_config/xgboost-neo.json +++ b/src/sagemaker/image_uri_config/xgboost-neo.json @@ -28,7 +28,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009" + "us-west-2": "301217895009", + "il-central-1": "275950707576" }, "repository": "xgboost-neo" } diff --git a/tests/unit/sagemaker/image_uris/test_neo.py b/tests/unit/sagemaker/image_uris/test_neo.py index 61d758b403..f6b3d14647 100644 --- a/tests/unit/sagemaker/image_uris/test_neo.py +++ b/tests/unit/sagemaker/image_uris/test_neo.py @@ -45,7 +45,8 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009", + "us-west-2": "301217895009",, + "il-central-1": "275950707576" } INFERENTIA_REGIONS = ACCOUNTS.keys() From 0ec766beabc2a93501956cca48d8655de147bcab Mon Sep 17 00:00:00 2001 From: Qingzi-Lan Date: Mon, 28 Aug 2023 12:22:44 -0700 Subject: [PATCH 2/3] fix typo --- tests/unit/sagemaker/image_uris/test_neo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/sagemaker/image_uris/test_neo.py b/tests/unit/sagemaker/image_uris/test_neo.py index f6b3d14647..702c20ff7b 100644 --- a/tests/unit/sagemaker/image_uris/test_neo.py +++ b/tests/unit/sagemaker/image_uris/test_neo.py @@ -45,7 +45,7 @@ "us-iso-east-1": "167761179201", "us-isob-east-1": "406031935815", "us-west-1": "710691900526", - "us-west-2": "301217895009",, + "us-west-2": "301217895009", "il-central-1": "275950707576" } From 93ca88b85739f0660dcd2666045ab8523aff3caf Mon Sep 17 00:00:00 2001 From: Qingzi-Lan Date: Mon, 28 Aug 2023 12:23:52 -0700 Subject: [PATCH 3/3] format --- tests/unit/sagemaker/image_uris/test_neo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/sagemaker/image_uris/test_neo.py b/tests/unit/sagemaker/image_uris/test_neo.py index 702c20ff7b..2b51f0e9ea 100644 --- a/tests/unit/sagemaker/image_uris/test_neo.py +++ b/tests/unit/sagemaker/image_uris/test_neo.py @@ -46,7 +46,7 @@ "us-isob-east-1": "406031935815", "us-west-1": "710691900526", "us-west-2": "301217895009", - "il-central-1": "275950707576" + "il-central-1": "275950707576", } INFERENTIA_REGIONS = ACCOUNTS.keys()