aclrtReset Device

本文分享自华为云社区《【2023 · CANN训练营第一季】——Ascend C sqrt算子实战》，作者：dayao。 前言 编写一个Ascend C的sqrt算子，并通过内核调用方式在cpu和npu模式下进行验证。在训练营沙箱环境下，cpu模式工作正常结果正确。 一、概述 先简单回顾下TIK C++算子矢量编程的流程和实现。 矢量算子开发流程如下： 主要工作内容有： 1、算子分析：确定输入输出，确定数学表达式以及底层实现接口，确定核函数定义。 2、算子类的实现：实现init()和process()。init()完成内存初始化，实质上体现的是多核运行，和单核数据切分以及是否开启double buffer优化；Process()实现的是CopyIn，Compute、CopyOut三个流水任务。 3、算子验证：通过核函数的内核调用符的方式调用算子，计算出结果，并于使用相同输入用numpy计算结果进行比对，误差在一定范围内即可。实际应用中，需要使用原有框架的算子进行计算精度比对。 二、算子分析 算子定义如下：假定仍是8个逻辑核。 查询TIK C++的API可知，可以使用（TIK...