察元AI教程库 · 仓库 · 部署

cider

W8A8/W4A8推理 + 优化的SDPA在Apple Silicon上运行——解锁M5中未使用的INT8张量运算,实现1.2–1.9倍更快的LLM预填充,以及受FlashInfer启发的GQA解码注意力,实现最多1.6倍的SDPA加速,作为MLX自定义原语构建。

作者:Mininglamp-AI · 语言:en

前往原址学习

简介

W8A8/W4A8推理 + 优化的SDPA在Apple Silicon上运行——解锁M5中未使用的INT8张量运算,实现1.2–1.9倍更快的LLM预填充,以及受FlashInfer启发的GQA解码注意力,实现最多1.6倍的SDPA加速,作为MLX自定义原语构建。

返回教程库 · 察元AI产品家族