设置block dim。
ge::graphStatus SetBlockDim(const uint32_t block_dim);
参数 |
输入/输出 |
说明 |
---|---|---|
block_dim |
输入 |
参与计算核数 |
无
ge::graphStatus Tiling4XXX(TilingContext* context) { auto ret = context->SetBlockDim(32); // ... }