| 88 | |
| 89 | @dataclass |
| 90 | class GPUSpec: |
| 91 | name: str = "Unknown" |
| 92 | sm_count: int = 0 |
| 93 | memory_gb: float = 0.0 |
| 94 | peak_tflops_fp16: float = 0.0 |
| 95 | peak_tflops_bf16: float = 0.0 |
| 96 | peak_tflops_fp32: float = 0.0 |
| 97 | peak_bandwidth_gb_s: float = 0.0 |
| 98 | l2_cache_mb: float = 0.0 |
| 99 | compute_capability: Tuple[int, int] = (0, 0) |
| 100 | |
| 101 | |
| 102 | # Known GPU database: name_fragment -> (peak_fp16_tflops, peak_bandwidth_gb_s, l2_cache_mb) |