From 1f2d7a1b1c39e0511b38c51da7085096fe791eaa Mon Sep 17 00:00:00 2001
From: Isaac Vidas <tzahi.vidas@gmail.com>
Date: Wed, 13 Nov 2024 06:55:36 -0500
Subject: [PATCH] Add rtx 4080 super to chips dictionary (#20285)

---
 src/lightning/fabric/utilities/throughput.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/lightning/fabric/utilities/throughput.py b/src/lightning/fabric/utilities/throughput.py
index 6743da7b34..598a32228a 100644
--- a/src/lightning/fabric/utilities/throughput.py
+++ b/src/lightning/fabric/utilities/throughput.py
@@ -347,6 +347,14 @@ _CUDA_FLOPS: Dict[str, Dict[Union[str, torch.dtype], float]] = {
         torch.int8: 389.9e12,
         "int4": 779.8e12,
     },
+    "rtx 4080 super": {
+        torch.float32: 52.2e12,
+        "tfloat32": 52.2e12,
+        torch.bfloat16: 52.2e12,
+        torch.float16: 52.2e12,
+        torch.int8: 417.6e12,
+        "int4": 835.2e12,
+    },
     "l4": {
         torch.float32: 30.3e12,
         "tfloat32": 60e12,