Skip to content

mha: size the K/V arg specs by num_KV_heads, not num_heads - #169

Open
atassis wants to merge 1 commit into
amd:develfrom
atassis:fix/mha-gqa-kv-arg-spec
Open

mha: size the K/V arg specs by num_KV_heads, not num_heads#169
atassis wants to merge 1 commit into
amd:develfrom
atassis:fix/mha-gqa-kv-arg-spec

mha: size the K/V arg specs by num_KV_heads, not num_heads

8726f85
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs