SetBlockDim

Function Usage

Sets BlockDim for an operator.

Prototype

graphStatus SetBlockDim(uint32_t block_dim)

Parameters

Parameter	Input/Output	Description
block_dim	Input	BlockDim value of the operator.

Returns

ge::GRAPH_SUCCESS on success.

For details about the definition of graphStatus, see ge::graphStatus.

Constraints

None

Examples

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);
  ...
}

Parent topic: KernelLaunchInfo