Add SuperNova on trn1

results.py

@@ -157,6 +157,13 @@ instance_type_mappings = {
         "url": "https://instances.vantage.sh/aws/ec2/inf2.48xlarge",
         "price": 12.981,
     },
+    "trn1.32xlarge": {
+        "cloud": "AWS",
+        "gpu": "32xNeuronCore v2",
+        "gpuRAM": "512 GB",
+        "url": "https://instances.vantage.sh/aws/ec2/trn1.32xlarge",
+        "price": 21.50,
+    },
 }
 
 results = {

results_arcee_supernova.py

@@ -45,6 +45,13 @@ results_arcee_supernova = {
                     "tokensPerSecond": "-",
                     "notes": "OOM bs=2,seqlen=2048 - SDK 2.19.1",
                 },
+                {
+                    "quantization": "8-bit",
+                    "container": "LMI 0.29+transformers-neuronx 0.11.351",
+                    "status": "???",
+                    "tokensPerSecond": "???",
+                    "notes": "bs=2,seqlen=8192 - SDK 2.19.1",
+                },
             ],
         },
         {
@@ -73,6 +80,18 @@ results_arcee_supernova = {
                 },
             ],
         },
+        {
+            "instanceType": "trn1.32xlarge",
+            "configurations": [
+                {
+                    "quantization": "none",
+                    "container": "LMI 0.29+transformers-neuronx 0.11.351",
+                    "status": "OK",
+                    "tokensPerSecond": "32",
+                    "notes": "bs=2,seqlen=8192 - SDK 2.19.1",
+                },
+            ],
+        },
         {
             "instanceType": "p4d.24xlarge",
             "quantization": "none",