|
16 | 16 |
|
17 | 17 |
|
18 | 18 | @pytest.mark.skipif( |
19 | | - current_platform.has_device_capability(100), |
20 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 19 | + not current_platform.has_device_capability(90), |
| 20 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
21 | 21 | ) |
22 | 22 | @pytest.mark.skipif( |
23 | 23 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
@@ -71,8 +71,8 @@ def test_rms_norm_batch_invariant_vs_standard( |
71 | 71 |
|
72 | 72 |
|
73 | 73 | @pytest.mark.skipif( |
74 | | - current_platform.has_device_capability(100), |
75 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 74 | + not current_platform.has_device_capability(90), |
| 75 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
76 | 76 | ) |
77 | 77 | @pytest.mark.skipif( |
78 | 78 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
@@ -119,8 +119,8 @@ def test_rms_norm_3d_input(batch_size: int, seq_len: int, hidden_size: int): |
119 | 119 |
|
120 | 120 |
|
121 | 121 | @pytest.mark.skipif( |
122 | | - current_platform.has_device_capability(100), |
123 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 122 | + not current_platform.has_device_capability(90), |
| 123 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
124 | 124 | ) |
125 | 125 | @pytest.mark.skipif( |
126 | 126 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
@@ -185,8 +185,8 @@ def test_rms_norm_numerical_stability(): |
185 | 185 |
|
186 | 186 |
|
187 | 187 | @pytest.mark.skipif( |
188 | | - current_platform.has_device_capability(100), |
189 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 188 | + not current_platform.has_device_capability(90), |
| 189 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
190 | 190 | ) |
191 | 191 | @pytest.mark.skipif( |
192 | 192 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
@@ -224,8 +224,8 @@ def test_rms_norm_formula(): |
224 | 224 |
|
225 | 225 |
|
226 | 226 | @pytest.mark.skipif( |
227 | | - current_platform.has_device_capability(100), |
228 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 227 | + not current_platform.has_device_capability(90), |
| 228 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
229 | 229 | ) |
230 | 230 | @pytest.mark.skipif( |
231 | 231 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
@@ -268,8 +268,8 @@ def test_rms_norm_different_hidden_sizes(hidden_size: int): |
268 | 268 |
|
269 | 269 |
|
270 | 270 | @pytest.mark.skipif( |
271 | | - current_platform.has_device_capability(100), |
272 | | - reason="Batch invariance tests not yet supported on B200 (SM100)", |
| 271 | + not current_platform.has_device_capability(90), |
| 272 | + reason="Batch invariance tests only supported on Hopper (SM90)", |
273 | 273 | ) |
274 | 274 | @pytest.mark.skipif( |
275 | 275 | not torch.cuda.is_available(), reason="Requires CUDA for RMS norm kernels" |
|
0 commit comments