尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
CANN AscendC FMA临时缓冲区因子大小获取
GetFmaTmpBufferFactorSize【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明该接口用于获取maxLivedNodeCount和extraBuf在固定空间大小的情况下通过maxLivedNodeCount和extraBuf可以推算算子单次最大计算元素数量。maxLivedNodeCount表示临时空间是单次计算数据量所占空间的多少倍extraBuf表示使用的额外临时空间大小。推算示例如下算子实现需要调用Fma接口开发者为其预留currBuff大小的空间利用GetFmaTmpBufferFactorSize接口得到maxLivedNodeCount、extraBuf输出值可推导算子单次最大计算元素数量为currentShapeSize (currBuff - extraBuf) / maxLivedNodeCount / typeSize算子实现需要调用两个kernel侧API KernelIntf1、KernelIntf2利用两个GetXxxTmpBufferFactorSize其中Xxx为需要调用的两个高阶API接口的两组输出值(maxLivedNodeCount、extraBuf)以及当前现有的临时空间推导单次最大计算元素数量currentShapeSize为currentShapeSize1 (currBuff - extraBuf1) / maxLivedNodeCount1 / typeSizecurrentShapeSize2 (currBuff - extraBuf2) / maxLivedNodeCount2 / typeSizecurrentShapeSize min(currentShapeSize1, currentShapeSize2)注意上文中的currBuff表示接口计算可用的空间需要去除用户输入输出等空间另外接口获取的maxLivedNodeCount值可能为0计算时需要判断该值非0避免除零错误。函数原型void GetFmaTmpBufferFactorSize(const platform_ascendc::PlatformAscendC ascendcPlatform, const uint32_t typeSize, uint32_t maxLivedNodeCount, uint32_t extraBuf)参数说明表1参数列表参数名输入/输出功能ascendcPlatform输入输入的平台信息。PlatformAscendC的定义请参见构造及析构函数。typeSize输入输入的数据类型大小单位为字节。比如输入的数据类型为half此处应传入2。maxLivedNodeCount输出最大存活节点数表示临时空间是单次计算数据量所占空间的多少倍。extraBuf输出使用的额外临时空间大小单位为字节。返回值说明无约束说明当利用maxLivedNodeCount、extraBuf反推出的currentShapeSize * typeSize 256B时currentShapeSize按照256B/typeSize的值向上取整。调用示例完整的调用样例请参考更多样例。uint32_t maxLivedNodeCount 0; uint32_t extraBuf 0; auto plat platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); AscendC::GetFmaTmpBufferFactorSize(plat, typeSize, maxLivedNodeCount, extraBuf);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED

相关推荐

AI编程工具横评:Cursor、Claude Code与Copilot在真实开发流中的协同逻辑

AI编程工具横评:Cursor、Claude Code与Copilot在真实开发流中的协同逻辑

1. 这场横评不是选“最聪明”的AI,而是找“最懂你写代码节奏”的搭档 2026年春天,我清空了IDE里所有AI编程插件,重新装上Cursor、Claude Code和GitHub Copilot Pro,用同一套真实项目——一个基于Spring Boot 3.3 React 19的供应链…

📅 2026/10/7 13:11:37
CANN/asc-devkit Exp指数函数接口

CANN/asc-devkit Exp指数函数接口

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

📅 2026/10/4 19:57:05
AscendC Erfc临时空间接口

AscendC Erfc临时空间接口

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

📅 2026/9/11 6:27:37
MORE NEWS

更多资讯

📰

工地太忙根本没时间复习考试?真伪电工证官方报名入口全解析

工地太忙根本没时间复习考试?真伪电工证官方报名入口全解析 为什么工地电工证越来越难考? 你是不是也遇到过这种情况,工地项目一忙,根本没有时间复习电工证考试?其实这不是个例,很多电工都因为工作节奏快、培训时间紧而错过考试。而如今,…

📰

金州区电工证怎么考?良心建议别踩这些坑

金州区电工证怎么考?良心建议别踩这些坑 你现在急着找工作,但手上还没个 金州区电工证 ,根本进不了工地、工厂、水电安装公司这些地方。别人都在忙面试,你却在为这个证发愁,这事儿真不是小事。听我一句 良心建议…

📰

香港认可内地电工证吗保姆级教程:外省电工证转过来全攻略

香港认可内地电工证吗保姆级教程:外省电工证转过来全攻略 之前考的证在外省能不能转过来?这是很多电工从业者心头的大石头,特别是像你这样在信阳浉河区从事市政公用工程的,常常会遇到证书转接、资格认证、跨省使用等问题。尤其是想在港企或跨区域项目中工…

📰

金州区电工证值不值得考?怕考不过白交培训费的看过来

金州区电工证值不值得考?怕考不过白交培训费的看过来 怕考不过白交培训费?金州区电工证值不值得考,这可能是很多人在报名前的第一反应。电工证虽然看起来只是个资格证明,但背后关系到从业门槛、薪资水平和职业发展。尤其是在当前电力行业日益规范的背景下…

📰

电工证复审算上班吗?考前押题帮你顺利通过复审

电工证复审算上班吗?考前押题帮你顺利通过复审 证书过期了不知道怎么复审延期?最近一位来自平顶山的电工朋友小李,就遇到了这样的难题。他原本以为电工证复审和上班无关,结果在办理过程中发现,复审不仅需要时间,还涉及到单位配合、考试内容和流程,让他…

📰

工地太忙根本没时间复习考试?2026河南最新政策下电工证理论不及格怎么办

工地太忙根本没时间复习考试?2026河南最新政策下电工证理论不及格怎么办 工地太忙根本没时间复习考试?很多人在考电工证的时候,都遇到过理论考试不及格的烦恼。尤其是2026年河南最新政策出台后,电工证的考试难度和要求都有所变化,如果还是按照老…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

读完文章,想聊聊您的网站?

告诉我们您的行业与需求,资深顾问一对一梳理方案与报价,全程免费。

📞 💬