iree-org / iree-org/wave

Core kernels to implement for Wave

Open
#63 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
59
Forks
32
PR merge metrics
No merged PRs in 30d

Description

Functionality
- [ ] Int8 Group-quantized linear / Dequant-GEMM
- [ ] Int4 Group-quantized linear / Dequant-GEMM
- [ ] MoE align block
- [ ] MLA ([link](https://github.com/ROCm/aiter/blob/main/aiter/mla.py))

Need polishing
- [ ] Flash attention backwards - Need to improve performance
- [ ] Quant Linear - Need to improve accuracy

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.