尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
cann/asc-devkit:gather_ld_reg样例
gather_ld_reg样例概述本样例基于Reg编程接口实现将离散数据从UB(Unified Buffer)搬入到RegTensor(Reg矢量计算基本单元)的功能支持两种场景通过 CMake 编译参数SCENARIO_NUM选择场景。SCENARIO_NUM搬入场景1Gather从UB按索引将单点数据收集到RegTensor2GatherB从UB按索引将DataBlock大小的数据收集到RegTensor本样例支持的产品及CANN软件版本产品CANN软件版本Ascend 950PR/Ascend 950DT CANN 9.1.0目录结构介绍├── reg_load_gather │ ├── scripts │ │ ├── gen_data.py // 输入数据和真值数据生成脚本 │ ├── CMakeLists.txt // 编译工程文件 │ ├── data_utils.h // 数据读入写出函数 │ ├── gather_ld_reg.asc // AscendC样例实现 调用样例 │ └── README.md // 样例介绍样例描述样例功能根据索引从源数据中离散收集元素到目的地址。场景1Gather模式使用Gather接口从UB按元素索引收集数据到RegTensor按元素收集dst[i] src[index[i]]源数据量为1024个元素索引数据量为128个元素输出数据量为128个元素。样例规格样例类型(OpType)AIV样例样例输入nameshapedata typesrc[1, 1024]halfindex[1, 128]uint16样例输出dst[1, 128]half核函数名gather_ld_reg样例实现 GatherVF函数内先通过LoadAlign加载索引数据到indexReg再调用Gather接口从源地址按索引收集数据到dstReg最后通过StoreAlign将结果写回UB。调用实现 使用内核调用符调用核函数启动1个核。场景2GatherB模式使用GatherB接口按32字节DataBlock收集源数据量为1024个元素索引数据量为8个元素对应8个DataBlock输出数据量为128个元素。样例规格样例类型(OpType)AIV样例样例输入nameshapedata typesrc[1, 1024]halfindex[1, 8]uint32样例输出dst[1, 128]half核函数名gather_ld_reg样例实现 GatherBVF函数内先通过LoadAlign加载索引数据到indexReg再调用GatherB接口按DataBlock从源地址收集数据到dstReg最后通过StoreAlign将结果写回UB。调用实现 使用内核调用符调用核函数启动1个核。编译运行在本样例根目录下执行如下步骤编译并执行样例。配置环境变量 请根据当前环境上CANN开发套件包的安装方式配置环境变量。source ${install_path}/cann/set_env.sh说明${install_path}为CANN包安装目录未指定安装目录时默认安装至/usr/local/Ascend下。样例执行在本样例目录下执行如下命令。SCENARIO_NUM1 # 选择执行场景1Gather2GatherB mkdir -p build cd build; # 创建并进入build目录 cmake -DSCENARIO_NUM$SCENARIO_NUM -DCMAKE_ASC_ARCHITECTURESdav-3510 ..;make -j; # 编译工程默认npu模式 python3 ../scripts/gen_data.py -scenarioNum $SCENARIO_NUM # 生成测试输入数据 ./demo # 执行编译生成的可执行程序执行样例使用 CPU调试 或 NPU仿真 模式时添加-DCMAKE_ASC_RUN_MODEcpu或-DCMAKE_ASC_RUN_MODEsim参数即可。示例如下cmake -DSCENARIO_NUM$SCENARIO_NUM -DCMAKE_ASC_RUN_MODEcpu -DCMAKE_ASC_ARCHITECTURESdav-3510 ..;make -j; # cpu调试模式 cmake -DSCENARIO_NUM$SCENARIO_NUM -DCMAKE_ASC_RUN_MODEsim -DCMAKE_ASC_ARCHITECTURESdav-3510 ..;make -j; # NPU仿真模式注意切换编译模式或场景前需清理 cmake 缓存可在 build 目录下执行rm CMakeCache.txt后重新 cmake。编译选项说明选项可选值说明SCENARIO_NUM1、2样例执行场景场景1Gather模式、场景2GatherB模式CMAKE_ASC_RUN_MODEnpu默认、cpu、sim运行模式NPU 运行、CPU调试、NPU仿真CMAKE_ASC_ARCHITECTURESdav-3510NPU 架构dav-3510 对应 Ascend 950PR/Ascend 950DT执行结果执行结果如下说明精度对比成功。test pass!创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED

相关推荐

掌握LZ-String压缩算法:JavaScript数据压缩的5大实战技巧

掌握LZ-String压缩算法:JavaScript数据压缩的5大实战技巧

掌握LZ-String压缩算法:JavaScript数据压缩的5大实战技巧 【免费下载链接】lz-string LZ-based compression algorithm for JavaScript 项目地址: https://gitcode.com/gh_mirrors/lz/lz-string LZ-String是一个基于LZ77压缩算法的高效JavaScript库&#xff…

📅 2026/10/30 17:26:49
RimWorld终极角色定制指南:如何用EdB Prepare Carefully打造完美殖民者

RimWorld终极角色定制指南:如何用EdB Prepare Carefully打造完美殖民者

RimWorld终极角色定制指南:如何用EdB Prepare Carefully打造完美殖民者 【免费下载链接】EdBPrepareCarefully EdB Prepare Carefully, a RimWorld mod 项目地址: https://gitcode.com/gh_mirrors/ed/EdBPrepareCarefully 厌倦了随机生成的殖民者无法满足你的…

📅 2026/9/13 2:33:42
CANN/asc-devkit:自定义算子工程多Vendor并行编译、打包和部署样例

CANN/asc-devkit:自定义算子工程多Vendor并行编译、打包和部署样例

自定义算子工程多 Vendor 并行编译、打包和部署样例 概述 本样例展示如何在一个顶层 CMake 工程中,使用 ExternalProject_Add 并行编译两个独立自定义算子工程: add_custom(AddCustom)leaky_relu_custom(LeakyReluC…

📅 2026/9/27 3:44:02
MORE NEWS

更多资讯

📰

电工证可以从事的行业保姆级教程:新手必看的就业方向和报考指南

电工证可以从事的行业保姆级教程:新手必看的就业方向和报考指南 你现在手里还拿着电工证吗?有没有遇到过这种情况:拿着电工证,却不知道能去哪家公司上班?别急,这篇文章就是专门为你写的。 急需特种作业证才能上岗找工作…

📰

电工证到期怎么年审千万别踩坑 真实案例教你少走弯路

电工证到期怎么年审千万别踩坑 真实案例教你少走弯路 你手里那本电工证快到期了?别慌,但 千万别踩坑 ,很多人因为复审没搞明白,结果直接被单位辞退,连新工作都找不到。特别是像我这样从三门峡转岗过来的电工,跨省办证、复审流程都不一样,稍不留神就…

📰

电工证的证号是编号吗 怎么报名能顺利通过

电工证的证号是编号吗 怎么报名能顺利通过 你是不是也这样想: 学历不高怕报不上名 ,想着去考个电工证,但又担心流程太复杂,根本不知道从哪开始?别急,这篇文章就帮你把电工证的报考全流程拆解清楚,特别是 电工证的证号是编号吗…

📰

南京市报考电工证怎么考?工地太忙根本没时间复习考试,考前押题全搞定

南京市报考电工证怎么考?工地太忙根本没时间复习考试,考前押题全搞定 工地太忙根本没时间复习考试,这是很多电工朋友的真实写照,尤其是南京本地的建筑工人,经常一忙就是几个月,根本没时间准备电工证考试。别急,本文就是为你们量身定制的,从报名入口到…

📰

萧县电工证在哪考的啊,真实经历告诉你怎么报名

萧县电工证在哪考的啊,真实经历告诉你怎么报名 你是不是也和我一样,学历不高,担心自己连电工证都报不上名?别急,我当年就是个初中毕业的电工,靠着自己的努力,顺利考下低压电工证,现在在萧县做电工已经快五年了。今天就把我这条“真实经历”分享给你,…

📰

外省井下特种电工证能转过来吗?含金量还在不在?

外省井下特种电工证能转过来吗?含金量还在不在? 之前考的证在外省能不能转过来?这是很多特种作业人员,特别是井下特种电工证持有者最关心的问题。其实,不少朋友在换工作、跨地区发展时,都遇到过证书不能直接使用的情况。井下特种电工证的含金量不低,但…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

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

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

📞 💬