打赏

相关文章

CANN/pyasc矩阵乘法N批处理迭代

asc.language.adv.Matmul.iterate_n_batch 【免费下载链接】pyasc 本项目为Python用户提供算子编程接口,支持在昇腾AI处理器上加速计算,接口与Ascend C一一对应并遵守Python原生语法。 项目地址: https://gitcode.com/cann/pyasc Matmul.iterate_…

华为CANN TensorFlow AllGather算子

allgather 【免费下载链接】tensorflow Ascend TensorFlow Adapter 项目地址: https://gitcode.com/cann/tensorflow 功能说明 集合通信算子AllGather的操作接口,将通信域内所有节点的输入按照rank id重新排序,然后拼接起来,再将结果…

CANN/ops-cv加法算子示例

AddExample 【免费下载链接】ops-cv 本项目是CANN提供的图像处理、目标检测相关的算子库,实现网络在NPU上加速计算。 项目地址: https://gitcode.com/cann/ops-cv 产品支持情况 产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列…

CATLASS FlashAttention推理示例

FlashAttentionInfer Example Readme 【免费下载链接】catlass 本项目是CANN的算子模板库,提供NPU上高性能矩阵乘及其相关融合类算子模板样例。 项目地址: https://gitcode.com/cann/catlass 代码组织 ├── 23_flash_attention_infer │ ├── CMakeLi…

手机版浏览

扫一扫体验

微信公众账号

微信扫一扫加关注

返回
顶部