Skip to content

[ROCm] Jax Add softmax sink (learnable off-by-one) support for the ROCm/CK fused attention backend - #678

Open
shurale-nkn wants to merge 12 commits into
devfrom
sink_support_ck
Open

[ROCm] Jax Add softmax sink (learnable off-by-one) support for the ROCm/CK fused attention backend#678
shurale-nkn wants to merge 12 commits into
devfrom
sink_support_ck

CI fix. Fixed hlo check for Distributed test

17cec09
Select commit
Loading
Failed to load commit list.

Select a check to view from the sidebar