lightllm icon indicating copy to clipboard operation
lightllm copied to clipboard

weight only int4 is slower than cutlass int4

Open zhoutianzi666 opened this issue 1 year ago • 1 comments

https://github.com/ModelTC/lightllm/blob/main/lightllm/common/basemodel/triton_kernel/dequantize_gemm_int4.py

The algorithm in the above file implements weight only int4, but its speed is only 50% of cutpass int4. How can this be resolved?

zhoutianzi666 avatar Mar 19 '24 05:03 zhoutianzi666

You can compile your own operator interface implemented with pybind, and accelerate the inference by modifying the source code to replace the implementation used during inference.

hiworldwzj avatar Mar 19 '24 06:03 hiworldwzj