From ec5e7aa197915a96c5d170bb692388ae40b0641a Mon Sep 17 00:00:00 2001
From: haixiw <haixiw@amazon.com>
Date: Thu, 6 Jun 2024 17:32:50 +0000
Subject: [PATCH] chore: cover Nvidia T4/L4 GPU

---
 sagemaker-entrypoint-cuda-all.sh | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/sagemaker-entrypoint-cuda-all.sh b/sagemaker-entrypoint-cuda-all.sh
index 4cf645ff..a3c63cbb 100644
--- a/sagemaker-entrypoint-cuda-all.sh
+++ b/sagemaker-entrypoint-cuda-all.sh
@@ -58,6 +58,12 @@ get_compute_cap() {
     # Check if the GPU name contains "H100"
     elif [[ "$gpu_name" == *"H100"* ]]; then
         echo "90"
+    # Cover Nvidia T4
+    elif [[ "$gpu_name" == *"T4"* ]]; then
+        echo "75"
+    # Cover Nvidia L4
+    elif [[ "$gpu_name" == *"L4"* ]]; then
+        echo "89"
     else
         echo "80"  # Default compute capability
     fi