CANN/asc-devkit Rint数学计算接口-北京尧图网络

Rint【免费下载链接】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 训练系列产品不支持功能说明获取与输入数据最接近的整数若存在两个同样接近的整数则获取其中的偶数。计算公式如下举例如下Rint(3.9) 4Rint(3.3) 3Rint(3.5) 4函数原型通过sharedTmpBuffer入参传入临时空间template const RintConfig config DEFAULT_RINT_CONFIG, typename T __aicore__ inline void Rint(const LocalTensorT dst, const LocalTensorT src, const LocalTensoruint8_t sharedTmpBuffer, const uint32_t count)接口框架申请临时空间template const RintConfig config DEFAULT_RINT_CONFIG, typename T __aicore__ inline void Rint(const LocalTensorT dst, const LocalTensorT src, const uint32_t count)由于该接口的内部实现中涉及复杂的数学计算需要额外的临时空间来存储计算过程中的中间变量。临时空间支持开发者通过sharedTmpBuffer入参传入和接口框架申请两种方式。通过sharedTmpBuffer入参传入使用该tensor作为临时空间进行处理接口框架不再申请。该方式开发者可以自行管理sharedTmpBuffer内存空间并在接口调用完成后复用该部分内存内存不会反复申请释放灵活性较高内存利用率也较高。接口框架申请临时空间开发者无需申请但是需要预留临时空间的大小。通过sharedTmpBuffer传入的情况开发者需要为tensor申请空间接口框架申请的方式开发者需要预留临时空间。临时空间大小BufferSize的获取方式如下通过GetRintMaxMinTmpSize中提供的接口获取需要预留空间范围的大小。参数说明表1模板参数说明参数名描述RintConfigRint算法的相关配置。此参数可选配RintConfig类型具体定义如下方代码所示其中参数的含义为isReuseSource是否允许修改源操作数。该参数预留传入默认值false即可。T操作数的数据类型。支持的数据类型为half、float。struct RintConfig { bool isReuseSource; };表2接口参数说明参数名输入/输出描述dst输出目的操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。src输入源操作数。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。源操作数的数据类型与目的操作数保持一致。sharedTmpBuffer输入临时缓存。类型为LocalTensor支持的TPosition为VECIN/VECCALC/VECOUT。用于Rint内部复杂计算时存储中间变量由开发者提供。临时空间大小BufferSize的获取方式请参考GetRintMaxMinTmpSize。count输入参与计算的元素个数。返回值说明无约束说明不支持源操作数与目的操作数地址重叠。不支持sharedTmpBuffer与源操作数和目的操作数地址重叠。操作数地址对齐要求请参见通用地址对齐约束。调用示例完整样例请参考Rint算子样例。通过sharedTmpBuffer入参传入AscendC::TPipe pipe; AscendC::TQueAscendC::TPosition::VECCALC, 1 tmpQue; pipe.InitBuffer(tmpQue, 1, bufferSize); // bufferSize通过Host侧tiling参数获取 AscendC::LocalTensoruint8_t sharedTmpBuffer tmpQue.AllocTensoruint8_t(); // 输入tensor长度为1024,算子输入的数据类型为half,实际计算个数为512 static constexpr AscendC::RintConfig rintConfig {false}; // 不修改源操作数 // dst、src为half类型的LocalTensor AscendC::RintrintConfig, half(dst, src, sharedTmpBuffer, 512);接口框架申请临时空间AscendC::TPipe pipe; AscendC::TQueAscendC::TPosition::VECCALC, 1 tmpQue; pipe.InitBuffer(tmpQue, 1, bufferSize); // bufferSize通过Host侧tiling参数获取 // 输入tensor长度为1024,算子输入的数据类型为half,实际计算个数为512 static constexpr AscendC::RintConfig rintConfig {false}; // 不修改源操作数 // dst、src为half类型的LocalTensor AscendC::RintrintConfig, half(dst, src, 512);结果示例如下输入数据(src): [-1.5, -1.3, -0.9, -0.5, 0.5, 0.9, 1.1, 1.5, 2.2, 2.5, 2.8, 3.2] 输出数据(dst): [-2, -1, -1, 0, 0, 1, 1, 2, 2, 2, 3, 3]【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

CANN/asc-devkit Rint数学计算接口

相关推荐

SmoothQuant技术解析：大模型无损量化的关键突破

昇腾C数学计算API使用样例

昇腾C数据过滤临时空间API

更多资讯

从一条氨基酸序列到3D结构：AlphaFold Web服务实操全流程

PCIe如何成为机器人控制器的实时性锚点与系统中枢

AI数据集划分：核心原理与OpenClaw实践

基于GSM短信模块与单片机的远程开锁系统设计解析

PM Skills Marketplace 贡献指南：从 frontmatter 规范到自动化校验的插件开发实战

风电电力系统低碳调度模型与Matlab实现

今日更新

Linux 内核 ARM 虚拟内存布局全解析：从 4GB 地址空间划分到源码级验证

AIGC降重工具测评与教育应用指南

数据容灾核心指标与实战方案解析

本周精选

超人会飞不算本事：系统稳定依赖清晰规则与边界设计

超人VS蜘蛛侠：拆解超级IP的影响力与传播方法论

基于CNN的调制信号识别：MATLAB实现时频图分类实战

本月热门

基于YOLO的焊缝缺陷检测：从数据标注到工业部署全流程实践

分布式服务的决策记录

创业产品的首版范围

读完文章，想聊聊您的网站？