Add rtx 4080 super to chips dictionary (#20285)

2024-11-13 06:55:36 -05:00 · 2024-11-13 06:55:36 -05:00 · 1f2d7a1b1c
parent 9358898c6e
commit 1f2d7a1b1c
1 changed files with 8 additions and 0 deletions
--- 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,