news 2026/7/17 8:23:34

CANN Ascend C矩阵计算SetFixSplit函数

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
CANN Ascend C矩阵计算SetFixSplit函数

SetFixSplit

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

功能说明

设置固定的baseM、baseN、baseK,单位为元素个数。

函数原型

int32_t SetFixSplit(int32_t baseMIn = -1, int32_t baseNIn = -1, int32_t baseKIn = -1)

参数说明

表1参数说明

参数名输入/输出描述
baseMIn输入设置固定的baseM,默认值为-1,表示不设置固定baseM,由tiling函数进行计算。
baseNIn输入设置固定的baseN,默认值为-1,表示不设置固定baseN,由tiling函数进行计算。
baseKIn输入当前仅支持取值为-1,暂不支持设置其它值。

返回值说明

-1表示设置失败;0表示设置成功。

约束说明

  • baseM*baseN个输出元素所占的存储空间大小不能超过L0C Buffer大小,即baseM * baseN * sizeof(C_TYPE) <= L0CSize。
  • baseM需要小于等于singleM按16个元素向上对齐后的值(如ceil(singleM/16)*16),baseN需要小于等于singleN以C0_size个元素向上对齐的值,其中singleM为单核内M轴长度,singleN为单核内N轴长度,half/bfloat16_t数据类型的C0_size为16,float数据类型的C0_size为8,int8_t数据类型的C0_size为32,int4b_t数据类型的C0_size为64。例如singleM=12,则baseM需要小于等于16,同时baseM需要满足分形对齐的要求,所以baseM只能取16;如果baseM取其他超过16的值,获取Tiling将失败。

调用示例

auto ascendcPlatform = platform_ascendc::PlatformAscendC(context->GetPlatformInfo()); matmul_tiling::MatmulApiTiling tiling(ascendcPlatform); tiling.SetFixSplit(16, 16, -1); // 设置固定的baseM、baseN

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/7/17 8:22:26

广东广氟新材料|高品质 PTFE 系列功能膜材应用指南

在高端电子、通讯、精密制造、防护封装等产业快速迭代的当下&#xff0c;高性能膜材成为决定产品稳定性、使用寿命与核心性能的关键基础材料。PTFE聚四氟乙烯膜凭借优异的介电性能、耐高低温、耐腐蚀、低摩擦及高稳定性等特性&#xff0c;被广泛应用于各类高精尖工业场景&#…

作者头像 李华
网站建设 2026/7/17 8:21:45

3步解决网易云音乐NCM格式限制的终极方案

3步解决网易云音乐NCM格式限制的终极方案 【免费下载链接】ncmdump 项目地址: https://gitcode.com/gh_mirrors/ncmd/ncmdump 你是否曾经为心爱的音乐只能在网易云音乐客户端播放而烦恼&#xff1f;精心收藏的歌曲在车载音响、专业播放器或智能音箱上变成了一堆无法识别…

作者头像 李华
网站建设 2026/7/17 8:21:45

Ascend C Warp函数简介

Warp函数 【免费下载链接】asc-devkit 本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言&#xff0c;原生支持C和C标准规范&#xff0c;主要由类库和语言扩展层构成&#xff0c;提供多层级API&#xff0c;满足多维场景算子开发诉求。 项目地址: https://gitcode.com/c…

作者头像 李华
网站建设 2026/7/17 8:20:59

CW32开发板光敏电阻光照检测实践指南

1. CW32饭盒派开发板与光照检测入门 第一次拿到CW32饭盒派开发板时&#xff0c;这块蓝色的小板子确实让我想起了学生时代的便当盒。作为武汉芯源半导体推出的大学计划开发板&#xff0c;它搭载了CW32F030系列MCU&#xff0c;板载资源丰富得让人惊喜——从基本的GPIO、ADC到UART…

作者头像 李华
网站建设 2026/7/17 8:18:05

Ubuntu命令行操作全指南:从基础到高级实战

1. Ubuntu命令操作体系全景解析 作为Linux发行版中最受欢迎的桌面系统之一&#xff0c;Ubuntu的命令行操作是每位用户必须掌握的生存技能。不同于Windows的图形化操作&#xff0c;命令行提供了更高效、更灵活的系统控制方式。我将从实际应用场景出发&#xff0c;系统性地梳理Ub…

作者头像 李华
网站建设 2026/7/17 8:16:50

大模型API输出优化:指令工程与参数调优实战指南

1. 大模型API输出优化的核心挑战当你第一次调用大模型API时&#xff0c;大概率会遇到这样的场景&#xff1a;精心设计的prompt投喂进去&#xff0c;返回的却是驴唇不对马嘴的结果。这就像点了一份牛排却收到一盘西红柿炒蛋——不是厨师不行&#xff0c;而是点单方式出了问题。大…

作者头像 李华