开发者
资源

HcclThreadResGetInfo

产品支持情况

产品

是否支持

Atlas 350 加速卡

Atlas A3 训练系列产品/Atlas A3 推理系列产品

Atlas A2 训练系列产品/Atlas A2 推理系列产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品

Atlas 训练系列产品

功能说明

该接口用于获取Thread底层资源,例如stream等。

函数原型

1
HcclResult HcclThreadResGetInfo(HcclComm comm, ThreadHandle thread, ThreadResType resType, uint32_t infoLen, void **info)

参数说明

参数名

输入/输出

说明

comm

输入

通信域句柄。

HcclComm类型的定义如下:

1
typedef void *HcclComm;

thread

输入

线程句柄。

ThreadHandle类型的定义可参见ThreadHandle

resType

输入

底层资源类型(如STREAM等)。

ThreadResType类型的定义可参见ThreadResType

infoLen

输入

目标资源信息大小。

info

输出

资源信息输出缓冲区。

返回类型为获取的对应资源类型,目前已有资源类型定义如下:

1
typedef aclrtStream ThreadResTypeStream; //stream资源

返回值

HcclResult:接口成功返回HCCL_SUCCESS,其他失败。

约束说明

调用示例

1
2
3
4
5
6
HcclComm comm;
ThreadHandle thread;          //HcclThreadAcquire创建出来的thread的句柄
ThreadResTypeStream stream;   //info缓冲区必须按资源类型对齐且可写
uint32_t size = sizeof(ThreadResTypeStream);  // 必须等于目标类型大小  
CHK_RET(HcommThreadResGetInfo(comm, thread, ThreadResType::THREAD_RES_TYPE_STREAM, size, &stream));
//使用stream资源