From 645899386f0ed6d6d3fd51c934f51f1b42caf307 Mon Sep 17 00:00:00 2001 From: Isaac Vidas Date: Tue, 17 Sep 2024 13:42:39 -0400 Subject: [PATCH] Add rtx 4080 super to chips dictionary --- 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 6743da7b34085..598a32228ac75 100644 --- a/src/lightning/fabric/utilities/throughput.py +++ b/src/lightning/fabric/utilities/throughput.py @@ -347,6 +347,14 @@ def measure_flops( 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,