每个算子分为,必须先调用[object Object]接口获取入参并根据计算流程计算所需workspace大小,再调用[object Object]接口执行计算。
[object Object]
[object Object]
参数说明:
[object Object][object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:
[object Object]的数据类型与入参[object Object]的数据类型具有一定对应关系:- 当
[object Object]的数据类型为FLOAT32、BFLOAT16时,[object Object]的数据类型支持FLOAT32、DOUBLE。 - 当
[object Object]的数据类型为FLOAT16时,[object Object]的数据类型支持FLOAT16、DOUBLE。 - 当
[object Object]的数据类型为INT32时,[object Object]的数据类型支持INT32、INT64。
- 当
Atlas 350 加速卡:
- 入参
[object Object]和出参[object Object]支持包含的最大Tensor个数均为50。 [object Object]的数据类型与入参[object Object]的数据类型具有一定对应关系:- 当
[object Object]的数据类型为BFLOAT16、FLOAT32时,[object Object]的数据类型支持FLOAT32、DOUBLE。 - 当
[object Object]的数据类型为FLOAT16时,[object Object]的数据类型支持FLOAT16、FLOAT32、DOUBLE。 - 当
[object Object]的数据类型为INT32时,[object Object]的数据类型支持INT32、INT64。
- 当
- 入参
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
- 确定性计算:
- aclnnForeachMulScalarV2默认确定性实现。
[object Object]