Merge pull request #739 from Ahuge/patch-1

Add G6/GR6 L4 GPU specifications to scrape.py
vantage-sh · May 2, 2024 · 7cbb727 · 7cbb727
2 parents 9ab540f + a8d3956
commit 7cbb727
Showing 1 changed file with 74 additions and 0 deletions.
diff --git a/scrape.py b/scrape.py
@@ -534,6 +534,8 @@ def add_pretty_names(instances):
         "g3": "G3 Graphics GPU",
         "g4": "G4 Graphics and Machine Learning GPU",
         "g5": "G5 Graphics and Machine Learning GPU",
+        "g6": "G6 Graphics and Machine Learning GPU",
+        "gr6": "Gr6 Graphics and Machine Learning GPU High RAM ratio",
         "p2": "P2 General Purpose GPU",
         "p3": "P3 High Performance GPU",
         "p4d": "P4D Highest Performance GPU",
@@ -817,6 +819,78 @@ def add_gpu_info(instances):
             "cuda_cores": 76928,
             "gpu_memory": 192,
         },
+        "g6.xlarge": {  
+            # GPU core count found from the whitepaper
+            # https://images.nvidia.com/aem-dam/Solutions/Data-Center/l4/nvidia-ada-gpu-architecture-whitepaper-v2.1.pdf
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "g6.2xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "g6.4xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "g6.8xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "gr6.4xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "gr6.8xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "g6.16xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 1,
+            "cuda_cores": 7424,
+            "gpu_memory": 24,
+        },
+        "g6.12xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 4,
+            "cuda_cores": 29696,
+            "gpu_memory": 96,
+        },
+        "g6.24xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 4,
+            "cuda_cores": 29696,
+            "gpu_memory": 96,
+        },
+        "g6.48xlarge": {
+            "gpu_model": "NVIDIA L4",
+            "compute_capability": 8.9,
+            "gpu_count": 8,
+            "cuda_cores": 59392,
+            "gpu_memory": 192,
+        },
         "p4d.24xlarge": {
             "gpu_model": "NVIDIA A100",
             "compute_capability": 8.0,