昇腾社区首页
中文
注册

Swap

功能描述

  • asdBlasMakeSwapPlan

    接口功能:初始化该句柄对应的Swap算子配置。

  • asdBlasSswap

    接口功能:交换两个实数向量。

  • asdBlasCswap

    接口功能:交换两个复数向量。

计算公式

  • asdBlasSswap
    • 计算公式:
    • 示例:

      输入“x”为:

      [1.0, 2.0]

      输入“y”为:

      [3.0, 4.0]

      调用“Sswap”算子后,

      输出“x”为:

      [3.0, 4.0]

      输出“y”为:

      [1.0, 2.0]
  • asdBlasCswap
    • 计算公式:
    • 示例:

      输入“x”为:

      [1.0 + 1.0i, 2.0 + 2.0i]

      输入“y”为:

      [3.0 + 3.0i, 4.0 + 4.0i]

      调用“Cswap”算子后,

      输出“x”为:

      [3.0 + 3.0i, 4.0 + 4.0i]

      输出“y”为:

      [1.0 + 1.0i, 2.0 + 2.0i]

接口原型

  • AspbStatus asdBlasMakeSwapPlan(asdBlasHandle handle)
  • AspbStatus asdBlasSswap(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *y, const int64_t incy)
  • AspbStatus asdBlasCswap(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *y,const int64_t incy)

参数列表

  • asdBlasMakeSwapPlan

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    swap算子的句柄。

  • asdBlasSswap & asdBlasCswap

    参数名称

    Input/Output

    类型

    描述

    handle

    Input

    asdBlasHandle

    swap算子的句柄。

    n

    Input

    const int64_t

    向量x(y)中元素的个数。

    x

    Input/Output

    aclTensor *

    公式中的x,Device侧的Tensor,数据格式支持ND,shape为[n]。

    • Sswap下数据类型支持FLOAT32;
    • Cswap下数据类型支持COMPLEX64。

    incx

    Input

    const int64_t

    x相邻元素间的内存地址偏移量(当前约束为1)。

    y

    Input

    /Output

    aclTensor *

    公式中的y,Device侧的Tensor,数据格式支持ND,shape为[n]。

    • Sswap下数据类型支持FLOAT32;
    • Cswap下数据类型支持COMPLEX64。

    incy

    Input

    const int64_t

    y相邻元素间的内存地址偏移量(当前约束为1)。

规格约束

  • asdBlasMakeSwapPlan:无。
  • asdBlasSswap
    • 输入和输出数据格式、数据类型、shape保持一致。
    • 输入的元素个数n当前覆盖支持[1,6.71e+07];
    • 算子输入shape为[n]、[n],输出shape为[n]、[n];
    • 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
  • asdBlasCswap
    • 输入和输出数据格式、数据类型、shape保持一致。
    • 输入的元素个数n当前覆盖支持[1,6.71e+07];
    • 算子输入shape为[n]、[n],输出shape为[n]、[n];
    • 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。