CANN/asc-devkit: SetALayout矩阵布局设置

发布时间:2026/11/3 17:00:11

CANN/asc-devkit: SetALayout矩阵布局设置 SetALayout【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置A矩阵的Layout轴信息包括B、S、N、G、D轴。对于BSNGD、SBNGD、BNGS1S2 Layout格式调用IterateBatch接口之前需要在Host侧Tiling实现中通过本接口设置A矩阵的Layout轴信息。函数原型int32_t SetALayout(int32_t b, int32_t s, int32_t n, int32_t g, int32_t d)参数说明表1参数说明参数名输入/输出描述b输入A矩阵Layout的B轴信息s输入A矩阵Layout的S轴信息n输入A矩阵Layout的N轴信息g输入A矩阵Layout的G轴信息d输入A矩阵Layout的D轴信息返回值说明-1表示设置失败 0表示设置成功。约束说明对于BSNGD、SBNGD、BNGS1S2 Layout格式调用IterateBatch接口之前需要在Host侧Tiling实现中通过本接口设置A矩阵的Layout轴信息。调用示例auto ascendcPlatform platform_ascendc::PlatformAscendC(context-GetPlatformInfo()); matmul_tiling::MultiCoreMatmulTiling tiling(ascendcPlatform); int32_t M 32; int32_t N 256; int32_t K 64; tiling.SetDim(1); tiling.SetAType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); tiling.SetBType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT16); tiling.SetCType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT); tiling.SetBiasType(matmul_tiling::TPosition::GM, matmul_tiling::CubeFormat::ND, matmul_tiling::DataType::DT_FLOAT); tiling.SetShape(M, N, K); tiling.SetOrgShape(M, N, K); tiling.SetBias(true); tiling.SetBufferSpace(-1, -1, -1); constexpr int32_t A_BNUM 2; constexpr int32_t A_SNUM 32; constexpr int32_t A_GNUM 3; constexpr int32_t A_DNUM 64; constexpr int32_t B_BNUM 2; constexpr int32_t B_SNUM 256; constexpr int32_t B_GNUM 3; constexpr int32_t B_DNUM 64; constexpr int32_t C_BNUM 2; constexpr int32_t C_SNUM 32; constexpr int32_t C_GNUM 3; constexpr int32_t C_DNUM 256; constexpr int32_t BATCH_NUM 3; tiling.SetALayout(A_BNUM, A_SNUM, 1, A_GNUM, A_DNUM); // 设置A矩阵排布 tiling.SetBLayout(B_BNUM, B_SNUM, 1, B_GNUM, B_DNUM); tiling.SetCLayout(C_BNUM, C_SNUM, 1, C_GNUM, C_DNUM); tiling.SetBatchNum(BATCH_NUM); tiling.SetBufferSpace(-1, -1, -1); optiling::TCubeTiling tilingData; int ret tiling.GetTiling(tilingData);【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
延伸阅读

更多相关文章

2026/11/1 6:09:24

AI技术前沿动态简报(2026.07.16)

第1条:英伟达发布 Audex 统一音频语言模型,缓解多模态训练中的能力退化核心内容:英伟达研究团队于 2026 年 7 月发布技术报告(arXiv:2607.05196),提出 Nemotron-Labs-Audex-30B-A3B(简称 Audex&…

2026/10/25 14:47:21

AI商业洞察动态简报(2026.07.15)

第1条:OPPO与支付宝推出多智能体协同方案,打通近200项生活服务商业价值:OPPO系统智能体"小布助手"与AI版支付宝"阿宝"完成深度协同,用户通过一句话即可调用近200项支付宝常用服务。该合作将手机厂商的端侧感知…

2026/11/3 16:56:08

Axon 4.4 事件溯源与 CQRS 实战:核心机制、配置与避坑指南

1. 拆解“Axon 4.4 中文版文档(九)”这个标题背后的真实需求 看到“Axon 4.4 中文版文档(九)”这个标题,很多没接触过事件驱动架构的朋友可能会一头雾水。我先把这个标题拆开揉碎讲清楚:Axon 是 Java 生态里…

2026/11/3 16:56:08

区块链技术:从记账演进到信用革命的核心解析

1. 区块链的本质:从记账技术到信用革命区块链技术本质上是一种分布式账本系统,它通过密码学算法和共识机制实现了去中心化的记账方式。这项技术的核心创新在于解决了传统金融体系中必须依赖中心化机构进行信用背书的根本问题。在传统银行体系中&#xff…

2026/11/3 16:56:08

findAllPeak雷达峰值检测:一维距离像到RD图多目标提取

简介:面向雷达信号处理与目标检测研究的工程师和高校师生,这份资源围绕雷达IQ原始数据的完整处理链路展开,聚焦从DCA1000采集数据读取到峰值精确识别的关键环节。压缩包内共2个文件,均为MATLAB脚本(.m)&…

2026/11/3 16:56:08

OpenClaw为何用WebAssembly做AI Agent沙箱?

1. “龙虾之父”是谁?——一场被误读的开源身份争夺战“OpenAI抢走‘龙虾之父’”这个标题乍看像一则科技圈八卦,实则戳中了当前开源AI生态里最敏感的一根神经:命名权、归属感与事实溯源的撕裂。它不是关于某个人被挖角,而是一场围…

2026/11/3 16:56:08

C#文本解析利器Superpower:原理、实战与性能优化

1. 文本解析的痛点与Superpower的诞生在C#开发领域,文本处理一直是高频需求却又是隐藏的痛点。我曾参与过一个物流系统的开发,需要解析来自不同承运商的货运单号——有的用竖线分隔,有的用逗号,还有的夹杂着不规则空格。用String.…

2026/11/3 16:51:06

量子计算测试:从传统到量子范式的技术跃迁

1. 量子计算测试的范式转变 那天凌晨三点,实验室的量子比特保真度突然突破阈值时,我正盯着示波器上那组异常的数据曲线发呆。作为传统半导体测试出身的工程师,第一次亲眼目睹量子态叠加产生的概率云分布,那种认知冲击不亚于当年从…

2026/11/3 7:47:06

面向复杂光照与角度的宠物面部识别鲁棒性增强策略

在智慧养宠时代,基于视觉的宠物身份识别技术正逐步从实验室走向家庭应用。然而,与配合度高的人类不同,宠物是典型的“非合作生物识别对象”——它们不会主动配合摄像头调整姿态,更无法控制光照环境。实际应用中,我们常…

2026/11/3 14:24:53

SSM电子书店源码全解析:结构、原理、部署与排错实战

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/11/3 1:04:48

STM32实战:基于MAX14431数字隔离器的SPI隔离通信方案

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/11/3 1:04:48

工业SPI隔离通信设计:MAX14431BAEE+与STM32L4A6RG实战

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/11/3 1:04:48

STM32+NRF24L01无线抢答器:硬件设计、通信协议与低功耗实现

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

2026/11/3 7:47:06

面向复杂光照与角度的宠物面部识别鲁棒性增强策略

在智慧养宠时代,基于视觉的宠物身份识别技术正逐步从实验室走向家庭应用。然而,与配合度高的人类不同,宠物是典型的“非合作生物识别对象”——它们不会主动配合摄像头调整姿态,更无法控制光照环境。实际应用中,我们常…

2026/11/3 14:24:53

SSM电子书店源码全解析:结构、原理、部署与排错实战

/* MD / 富文本中的 .toc(含博客园搬家等嵌套结构);.toc-box 在侧栏,不受影响 */#content_views .toc,/* 编辑器常在目录前后插入空 p(:empty 仍占 20px),一并去掉避免顶空隙 */#content_views.markdown_views > p:empty:has(+ .toc),#content_views.markdown_views …

还想了解更多?直接咨询顾问

免费诊断 + 免费方案 + 透明报价。

全国咨询热线400-8866-253
免费获取方案
☎咨询二维码 ☎ ↑