SetScheduleMode

Description

Sets the scheduling mode of operators when they are executed on NPUs.

Prototype

ge::graphStatus SetScheduleMode(const uint32_t schedule_mode)

Parameters

Returns

ge::GRAPH_SUCCESS on success.

ge::GRAPH_FAILED on failure.

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

Restrictions

None

Example

ge::graphStatus TilingForAdd(TilingContext *context) {
  uint32_t batch_mode = 1U;
  auto ret = context->SetScheduleMode(batch_mode);  
  GE_ASSERT_SUCCESS(ret);
  ...
}