SetMMLayoutTransform

Supported Products

Product

Supported/Unsupported

Atlas A3 training products/Atlas A3 inference products

Atlas A2 training products/Atlas A2 inference products

Atlas 200I/500 A2 inference products

Atlas inference product's AI Core

x

Atlas inference product's Vector Core

x

Atlas training products

x

Function Usage

Sets the direction (M or N) to be preferentially passed through during the MMAD computation.

Prototype

1
__aicore__ inline void SetMMLayoutTransform(bool mmLayoutMode)

Parameters

Table 1 Parameters

Parameter

Input/Output

Description

mmLayoutMode

Input

Controls the direction (M or N) through which the MMAD preferentially passes. The value is of the Boolean type. The options are as follows:
  • true: The CUBE passes through the N direction and then generates results in the M direction.
  • false: The CUBE passes through the M direction and then generates the result in the N direction.

Returns

None

Constraints

None

Example

1
2
bool mmLayoutMode = true;
AscendC::SetMMLayoutTransform(mmLayoutMode);