Disable jagged tests for speed

pytorch-labs · Nov 19, 2024 · 7e177c6 · 7e177c6
1 parent e8a588e
commit 7e177c6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/test/test_gpu/skip_tests_h100_pytorch.yaml b/test/test_gpu/skip_tests_h100_pytorch.yaml
@@ -39,6 +39,11 @@ gemm:
   - colfax_cutlass_matmul
   # FIXME: PT2 CUTLASS backend failed
   - pt2_cutlass_matmul
+# jagged tests are slow, so disable them in OSS
+jagged_layer_norm:
+jagged_mean:
+jagged_softmax:
+jagged_sum:
 ragged_attention:
   - hstu_triton_ragged_attention_persistent
 test_op: