SetBlockDim

产品支持情况

产品	是否支持
Atlas 350 加速卡	√
Atlas A3 训练系列产品 / Atlas A3 推理系列产品	√
Atlas A2 训练系列产品 / Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	√
Atlas 推理系列产品	√
Atlas 训练系列产品	√

头文件/库文件

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

功能说明

设置算子BlockDim。

函数原型

      
           graphStatus SetBlockDim(uint32_t block_dim)

参数说明

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

返回值说明

参数名	类型	说明
-	graphStatus	GRAPH_SUCCESS(0)：成功。其他值：失败，详情请参见"ge::graphStatus"。

参数名

类型

说明

graphStatus

GRAPH_SUCCESS(0)：成功。

其他值：失败，详情请参见"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