Skip to content

Enable SLL tests on ROCm - #6325

Open
avbokovoy wants to merge 2 commits into
pytorch:mainfrom
ROCm:abokovoi/enable-sll-tests-on-rocm
Open

avbokovoy wants to merge 2 commits into
pytorch:mainfrom
ROCm:abokovoi/enable-sll-tests-on-rocm

Conversation

@avbokovoy

Copy link
Copy Markdown
Contributor

Summary

This PR removes the ROCm skip guards from 10 SLL Triton test files, enabling 18 previously skipped tests. Four files were skipped at class level (jagged_jagged_bmm, jagged_softmax, jagged_dense_bmm, jagged_dense_elementwise_mul_jagged_out) and six at method level (jagged_self_substraction_jagged_out, jagged_dense_flash_attention, jagged_flash_attention_basic, jagged2_to_padded_dense, dense_jagged_cat_jagged_out, array_jagged_bmm_jagged_out). Each change drops the skipIf decorator and the now unused running_on_rocm import. It also removes a stale entry from the ROCm ignore list in the CI test script.

Motivation

The guards carried only the generic reason string from test_utils.py, "Test currently doesn't work on the ROCm stack", with no per suite rationale. These are Triton kernels and are expected to be hardware portable, so the guards appear to be stale rather than load bearing.

@meta-codesync

meta-codesync Bot commented Sep 18, 2026

Copy link
Copy Markdown
Contributor

@q10 has imported this pull request. If you are a Meta employee, you can view this in D120726948.

This branch has not been deployed

No deployments
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant