Skip to content

feat: run rocm kernel tests on the amd mi300 runners - #1313

Merged
drbh merged 1 commit into
mainfrom
rocm-mi300-ci-tests
Nov 11, 2026
Merged

drbh merged 1 commit into
mainfrom
rocm-mi300-ci-tests

Conversation

@drbh

@drbh drbh commented Nov 9, 2026 •

Copy link
Copy Markdown
Collaborator

this pr adds a test-kernel-rocm job that runs the kernels_ci tests on the amd mi300 runners used by the transformers amd ci https://github.com/huggingface/transformers/blob/main/.github/workflows/self-comment-ci-amd.yml

its split out like the stable abi jobs so it cant block the cuda tests. .github/scripts/rocm_ci.py picks the rocm variant and skips kernels that dont target gfx942

note: build.yaml needs access to the hfc-amd-mi300-1gpu runner group before this can run

8 tests pass in .github/scripts/test_rocm_ci.py

Signed-off-by: David Holtz <david.richard.holtz@gmail.com>
@drbh

drbh commented Nov 9, 2026

Copy link
Copy Markdown
Collaborator Author

pending support for the runner from the infra team - once added I'll move the pr out of draft

@drbh
drbh marked this pull request as ready for review November 10, 2026 15:43
@drbh
drbh merged commit b7893e4 into main Nov 11, 2026
6 checks passed
@drbh
drbh deleted the rocm-mi300-ci-tests branch November 11, 2026 18:50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants