1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-03 05:58:09 +00:00
ColossalAI/tests/test_smoothquant
Xu Kai 611a5a80ca
[inference] Add smmoothquant for llama ()
* [inference] add int8 rotary embedding kernel for smoothquant ()

* [inference] add smoothquant llama attention ()

* add smoothquant llama attention

* remove uselss code

* remove useless code

* fix import error

* rename file name

* [inference] add silu linear fusion for smoothquant llama mlp  ()

* add silu linear

* update skip condition

* catch smoothquant cuda lib exception

* prcocess exception for tests

* [inference] add llama mlp for smoothquant ()

* add llama mlp for smoothquant

* fix down out scale

* remove duplicate lines

* add llama mlp check

* delete useless code

* [inference] add smoothquant llama ()

* add smoothquant llama

* fix attention accuracy

* fix accuracy

* add kv cache and save pretrained

* refactor example

* delete smooth

* refactor code

* [inference] add smooth function and delete useless code for smoothquant ()

* add smooth function and delete useless code

* update datasets

* remove duplicate import

* delete useless file

* refactor codes ()

* rafactor code

* add license

* add torch-int and smoothquant license
2023-10-16 11:28:44 +08:00
..
test_llama_attention.py [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00
test_llama_mlp.py [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00
test_smoothquant_linear.py [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00
test_sq_rotary_embedding.py [inference] Add smmoothquant for llama () 2023-10-16 11:28:44 +08:00