Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add rtx 4080 super to chips dictionary (#20285)

This commit is contained in:
Isaac Vidas 2024-11-13 06:55:36 -05:00 committed by GitHub
parent 9358898c6e
commit 1f2d7a1b1c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/lightning/fabric/utilities/throughput.py
View File

@ -347,6 +347,14 @@ _CUDA_FLOPS: Dict[str, Dict[Union[str, torch.dtype], float]] = {
torch.int8: 389.9e12, torch.int8: 389.9e12,
"int4": 779.8e12, "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": { "l4": {
torch.float32: 30.3e12, torch.float32: 30.3e12,
"tfloat32": 60e12, "tfloat32": 60e12,