intel / torch-xpu-ops

Apache License 2.0
30 stars 21 forks source link

[PT2.6] LLM customized kernels (GEMM INT4 serves for Torchao) #1005

Open riverliuintel opened 1 month ago

riverliuintel commented 1 month ago

🚀 The feature, motivation and pitch

Integrate oneDNN GEMM INT4 kernels, and serves for Torchao LLM usage. It requires pass UT and example workloads usage.

Alternatives

No response

Additional context

No response