Enable more operators.

pytorch-labs · Nov 19, 2024 · 34aeb9e · 34aeb9e
1 parent a6da2a3
commit 34aeb9e
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/test/test_gpu/skip_tests_h100_pytorch.yaml b/test/test_gpu/skip_tests_h100_pytorch.yaml
@@ -29,9 +29,6 @@ jagged_layer_norm:
 jagged_mean:
 jagged_softmax:
 jagged_sum:
-layer_norm:
-low_mem_dropout:
-rms_norm:
-rope:
-template_attention:
+ragged_attention:
+  - hstu_triton_ragged_attention_persistent
 test_op: