h2sqrt产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品xAtlas A2 训练系列产品/Atlas A2 推理系列产品xAtlas 200I/500 A2 推理产品xAtlas 推理系列产品AI CorexAtlas 推理系列产品Vector CorexAtlas 训练系列产品x功能说明获取输入数据x各元素的平方根。函数原型inline bfloat16x2_t h2sqrt(bfloat16x2_t x)参数说明表 1参数说明参数名输入/输出描述x输入源操作数输入数据。返回值说明输入数据各元素的平方根。当输入元素为0时返回值为0。当输入元素为inf时返回值为inf。当输入元素为-inf时返回值为nan。当输入元素为nan时返回值为nan。约束说明本接口支持的输入数据各元素范围为x大于等于0否则返回值为nan。需要包含的头文件使用bfloat16x2_t类型接口需要包含simt_api/asc_bf16.h头文件。#include simt_api/asc_bf16.h调用示例SIMT编程场景__global__ __launch_bounds__(1024) void KernelSqrt(bfloat16x2_t* dst, bfloat16x2_t* x) { int idx threadIdx.x blockIdx.x * blockDim.x; dst[idx] h2sqrt(x[idx]); }SIMD与SIMT混合编程场景__simt_vf__ __launch_bounds__(1024) inline void KernelSqrt(__gm__ bfloat16x2_t* dst, __gm__ bfloat16x2_t* x) { int idx threadIdx.x blockIdx.x * blockDim.x; dst[idx] h2sqrt(x[idx]); }创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考