onnxruntime
Implement reuse past and present tensor in Attention Ops.
#13791
Merged

Implement reuse past and present tensor in Attention Ops. #13791

zhanghuanrong
zhangleihuanrong Implement reuse kv_cache for past and present tensor in Attention Ops.
38566baf
zhanghuanrong zhanghuanrong requested a review from tianleiwu tianleiwu 3 years ago
zhanghuanrong zhanghuanrong requested a review from gh-yewang gh-yewang 3 years ago
zhanghuanrong Python file format
4e593997
tianleiwu
tianleiwu commented on 2022-12-01
tianleiwu
tianleiwu commented on 2022-12-01
tianleiwu
tianleiwu commented on 2022-12-01
zhangleihuanrong Fix vocab_size not set issue which cause random crash.
f6ec1d78
gh-yewang
gh-yewang commented on 2022-12-05
yufenglee
yufenglee commented on 2022-12-05
yufenglee
yufenglee commented on 2022-12-05
yufenglee
yufenglee commented on 2022-12-06
zhangleihuanrong Merge branch 'main' into zhalei/attention_reuse_past_present
2cb38017
zhanghuanrong according pr feedbacks.
9659987c
zhanghuanrong correct some on the LaunchAddBiasTransAppendKvToPresent
3a52f9fb
yufenglee
yufenglee commented on 2022-12-15
yufenglee
yufenglee commented on 2022-12-15
zhanghuanrong support past present share buffer for attention when bias is nullptr.
df2dd1b1
yufenglee
yufenglee commented on 2022-12-15
zhanghuanrong Merge branch 'main' into zhalei/attention_reuse_past_present
eb4b73c5
yufenglee
yufenglee dismissed these changes on 2022-12-15
zhanghuanrong Some condition check
aa9cca0e
zhanghuanrong zhanghuanrong dismissed their stale review via aa9cca0e 3 years ago
yufenglee
yufenglee dismissed these changes on 2022-12-15
yufenglee
yufenglee commented on 2022-12-15
zhanghuanrong optimized matrix number on qkv
ee8b9d71
zhanghuanrong zhanghuanrong dismissed their stale review via ee8b9d71 3 years ago
zhanghuanrong Fix CI build issues
05ffbf90
zhanghuanrong another python lint error
89ee6753
yufenglee
yufenglee dismissed these changes on 2022-12-16
zhanghuanrong Make past_present_share_buffer change do put into result model.
a7878f45
zhanghuanrong zhanghuanrong dismissed their stale review via a7878f45 3 years ago
zhanghuanrong fix prefast warnings.
624afa02
zhanghuanrong update op docs.
acbd0c99
yufenglee
yufenglee dismissed these changes on 2022-12-17
zhanghuanrong Using CI pipelines doc gen, local one seems not good.
eb2dc908
zhanghuanrong zhanghuanrong dismissed their stale review via eb2dc908 3 years ago
hariharans29
hariharans29 approved these changes on 2022-12-17
zhanghuanrong zhanghuanrong merged fba09faf into main 3 years ago
zhanghuanrong zhanghuanrong deleted the zhalei/attention_reuse_past_present branch 3 years ago
tianleiwu
tianleiwu commented on 2022-12-21
tianleiwu
tianleiwu commented on 2022-12-21

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone