Note: When using this API, ensure that the driver firmware package and CANN package are in the 8.0.RC2 version or later. Otherwise, an error, such as BUS ERROR, will be reported.
- API function: Integrates AllGather communication and MatMul compute.
- Formula:
Each operator has calls. First, [object Object] is called to obtain the workspace size required for computation and the executor that contains the operator computation process. Then, [object Object] is called to perform computation.
Parameters
[object Object]Returns
aclnnStatus: status code. For details, see .
The first-phase API implements input parameter verification. The following errors may be thrown.
[object Object]
Deterministic computing:
[object Object]defaults to a deterministic implementation.
Input
[object Object]is 2D (m, k).[object Object]must be 2D (k, n). The axes meet the input parameter requirements of the MatMul operator. Their k axes must be equal and fall within the range of [256, 65535).[object Object]and[object Object]support empty tensor scenarios. m and n can be empty, k cannot be empty, and the following conditions must be met:- m is empty, k is not empty, and n is not empty.
- If m is not empty, k is not empty, and n is empty.
- If m is empty, k is not empty, and n is empty.
The output is 2D with shape (m*rank_size, n), where rank_size indicates the number of devices.
[object Object]Atlas A2 training products/Atlas A2 inference products[object Object]: supports two, four, or eight devices and only the all mesh networking of HCCS links.
[object Object]Atlas A3 training products/Atlas A3 inference products[object Object]: supports two, four, eight, sixteen, or thirty-two devices and only the double ring networking of HCCS links.
[object Object]Atlas A2 training products/Atlas A2 inference products[object Object]: supports only one communication domain for MC2 operators within a model.
The following example is for reference only. For details, see .
[object Object]Atlas A2 training products/Atlas A2 inference products[object Object], [object Object]Atlas A3 training products/Atlas A3 inference products[object Object]:
[object Object]