获得多核切分后， 使用的BlockDim。
int32_t GetCoreNum(int32_t &dim, int32_t &mDim, int32_t &nDim)
int32_t GetCoreNum(int32_t &dim, int32_t &mDim, int32_t &nDim, int32_t &batchM, int32_t &batchN)
参数名 |
输入/输出 |
描述 |
|---|---|---|
dim |
输出 |
获取计算时所需要的核数， dim = mDim * nDim |
mDim |
输出 |
获取计算时M方向所需要的核数 |
nDim |
输出 |
获取计算时N方向所需要的核数 |
batchM |
输出 |
获取计算时batch M方向所需要的核数 |
batchN |
输出 |
获取计算时batch N方向所需要的核数 |
-1表示获取失败； 0表示获取成功。
无