SetBlockDim

Applicability

Product	Supported
Atlas 350 Accelerator Card	√
Atlas A3 training product / Atlas A3 inference product	√
Atlas A2 training product / Atlas A2 inference product	√
Atlas 200I/500 A2 inference product	√
Atlas inference product	√
Atlas training product	√

Header File/Library File

Header file: #include <graph/kernel_launch_info.h>
Library file: libregister.so

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

Parameter	Type	Description
-	graphStatus	GRAPH_SUCCESS(0): success. Other values: failure. For details, see ge::graphStatus.

Parameter

Type

Description

graphStatus

GRAPH_SUCCESS(0): success.

Other values: failure. For details, see ge::graphStatus.

Restrictions

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