aclnnCast

产品支持情况

[object Object]undefined

功能说明

将输入tensor转换为指定的dtype类型。

函数原型

每个算子分为，必须先调用“aclnnCastGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnCast”接口执行计算。

aclnnCastGetWorkspaceSize

• 参数说明：

  [object Object]
  • [object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas 推理系列产品[object Object]：不支持BFLOAT16、INT4。
  • [object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：不支持COMPLEX32、HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3FN、FLOAT4_E2M1、FLOAT4_E1M2、INT4。

• 返回值：

  aclnnStatus：返回状态码，具体参见。

  第一段接口会完成入参校验，出现以下场景时报错：

  [object Object]

aclnnCast

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

约束说明

• 确定性计算：

  • aclnnCast默认确定性实现。

• 针对数据类型从浮点数转换为整型的场景： 输入数据中存在nan，则将nan转换为0。

• 针对输入数据类型为BOOL、COMPLEX32、COMPLEX64、COMPLEX128、FLOAT4_E2M1、FLOAT4_E1M2的场景： 不支持输入为非连续。

• [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：

  • 针对数据类型从int32转换为int8的场景： 只能保证输入数据在(-2048, 1920)范围内精度无误差。
  • 针对数据类型从float64/complex64/complex128转换为uint8的场景： 只能保证输入数据为非负数精度无误差。

• [object Object]Atlas 推理系列产品[object Object]：

  • 针对数据类型从float32转换为int64和float32转换为uint8的场景： 只能保证输入数据在(-2147483648, 2147483583)范围内精度无误差。

  • 针对数据类型从int64转换为float32的场景： 只能保证输入数据在(-2147483648, 2147483647)范围内精度无误差。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。