kernels-ext-npu

Kernel sources for https://huggingface.co/kernels-ext-npu

介绍 kernels-npu

参考文档

从源码中构建算子

参考文档

当前支持

kernel	build from	category
flash-attn2-npu	source code	torch_npu
rmsnorm	source code	torch_npu
rmsnorm	source code	triton-ascend

其他说明

1. 目前，NPU 的主要优化手段仍是 torch_npu， triton-ascend 的优化 kernel 仅供参考。
2. triton-ascend 的安装可参考文档。