SetBlockDim

产品支持情况

全量芯片支持。

头文件/库文件

头文件：#include <graph/kernel_launch_info.h>
库文件：libregister.so

功能说明

设置算子BlockDim。

函数原型

      
           graphStatus SetBlockDim(uint32_t block_dim)

参数说明

参数	输入/输出	说明
block_dim	输入	算子BlockDim。

返回值说明

设置成功时返回“ge::GRAPH_SUCCESS”。

关于graphStatus的定义，请参见ge::graphStatus。

约束说明

无

调用示例

      
           graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  auto aicpu_task = KernelLaunchInfo::CreateAicpuKfcTask(context,
      "libccl_kernel.so", "RunAicpuKfcSrvLaunch");
  aicpu_task.SetBlockDim(4);
  ...
}

父主题： KernelLaunchInfo