trunk/610fee8d845047496bd0713e7adff4d79d08f137: [ROCm] Fix CK SDPA flash attention on gfx1200/gfx1201 (RDNA4) (#188114)
- PyTorch: 180 events in the last 90 days
- PyTorch: 178th Release in the last 90 days
- Previous: earlier the same day · trunk/29343e76e2342c777bb257a270bae1d805996700: Support non-builtin constexpr objects in user Triton kernels (#189692)
What happened
The CK Tile FMHA generator already compiles working kernels for gfx12 (--targets=gfx1201 passes through to the kernel template), but two guards prevented them from being used on RDNA4 hardware: launch_kernel_pt.hpp: kentry_pt kernel entry stubs added in PR #144777 ([ROCm] CK SDPA - Move arch check to CK patch) had a hard-coded whitelist of gfx90a/gfx942/gfx950 (updated for gfx950 in PR #155103 ). gfx1200/gfx1201 wer…
Summary assembled by rule from the sources below