Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

Backgroud（背景信息）

使用AscendC对TBE实现的Sqrt算子进行重构，实现了AscendC实现的Sqrt算子对Atlas 200/500 A2推理产品和Atlas 800I/T A2硬件的适配。

Origin（信息来源）

外部贡献

Benefit / Necessity （价值/作用）

Sqrt算子的主要功能是计算给定数值的平方根。在数学和工程领域中，平方根运算是一种基础且重要的操作，它被广泛应用于图像处理、信号处理、物理模拟等多个领域。Sqrt算子能够高效地处理批量数值的平方根计算，支持浮点数的输入。实现了Sqrt算子的AscendC实现，替代原有TBE算子在昇腾硬件上的适配。

支持设备：
Atlas 200/500 A2推理产品
Atlas 800I/T A2

Design（设计方案）

host侧设计：

tiling策略：
1)分核策略

优先使用满核的原则。
如果核间能均分，可视作无大小核区分，大核小核数据块一致；
如果核间不能均分，需要将余出的数据块分配到前几个核上。
数据量当能单核处理的情况下使用单核，否者使用满核进行加速处理。

2)单core内切分策略
充分使用UB空间的原则。
需要考虑不同硬件的UB大小不同、是否开启double buffer、kernel侧API实现过程中是否需要临时数据的储存，综合考虑单核内切分的大小。

3)tilingkey规划策略
需要tilingkey的情况：需要感知host侧信息对kernel侧走不同分支。
不需要tilingkey的情况：host侧的处理是通用的，无特殊属性等决定kernel的分支

数据检测：
对不支持AscendC::Cast()bfloat16向float32转换的硬件进行直接在tiling策略时返回Get Operator Workspace failed. error code is 561002报错。

kernel侧设计：

进行Init和Process两个阶段，其中Process包括数据搬入（CopyIn）、计算（Compute）、搬出（CopyOut）三个阶段。
1)由于支持Ascend C开发的硬件中AscendC::Sqrt都支持float32数据的输入，可以直接将float16和bfloat16的数据都转成float32进行计算，float32的类型保持原类型计算。
2)由于不支持“Intrinsic_vsqrt”指令的硬件，不支持Ascend C算子开发。所以可以直接使用Ascend C::Sqrt接口进行处理。
3)Ascend C的Sqrt算子流程见下图。