Skip to content

add support of sparse:#350

Open
STwangyingrui wants to merge 4 commits into
thu-ml:mainfrom
ModelTC:yr/sparse
Open

add support of sparse:#350
STwangyingrui wants to merge 4 commits into
thu-ml:mainfrom
ModelTC:yr/sparse

Conversation

@STwangyingrui

Copy link
Copy Markdown
  1. add core kernel of load_sparse() and mma_sparse()
  2. remove the redundant data type conversion of delta_s from fp32 to bf16
  3. optimize performance of prepare_qkv, merging pad_128 to quant kernel
  4. small adjustment of qm, moving substract-mean before padding.

1. add core kernel of load_sparse() and mma_sparse()
2. remove the redundant data type conversion of delta_s from fp32 to bf16
3. optimize performance of prepare_qkv, merging pad_128 to quant kernel
4. small adjustment of qm, moving substract-mean before padding.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant