Commit 9f6178e

authored

iakv: change attn_weights layout (#3248)

1 parent 97a5c4a commit 9f6178eCopy full SHA for 9f6178e

2 files changed

+818

-259

lines changed

csrc/cpu
- aten/kernels
  - MaskedMultiHeadAttentionKrnl.cpp
- vec/vec512/perf_kernel
  - add_softmax.h

+818

-259

lines changed

Comments

(0)