Simulates torch+cuda12.0 and a representative Ampere-class capability.
()
| 10 | |
| 11 | @pytest.fixture |
| 12 | def cuda120_spec() -> CUDASpecs: |
| 13 | """Simulates torch+cuda12.0 and a representative Ampere-class capability.""" |
| 14 | return CUDASpecs( |
| 15 | cuda_version_string="120", |
| 16 | highest_compute_capability=(8, 6), |
| 17 | cuda_version_tuple=(12, 0), |
| 18 | ) |
| 19 | |
| 20 | |
| 21 | @pytest.fixture |