设置blockDim,即参与计算的Vector或者Cube核数。
ge::graphStatus SetBlockDim(const uint32_t block_dim)
参数 |
输入/输出 |
说明 |
---|---|---|
block_dim |
输入 |
blockDim是逻辑核的概念,取值范围为[1,65535]。为了充分利用硬件资源,一般设置为物理核的核数或其倍数。对于耦合架构和分离架构,blockDim在运行时的意义和设置规则有一些区别,具体说明如下:
|
无。
ge::graphStatus Tiling4XXX(TilingContext* context) { auto ret = context->SetBlockDim(32); // ... }