Enable more tests

pytorch-labs · Dec 4, 2024 · fa27947 · fa27947
1 parent d0e2f76
commit fa27947
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 13 deletions.
diff --git a/test/test_gpu/skip_tests_h100_pytorch.yaml b/test/test_gpu/skip_tests_h100_pytorch.yaml
@@ -19,10 +19,6 @@ fp8_attention:
 # fp8_fused_quant_gemm_rowwise requires fb-only kernels
 fp8_fused_quant_gemm_rowwise:
 gemm:
-  # FIXME: out of shared memory
-  - triton_tma_persistent_matmul
-  # FIXME: out of shared memory
-  - triton_tma_persistent_cached_matmul
   # internal only kernels
   - hstu_triton_matmul
   - colfax_cutlass_matmul

diff --git a/test/test_gpu/skip_tests_h100_triton_main.yaml b/test/test_gpu/skip_tests_h100_triton_main.yaml
@@ -16,16 +16,7 @@ fp8_attention:
   - colfax_fmha
 # fb-only kernels
 fp8_fused_quant_gemm_rowwise:
-fp8_gemm:
-  # FIXME: out of shared memory
-  - triton_persistent_fp8_gemm
-  # FIXME: out of shared memory
-  - triton_tma_persistent_fp8_gemm
 gemm:
-  # FIXME: out of shared memory
-  - triton_tma_persistent_matmul
-  # FIXME: out of shared memory
-  - triton_tma_persistent_cached_matmul
   # internal only kernels
   - hstu_triton_matmul
   - colfax_cutlass_matmul