Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Interleaved MHA for CPU path (#17138) (#17211)
Browse files Browse the repository at this point in the history 
* init qk attention

* qk: fake backward

* cpu selfatt and encdec; move tests to test_operator.py

* coding style

* fix lint

* use random seed in tests

* remove ut in test_operator_gpu.py

* coding style

* retrigger ci

Co-authored-by: Tao Lv <tao.a.lv@intel.com>
  • Loading branch information
@eric-haibin-lin @TaoLv @ptrendx
2 people authored and ptrendx committed Jan 4, 2020
1 parent dafbb11 commit a0fcb9a
Show file tree
Hide file tree
Showing 3 changed files with 861 additions and 327 deletions.
Loading

0 comments on commit a0fcb9a

Please sign in to comment.