文件最后提交记录最后更新时间
5 个月前
9 个月前
8 个月前
3 个月前
3 个月前
6 个月前
8 个月前
5 个月前
README

Sqrt

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
Nice_try 西北工业大学-智能感知交互实验室 Sqrt 2024/12/24 新增Sqrt算子
liuxiqiang 个人开发者 Sqrt 2025/11/12 Sqrt算子适配开源仓

支持的产品型号

  • Atlas A2训练系列产品
  • Atlas 200I/500 A2推理产品

算子描述

  • 功能描述

    Sqrt算子返回输入数据经过开方运算的结果。

  • 原型信息

    算子类型(OpType)Sqrt
    nameTypedata typeformat
    算子输入
    xtensorfloat32,float16,bfloat16ND
    算子输出 ytensorfloat32,float16,bfloat16ND
    核函数名sqrt

约束与限制

  • x,y,out的数据类型仅支持float32,float16,bfloat16,数据格式仅支持ND

算子使用

使用该算子前,请参考社区版CANN开发套件包安装文档完成开发运行环境的部署。

编译部署

  • 进入到仓库目录

    cd ${git_clone_path}/ops-math
    
  • 执行编译

    bash build.sh --pkg --experimental --soc=ascend910b --ops=sqrt
    
  • 部署算子包

    ./build_out/cann-ops-<vendor_name>-linux.<arch>.run
    

算子调用

  • 执行调用

    bash build.sh --run_example sqrt eager cust --vendor_name=custom
    

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_sqrt 通过aclnnSqrt接口方式调用Sqrt算子。