| 75 | |
| 76 | @dataclass |
| 77 | class GPUSpec: |
| 78 | name: str = "Unknown" |
| 79 | sm_count: int = 0 |
| 80 | memory_gb: float = 0.0 |
| 81 | peak_tflops_fp16: float = 0.0 |
| 82 | peak_tflops_bf16: float = 0.0 |
| 83 | peak_tflops_fp32: float = 0.0 |
| 84 | peak_bandwidth_gb_s: float = 0.0 |
| 85 | l2_cache_mb: float = 0.0 |
| 86 | compute_capability: Tuple[int, int] = (0, 0) |
| 87 | |
| 88 | |
| 89 | def _fallback_detect_gpu() -> GPUSpec: |
no outgoing calls
no test coverage detected