CANN/Ascend C LogicalNot逻辑非接口文档

LogicalNot

【免费下载链接】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 训练系列产品:不支持

功能说明

按元素进行取反操作。当输入的数据类型不是bool时,零被视为False,非零数据被视为True。

函数原型

template <const LogicalNotConfig& config = DEFAULT_LOGICAL_NOT_CONFIG, typename T, typename U> __aicore__ inline void LogicalNot(const LocalTensor<T>& dst, const LocalTensor<U>& src, const uint32_t count)

参数说明

表1模板参数说明

参数名描述
configLogicalNot算法的相关配置。此参数可选配,LogicalNotConfig类型,具体定义如下方代码所示,其中参数的含义为:

isReuseSource:该参数预留,传入默认值false即可。
T目的操作数的数据类型。支持的数据类型为:bool。
U源操作数的数据类型。支持的数据类型为:bool、int8_t、uint8_t、int16_t、uint16_t、half、bfloat16_t、int32_t、uint32_t、float、int64_t、uint64_t。
struct LogicalNotConfig { bool isReuseSource; };

表2接口参数说明

参数名输入/输出描述
dst输出目的操作数。

类型为LocalTensor,支持的TPosition为VECIN/VECCALC/VECOUT。
src输入源操作数。

类型为LocalTensor,支持的TPosition为VECIN/VECCALC/VECOUT。
count输入参与计算的元素个数。

返回值说明

约束说明

  • 不支持源操作数与目的操作数地址重叠。
  • 操作数地址对齐要求请参见通用地址对齐约束。

调用示例

AscendC::LocalTensor<bool> dst; AscendC::LocalTensor<half> src; uint32_t count = 512; // 参与计算的元素个数 AscendC::LogicalNot(dst, src, count);

结果示例如下:

输入数据(src): [1, 2, 0, -1, -2, 0, 3, 4, 0, -3, -4, 0, 5, 6, 0, -5, -6, 0, ... 0] 输出数据(dst): [False, False, True, False, False, True, False, False, True, False, False, True, False, False, True, False, False, True, ... True]

【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言,原生支持C和C++标准规范,主要由类库和语言扩展层构成,提供多层级API,满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考