尧图网络 高端网站定制 · 原创设计
免费咨询热线
400-888-6620
免费获取方案
CANN/asc-devkit原子递增操作
asc_atomic_inc【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit产品支持情况Ascend 950PR/Ascend 950DT支持Atlas A3 训练系列产品/Atlas A3 推理系列产品不支持Atlas A2 训练系列产品/Atlas A2 推理系列产品不支持Atlas 200I/500 A2 推理产品不支持Atlas 推理系列产品AI Core不支持Atlas 推理系列产品Vector Core不支持Atlas 训练系列产品不支持功能说明对Unified Buffer或Global Memory上address的数值进行原子加1操作如果address上的数值大于等于指定数值val则对address赋值为0否则将address上数值加1。函数原型inline uint32_t asc_atomic_inc(uint32_t *address, uint32_t val)inline uint64_t asc_atomic_inc(uint64_t *address, uint64_t val)参数说明表1参数说明参数名输入/输出描述address输出Unified Buffer或Global Memory的地址。val输入源操作数。不同数据类型支持的内存范围说明如下表2不同数据类型支持的内存范围参数数据类型支持的内存空间uint32_tUnified Buffer、Global Memoryuint64_tGlobal Memory返回值说明Unified Buffer或Global Memory上的初始数据。约束说明原子操作保证对同一地址的读改写过程具有原子性但不保证多个线程之间的执行顺序。对于依赖返回值分配序号或槽位的场景返回值对应的序号唯一但分配给具体线程的顺序可能随线程调度变化而不同。需要包含的头文件使用该接口需要包含simt_api/device_atomic_functions.h头文件。#include simt_api/device_atomic_functions.h调用示例示例场景为多个线程向固定容量的环形缓冲区写入任务使用asc_atomic_inc接口获取递增并自动回绕的槽位编号。返回值是更新前的旧计数可作为本线程获得的槽位。输入参数说明如下名称说明ticketGlobal Memory中的环形计数器kernel启动前初始化。slots保存每个线程获得的槽位编号。capacity环形队列容量。n需要分配槽位的线程数。核心代码实现如下SIMT编程场景__global__ __launch_bounds__(256) void allocate_ring_slot(uint32_t *ticket, uint32_t *slots, uint32_t capacity, uint32_t n) { uint32_t idx blockIdx.x * blockDim.x threadIdx.x; if (idx n) { return; } uint32_t old_ticket asc_atomic_inc(ticket, capacity - 1U); slots[idx] old_ticket; }SIMD与SIMT混合编程场景SIMD与SIMT混合编程场景需要显式使用地址空间限定符表示地址空间__gm__表示Global Memory内存空间__ubuf__表示Unified Buffer内存空间。__simt_vf__ __launch_bounds__(1024) inline void allocate_ring_slot(__gm__ uint32_t *ticket, __gm__ uint32_t *slots, uint32_t capacity, uint32_t n) { uint32_t idx blockIdx.x * blockDim.x threadIdx.x; if (idx n) { return; } uint32_t old_ticket asc_atomic_inc(ticket, capacity - 1U); slots[idx] old_ticket; }输出结果示例如下ticket before: 0 capacity: 4 n: 6 slots: 0, 1, 2, 3, 0, 1 // 顺序由实际原子执行顺序决定 ticket after: 2【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
RELATED

相关推荐

国内外统计学教材差异对比:理论体系与实践导向的学习路径分析

国内外统计学教材差异对比:理论体系与实践导向的学习路径分析

这次我们来看一个很有意思的话题:国外统计学教材和国内高数教材的差异。很多人在学习过程中都会有这样的疑问——为什么国内教材总是按部就班地从基本概念开始,而国外教材似乎更注重实际应用?这背后其实反映了不同的教学理念和学习路径。 国…

📅 2026/7/20 14:46:25
RTX4050在Windows下编译3D Gaussian Splatting全链路踩坑指南

RTX4050在Windows下编译3D Gaussian Splatting全链路踩坑指南

1. 为什么Windows下装3DGS像在雷区跳探戈:从RTX4050显卡开始的真实困境 你手头刚到一台搭载RTX4050笔记本,满心欢喜想跑通3D Gaussian Splatting——这个最近刷爆CV圈的实时新视图合成技术。结果刚敲下 pip install -e . ,终端就弹出一串红…

📅 2026/9/9 20:57:21
让QQ音乐文件在任意设备自由播放:qmcdump解密转换方案

让QQ音乐文件在任意设备自由播放:qmcdump解密转换方案

让QQ音乐文件在任意设备自由播放:qmcdump解密转换方案 【免费下载链接】qmcdump 一个简单的QQ音乐解码(qmcflac/qmc0/qmc3 转 flac/mp3),仅为个人学习参考用。 项目地址: https://gitcode.com/gh_mirrors/qm/qmcdump 你是否…

📅 2026/7/21 13:39:59
MORE NEWS

更多资讯

📰

niri 浮动窗口(Floating Windows)完全指南:布局切换、窗口规则与精确定位

niri 浮动窗口(Floating Windows)完全指南:布局切换、窗口规则与精确定位 【免费下载链接】niri A scrollable-tiling Wayland compositor. 项目地址: https://gitcode.com/GitHub_Trending/ni/niri 导读:本文以 niri 官方 …

📰

从第一性原理推导 ML 系统:cs249r_book 的 ML Systems Design Grammar 原语目录与重写规则实战指南

从第一性原理推导 ML 系统:cs249r_book 的 ML Systems Design Grammar 原语目录与重写规则实战指南 【免费下载链接】cs249r_book Machine Learning Systems 项目地址: https://gitcode.com/GitHub_Trending/cs/cs249r_book 导读:本篇技术指南围绕…

📰

Arm-2D:Cortex-M上零动态内存的确定性2D像素操作原语

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

📰

WezTerm Lua API 详解:`color:complement_ryb()` 与基于 RYB 色彩模型的补色计算

WezTerm Lua API 详解:color:complement_ryb() 与基于 RYB 色彩模型的补色计算 【免费下载链接】wezterm A GPU-accelerated cross-platform terminal emulator and multiplexer written by wez and implemented in Rust 项目地址: https://gitcode.com/GitHub_Tr…

📰

Ente Locker Legacy 数字继承与账户恢复指南:Trusted Contacts 与 Legacy Kits 完整解析

Ente Locker Legacy 数字继承与账户恢复指南:Trusted Contacts 与 Legacy Kits 完整解析 【免费下载链接】ente 💚 End-to-end encrypted cloud for everything. 项目地址: https://gitcode.com/GitHub_Trending/en/ente 本文是一份基于 Ente 开源…

📰

daisyui Radial Progress 组件完全指南:从 `--value` 原理到无障碍进度环实战

daisyui Radial Progress 组件完全指南:从 --value 原理到无障碍进度环实战 【免费下载链接】daisyui 🌼 🌼 🌼 🌼 🌼  The most popular, free and open-source Tailwind CSS component library 项目地…

TODAY

今日更新

THIS WEEK

本周精选

THIS MONTH

本月热门

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

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

📞 💬