fix: Update Graviton supported instance families (aws#3443)

Author: knikure

src/sagemaker/fw_utils.py

@@ -144,7 +144,16 @@
 SMDISTRIBUTED_SUPPORTED_STRATEGIES = ["dataparallel", "modelparallel"]
 
 
-GRAVITON_ALLOWED_TARGET_INSTANCE_FAMILY = ["c6g", "t4g", "r6g", "m6g"]
+GRAVITON_ALLOWED_TARGET_INSTANCE_FAMILY = [
+    "m6g",
+    "m6gd",
+    "c6g",
+    "c6gd",
+    "c6gn",
+    "c7g",
+    "r6g",
+    "r6gd",
+]
 
 
 GRAVITON_ALLOWED_FRAMEWORKS = set(["tensorflow", "pytorch", "xgboost", "sklearn"])

tests/unit/sagemaker/image_uris/test_graviton.py

@@ -17,12 +17,26 @@
 
 import pytest
 
+GRAVITON_ALLOWED_TARGET_INSTANCE_FAMILY = [
+    "m6g",
+    "m6gd",
+    "c6g",
+    "c6gd",
+    "c6gn",
+    "c7g",
+    "r6g",
+    "r6gd",
+]
 GRAVITON_ALGOS = ("tensorflow", "pytotch")
 GRAVITON_INSTANCE_TYPES = [
-    "ml.c6g.4xlarge",
-    "ml.t4g.2xlarge",
-    "ml.r6g.2xlarge",
     "ml.m6g.4xlarge",
+    "ml.m6gd.2xlarge",
+    "ml.c6g.2xlarge",
+    "ml.c6gd.4xlarge",
+    "ml.c6gn.4xlarge",
+    "ml.c7g.2xlarge",
+    "ml.r6g.2xlarge",
+    "ml.r6gd.4xlarge",
 ]
 
 ACCOUNTS = {