From b6df651cb52d7605f602ecf667dbaca6aad47753 Mon Sep 17 00:00:00 2001
From: Andrew Smith <andjsmi@amazon.com>
Date: Thu, 5 Sep 2024 09:21:38 +1000
Subject: [PATCH] Added p5 as a supported NCCL instance

---
 src/sagemaker_training/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/sagemaker_training/__init__.py b/src/sagemaker_training/__init__.py
index 725935b9..6b18f749 100644
--- a/src/sagemaker_training/__init__.py
+++ b/src/sagemaker_training/__init__.py
@@ -81,6 +81,7 @@
     "ml.p3dn.24xlarge",
     "ml.p4d.24xlarge",
     "ml.p4de.24xlarge",
+    "ml.p5.48xlarge",
     "ml.trn1.32xlarge",
 ]