aclrtGetFunctionAddr
产品支持情况
产品 |
是否支持 |
---|---|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
|
√ |
功能说明
根据核函数句柄获取Device侧算子起始地址。
对于包含矩阵计算和矢量计算的算子,一个算子有两个起始地址,分别在Cube(矩阵)计算单元、Vector(向量)计算单元上执行,通过本接口可获取Cube计算单元、Vector计算单元上的算子起始地址。若通过本接口获取到aivAddr为空,则表示该算子只在Cube计算单元上执行。
函数原型
aclError aclrtGetFunctionAddr(aclrtFuncHandle funcHandle, void **aicAddr, void **aivAddr)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
funcHandle |
输入 |
核函数句柄。 |
aicAddr |
输出 |
Cube计算单元上的算子起始地址。 |
aivAddr |
输出 |
Vector计算单元上的算子起始地址。 |
返回值说明
返回0表示成功,返回其他值表示失败,请参见aclError。
父主题: Kernel加载与执行