PR #46268 Add some AITER kernel routing for ROCm

Add some AITER kernel routing for ROCm #46268

vasqu merged 4 commits into huggingface:main from Abdennacer-Badaoui:some-rocm-kernels

Abdennacer-Badaoui marked this pull request as draft 45 days ago

Abdennacer-Badaoui force pushed from 15efed55 to a43d58f5 40 days ago

Abdennacer-Badaoui requested a review from

vasqu 38 days ago

Abdennacer-Badaoui marked this pull request as ready for review 38 days ago

vasqu commented on 2026-06-04

Abdennacer-Badaoui force pushed from b8bf8f11 to 857dae59 38 days ago

vasqu approved these changes on 2026-06-04

use kernels-community/megablocks for rocm

2b7a7e86

add aiter-rope rocm mapping

59ac2bf3

add aiter-flash-attn rocm fallback for FA3

71f8ab0d

resolve_kernel_device_type for rocm

8ad82c8a

Abdennacer-Badaoui force pushed from 5ba56ad2 to 8ad82c8a 25 days ago

vasqu approved these changes on 2026-06-17

vasqu enabled auto-merge 25 days ago

vasqu merged 497d83c5 into main 25 days ago

Reviewers

vasqu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone