Skip to content

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

Open
atassis wants to merge 2 commits 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 2 commits into
amd:develfrom
atassis:fix/mha-gqa-kv-arg-spec

mha: tighten comments

ca9d83d
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs