SetMMLayoutTransform

Product Support

Product

Supported

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

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

Sets the direction (M or N) to be preferentially passed through during the Mmad computation. 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

Restrictions

None

Example

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