Duplicate

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

产品支持情况

功能说明

将一个变量或立即数复制多次并填充到向量中。

针对Ascend 950PR/Ascend 950DT，为方便开发者使用，tensor前n个数据计算接口同时也支持直接传入Tensor，此时会将Tensor的第一个元素复制多次并填充到向量中。

函数原型

• tensor前n个数据计算

  • 源操作数为标量

    template <typename T>
    __aicore__ inline void Duplicate(const LocalTensor<T>& dst, const T& scalarValue, const int32_t& count)
    

  • 源操作数为Tensor

    template <typename T>
    __aicore__ inline void Duplicate(const LocalTensor<T>& dst, const LocalTensor<T>& src, const int32_t& count)
    

• tensor高维切分计算

  • mask逐比特模式

    template <typename T, bool isSetMask = true>
    __aicore__ inline void Duplicate(const LocalTensor<T>& dst, const T& scalarValue, uint64_t mask[], const uint8_t repeatTime, const uint16_t dstBlockStride, const uint8_t dstRepeatStride)
    

  • mask连续模式

    template <typename T, bool isSetMask = true>
    __aicore__ inline void Duplicate(const LocalTensor<T>& dst, const T& scalarValue, uint64_t mask, const uint8_t repeatTime, const uint16_t dstBlockStride, const uint8_t dstRepeatStride)
    

参数说明

表 1 模板参数说明

表 2 参数说明

约束说明

• 操作数地址对齐要求请参见通用地址对齐约束。
• 针对Ascend 950PR/Ascend 950DT，bool、int8_t、uint8_t、fp4x2_e2m1_t、fp4x2_e1m2_t、hifloat8_t、fp8_e5m2_t、fp8_e4m3fn_t、 fp8_e8m0_t、complex32、int64_t、uint64_t、complex64数据类型仅支持tensor前n个数据计算接口。

返回值说明

无

调用示例

本示例仅展示Compute流程的部分代码。如需运行，请将代码段复制并粘贴到Duplicate样例中的Compute函数对应位置。

• tensor高维切分计算样例-mask连续模式

  uint64_t mask = 128;
  half scalar = 18.0;
  // repeatTime = 2, 128 elements one repeat, 256 elements total
  // dstBlkStride = 1, no gap between blocks in one repeat
  // dstRepStride = 8, no gap between repeats
  AscendC::Duplicate(dstLocal, scalar, mask, 2, 1, 8 );
  

• tensor高维切分计算样例-mask逐bit模式

  uint64_t mask[2] = { UINT64_MAX, UINT64_MAX };
  half scalar = 18.0;
  // repeatTime = 2, 128 elements one repeat, 256 elements total
  // dstBlkStride = 1, no gap between blocks in one repeat
  // dstRepStride = 8, no gap between repeats
  AscendC::Duplicate(dstLocal, scalar, mask, 2, 1, 8 );
  

• tensor前n个数据计算样例，源操作数为标量

  half inputVal(18.0);
  int32_t srcDataSize = 256; // 参与计算的元素个数
  AscendC::Duplicate<half>(dstLocal, inputVal, srcDataSize);
  

• tensor前n个数据计算样例，源操作数为Tensor

  AscendC::Duplicate<half>(dstLocal, srcLocal, srcDataSize);
  

结果示例如下：

scalar: 18.0
srcLocal: [18.0 1.0 2.0 ... 254.0 255.0]
dstLocal: [18.0 18.0 18.0 ... 18.0 18.0]

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