From b970a1afa87f3bd4f8c0cd564f384e30380e8bdf Mon Sep 17 00:00:00 2001
From: Pei Ran Li <richprli@amazon.com>
Date: Wed, 24 Jul 2024 21:26:15 -0700
Subject: [PATCH] added support for ml.g6 instances

---
 .../aws-sagemaker-alpha/lib/instance-type.ts  | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/packages/@aws-cdk/aws-sagemaker-alpha/lib/instance-type.ts b/packages/@aws-cdk/aws-sagemaker-alpha/lib/instance-type.ts
index fbd612b56c02b..3c69defc9ffa3 100644
--- a/packages/@aws-cdk/aws-sagemaker-alpha/lib/instance-type.ts
+++ b/packages/@aws-cdk/aws-sagemaker-alpha/lib/instance-type.ts
@@ -204,6 +204,46 @@ export class InstanceType {
    */
   public static readonly G5_XLARGE = InstanceType.of('ml.g5.xlarge');
 
+  /**
+   * ml.g6.12xlarge
+   */
+  public static readonly G6_12XLARGE = InstanceType.of('ml.g6.12xlarge');
+
+  /**
+   * ml.g6.16xlarge
+   */
+  public static readonly G6_16XLARGE = InstanceType.of('ml.g6.16xlarge');
+
+  /**
+   * ml.g6.24xlarge
+   */
+  public static readonly G6_24XLARGE = InstanceType.of('ml.g6.24xlarge');
+
+  /**
+   * ml.g6.2xlarge
+   */
+  public static readonly G6_2XLARGE = InstanceType.of('ml.g6.2xlarge');
+
+  /**
+   * ml.g6.48xlarge
+   */
+  public static readonly G6_48XLARGE = InstanceType.of('ml.g6.48xlarge');
+
+  /**
+   * ml.g6.4xlarge
+   */
+  public static readonly G6_4XLARGE = InstanceType.of('ml.g6.4xlarge');
+
+  /**
+   * ml.g6.8xlarge
+   */
+  public static readonly G6_8XLARGE = InstanceType.of('ml.g6.8xlarge');
+
+  /**
+   * ml.g6.xlarge
+   */
+  public static readonly G6_XLARGE = InstanceType.of('ml.g6.xlarge');
+
   /**
    * ml.inf1.24xlarge
    */