aclrtReduceAsync

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

执行Reduce操作,包括SUM、MIN、MAX等。异步接口。

函数原型

aclError aclrtReduceAsync(void *dst, const void *src, uint64_t count, aclrtReduceKind kind, aclDataType type, aclrtStream stream, void *reserve)

参数说明

参数名 输入/输出 说明
dst 输入 目的内存地址指针。
src 输入 源内存地址指针。
count 输入 源内存大小,单位为Byte。
kind 输入 操作类型。
type 输入 数据类型。
Atlas A3 训练系列产品/Atlas A3 推理系列产品支持如下类型:int8、int16、int32、fp16、fp32、bf16。
Atlas A2 训练系列产品/Atlas A2 推理系列产品支持如下类型:int8、int16、int32、fp16、fp32、bf16。
stream 输入 指定执行Reduce操作任务的Stream。
如果使用默认Stream,此处设置为NULL。
reserve 输入 预留参数。当前固定传NULL。

返回值说明

返回0表示成功,返回其他值表示失败,请参见aclError

约束说明

dts、src必须跟stream所在的Device是同一个设备。