GetBlockDim

Applicability

Product	Supported or Not
Atlas A3 training products/Atlas A3 inference products	√
Atlas A2 training products/Atlas A2 inference products	√
Atlas 200I/500 A2 inference products	√
Atlas inference products	√
Atlas training products	√

Header File

#include <graph/kernel_launch_info.h>

Function Usage

Obtains the BlockDim value of an operator.

Prototype

uint32_t GetBlockDim() const

Parameters

None

Returns

BlockDim value of the operator task. The default value is 0.

If an exception occurs, int32_max is returned.

Constraints

None

Examples

graphStatus Mc2GenTaskCallback(const gert::ExeResGenerationContext *context,
    std::vector<std::vector<uint8_t>> &tasks) {
  ...
  auto aicore_task = KernelLaunchInfo::LoadFromData(context, tasks.back());
  auto block_dim = aicore_task.GetBlockDim();
  ...
}

Parent topic: KernelLaunchInfo