gary-boon
Make QKV hook robust against shape mismatches
343dd57