MATRIX CALCULATION APPARATUS, METHOD, SYSTEM, CIRCUIT, AND DEVICE, AND CHIP

Information

  • Patent Application
  • 20230342419
  • Publication Number
    20230342419
  • Date Filed
    June 28, 2023
    a year ago
  • Date Published
    October 26, 2023
    8 months ago
Abstract
Example matrix calculation methods and apparatus are described One example method includes calculating, based on vector outer products, a first matrix and a second matrix that are in a compressed format. Vector outer product calculation is performed on a first column vector with reserved row coordinates and a second column vector with reserved column coordinates. Third element values with same position coordinates are accumulated based on indexes of position coordinates to obtain a result matrix.
Description
CROSS-REFERENCE TO RELATED APPLICATIONS

This application is a continuation of International Application No. PCT/CN2021/141000, filed on Dec. 23, 2021, which claims priority to Chinese Patent Application No. 202110181498.6, filed on Feb. 8, 2021, and Chinese Patent Application No. 202011617575.X, filed on Dec. 30, 2020. All of the aforementioned patent applications are hereby incorporated by reference in their entireties.


TECHNICAL FIELD

This application relates to the computer field, and in particular, to a matrix calculation apparatus, method, system, circuit, and device, and a chip.


BACKGROUND

Matrix calculation is an important computing type in different application scenarios such as artificial intelligence, scientific computing, and graphics computing. A matrix is a set of element values arranged according to a rectangular array. The element values in the matrix may include two values: a zero-element value and a non-zero-element value. When there are a large quantity of zero-element values in a matrix, to save storage space, only non-zero-element values in the matrix may be stored, that is, the matrix is compressed, and a matrix in a compressed format is stored.


In a current technology, a frequently-used method for calculating a matrix in a compressed format is as follows: First, the matrix in the compressed format needs to be decompressed. To be specific, the matrix in the compressed format is converted into a matrix in an uncompressed format. Then, matrix calculation is performed on the matrix in the uncompressed format. In the matrix calculation process, because the matrix in the compressed format needs to be decompressed, and data obtained through decompression occupies large memory space, a calculation speed of the matrix is limited by access bandwidth of memory. When the access bandwidth of the memory is fixed, the calculation speed of the matrix cannot be improved, and consequently, calculation efficiency is low.


SUMMARY

This application provides a matrix calculation apparatus, method, system, circuit, and device, and a chip, to directly calculate a matrix in a compressed format without decompressing the matrix in the compressed format, thereby improving calculation efficiency of the matrix in the compressed format.


According to a first aspect, an embodiment of this application provides a matrix calculation apparatus, where the matrix calculation apparatus includes a vector outer product processing engine and an accumulator. The vector outer product processing engine is configured to calculate vector outer products of N first column vectors and N first row vectors, to obtain N intermediate result matrices, where the first column vector includes first element values and row coordinates of the first element values, the first row vector includes second element values and column coordinates of the second element values, the intermediate result matrix includes third element values and position coordinates of the third element values, the position coordinates include the row coordinates of the first element values and the column coordinates of the second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1. The accumulator is configured to accumulate, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix. In this embodiment of this application, the matrix calculation apparatus calculates the first matrix and the second matrix in the compressed format based on the vector outer products. In the calculation process, row coordinates of element values in the first column vector are reserved, column coordinates of element values in the second column vector are reserved, and then third element values with same position coordinates are accumulated based on indexes of position coordinates, to obtain the result matrix obtained by calculating the two matrices in the compressed format. Compared with a conventional method in which a matrix in a compressed format needs to be first decompressed and then matrix calculation is performed on a matrix obtained through decompression, the matrix calculation method provided in embodiments of this application can effectively improve calculation efficiency of a matrix in a compressed format.


In an optional implementation, the N intermediate result matrices include at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates. The accumulator is configured to: write, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates; and then read, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache, and accumulate the third element values in the second intermediate result matrix and the cached values, to obtain a result matrix in an uncompressed format. In the foregoing optional implementation, the matrix calculation apparatus may output a matrix in an uncompressed format based on different application scenarios, thereby expanding an applicable scenario of the matrix calculation apparatus.


In an optional implementation, the matrix calculation apparatus further includes a matrix compression unit. The matrix compression unit is configured to compress the result matrix in the uncompressed format to obtain a result matrix in a compressed format. In the foregoing optional implementation, the matrix calculation apparatus may output a matrix in a compressed format based on different application scenarios, thereby expanding an applicable scenario of the matrix calculation apparatus. In addition, the matrix calculation apparatus compresses the result matrix, and outputs the matrix in the compressed format, thereby saving transmission resources or facilitating a next calculation operation.


In an optional implementation, the accumulator is further specifically configured to: sort third element values in the N intermediate result matrices based on position coordinates of the third element values, for example, sort the third element values based on row coordinates of the third element values, or sort the third element values based on column coordinates of the third element values; and then compare position coordinates in N intermediate result matrices obtained through sorting, add up third element values with same position coordinates, and delete position coordinates of a zero-element value, to obtain a result matrix in a compressed format. In the foregoing optional implementation, the matrix in the compressed format may be directly obtained, and the matrix in the compressed format output by the matrix calculation apparatus may be used in some application scenarios in which a matrix in a compressed format is required for subsequently calculation. In addition, because the matrix calculation apparatus outputs the matrix in the compressed format, transmission resources of a subsequent transmission matrix can be reduced.


In an optional implementation, the matrix calculation apparatus further includes a format conversion unit. The format conversion unit is configured to: obtain the first matrix and the second matrix; convert the first matrix into the N first column vectors and reserve the row coordinates of the first element values in the first column vectors; and convert the second matrix into the N first row vectors and reserve the column coordinates of the first element values in the first row vectors. In this way, the matrix calculation apparatus can calculate the two matrices in the compressed format based on the vector outer products.


In an optional implementation, the matrix calculation apparatus further includes the format conversion unit. The format conversion unit is further configured to: obtain a fifth matrix and a sixth matrix, perform format conversion on the fifth matrix to obtain the first matrix, and perform format conversion on the sixth matrix to obtain the second matrix, where at least one of the fifth matrix and the sixth matrix is a matrix in an uncompressed format. In the foregoing optional implementation, the matrix calculation apparatus may receive the matrix in the uncompressed format, and then convert the matrix in the uncompressed format into the matrix in the compressed format, so that the matrix calculation apparatus can support calculation of matrices in a plurality of formats.


In an optional implementation, the matrix calculation apparatus further includes the format conversion unit. The format conversion unit is further configured to split the first column vector into X second column vectors, and split the first row vector into X second row vectors, where precision of element values included in the second column vector and the second row vector is second precision, precision of element values included in the first column vector and the first row vector is first precision, the first precision is higher than the second precision, and X is an integer greater than or equal to 2. The vector outer product processing engine is further configured to calculate vector outer products of the X second column vectors and the X second row vectors to obtain X2 fourth matrices, where the fourth matrix includes fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values include the row coordinates of the first element values and the column coordinates of the second element values, and precision of the fourth element values is the first precision. Then, the accumulator is further configured to accumulate, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices, to obtain the intermediate result matrix, where precision of the third element values in the intermediate result matrix is the first precision. In the foregoing optional implementation, the matrix calculation apparatus may implement high-precision matrix calculation based on a low-precision matrix calculation apparatus, thereby improving applicability of the matrix calculation unit.


According to a second aspect, an embodiment of this application provides a matrix calculation method, where the method is applied to a matrix calculation apparatus, and the method includes: first, obtaining a first calculation instruction, where the first calculation instruction includes N first column vectors and N first row vectors; then, calculating vector outer products of the N first column vectors and the N first row vectors, to obtain N intermediate result matrices, where the first column vector includes first element values and row coordinates of the first element values, the first row vector includes second element values and column coordinates of the second element values, the intermediate result matrix includes third element values and position coordinates of the third element values, the position coordinates include the row coordinates of the first element values and the column coordinates of the second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1; and finally, accumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix. In this embodiment of this application, the first matrix and the second matrix are calculated based on the vector outer products of the N first column vectors and the N first row vectors. In the calculation process, row coordinates of element values in the first column vector are reserved, column coordinates of element values in the second column vector are reserved, and then third element values with same position coordinates are accumulated based on indexes of position coordinates, to obtain the result matrix obtained by calculating the two matrices in the compressed format. Compared with a conventional method in which a matrix in a compressed format needs to be first decompressed and then matrix calculation is performed on a matrix obtained through decompression, the matrix calculation apparatus provided in embodiments of this application can effectively improve calculation efficiency of a matrix in a compressed format.


In an optional implementation, the N intermediate result matrices include at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates. In the method, the accumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix may specifically include: first, writing, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates; and then reading, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache, and accumulating the third element values in the second intermediate result matrix and the cached values, to obtain a result matrix in an uncompressed format. In the foregoing optional implementation, the matrix calculation apparatus may output a matrix in an uncompressed format based on different application scenarios, thereby expanding an applicable scenario of the matrix calculation apparatus.


In an optional implementation, the method further includes: compressing the result matrix in the uncompressed format to obtain a result matrix in a compressed format. In the foregoing optional implementation, the matrix calculation apparatus may output a matrix in a compressed format based on different application scenarios, thereby expanding an applicable scenario of the matrix calculation apparatus. In addition, the matrix calculation apparatus compresses the result matrix, and outputs the matrix in the compressed format, thereby saving transmission resources or facilitating a next calculation operation.


In an optional implementation, in the method, the accumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix may specifically include: first, sorting third element values in the N intermediate result matrices based on position coordinates of the third element values, for example, sorting the third element values based on row coordinates of the third element values, or sorting the third element values based on column coordinates of the third element values; and comparing position coordinates in N intermediate result matrices obtained through sorting, adding up third element values with same position coordinates, and deleting position coordinates of a zero-element value, to obtain a result matrix in a compressed format. In the foregoing optional implementation, the matrix in the compressed format may be directly obtained, and the matrix in the compressed format output by the matrix calculation apparatus may be used in some application scenarios in which a matrix in a compressed format is required for subsequently calculation. In addition, because the matrix calculation apparatus outputs the matrix in the compressed format, transmission resources of a subsequent transmission matrix can be reduced.


In an optional implementation, before the obtaining a first calculation instruction, the method further includes: obtaining a second calculation instruction, where the second calculation instruction includes the first matrix and the second matrix; and converting the first matrix into the N first column vectors and reserving the row coordinates of the first element values in the first column vectors; and converting the second matrix into the N first row vectors and reserving the column coordinates of the first element values in the first row vectors. In this way, the matrix calculation apparatus can calculate the two matrices in the compressed format based on the vector outer products.


In an optional implementation, before the obtaining a second calculation instruction, the method further includes: obtaining a third calculation instruction, where the third calculation instruction includes a fifth matrix and a sixth matrix, and at least one of the fifth matrix and the sixth matrix is a matrix in an uncompressed format; and then performing format conversion on the fifth matrix to obtain the first matrix in the compressed format, and performing format conversion on the sixth matrix to obtain the second matrix. In the foregoing optional implementation, the matrix calculation apparatus may receive the matrix in the uncompressed format, and then convert the matrix in the uncompressed format into the matrix in the compressed format, so that the matrix calculation apparatus can support calculation of matrices in a plurality of formats.


In an optional implementation, the first column vector may be first split into X second column vectors, and the first row vector is split into X second row vectors, where precision of element values included in the second column vector and the second row vector is second precision, precision of element values included in the first column vector and the first row vector is first precision, the first precision is higher than the second precision, and X is an integer greater than or equal to 2. In the method, the calculating vector outer products of the N first column vectors and the N first row vectors, to obtain N intermediate result matrices may include: calculating vector outer products of the X second column vectors and the X second row vectors to obtain X2 fourth matrices, where the fourth matrix includes fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values include the row coordinates of the first element values and the column coordinates of the second element values, and precision of the fourth element values is the first precision; and then, accumulating, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices, to obtain the intermediate result matrix, where precision of the third element values in the intermediate result matrix is the first precision. In the foregoing optional implementation, the matrix calculation apparatus may implement high-precision matrix calculation based on a low-precision matrix calculation apparatus, thereby improving applicability of the matrix calculation unit.


According to a third aspect, a matrix calculation circuit is provided. The matrix calculation circuit is configured to perform operation steps of the matrix calculation method according to any one of the second aspect or the possible implementations of the second aspect.


According to a fourth aspect, a matrix calculation system is provided. The system includes a processor and a matrix calculation apparatus. The processor is configured to send a calculation instruction to the matrix calculation apparatus. The matrix calculation apparatus is configured to perform operation steps of the matrix calculation method according to any one of the second aspect or the possible implementations of the second aspect.


According to a fifth aspect, a chip is provided. The chip includes a processor, a matrix calculation apparatus is integrated into the processor, and the matrix calculation apparatus is configured to perform operation steps of the matrix calculation method according to any one of the second aspect or the possible implementations of the second aspect.


According to a sixth aspect, a matrix calculation device is provided. The device includes the matrix calculation system according to the fourth aspect or the chip according to the fifth aspect.


According to a seventh aspect, a readable storage medium is provided. The readable storage medium stores instructions. When the readable storage medium runs on a device, the device is enabled to perform operation steps of the matrix calculation method according to any one of the second aspect or the possible implementations of the second aspect.


According to an eighth aspect, a computer program product is provided. When the computer program product runs on a computer, the computer is enabled to perform operation steps of the matrix calculation method according to any one of the second aspect or the possible implementations of the second aspect.


It may be understood that any apparatus, computer storage medium, or computer program product for implementing the matrix calculation method provided above is configured to perform the corresponding method provided above. Therefore, for beneficial effects that can be achieved by the apparatus, the computer storage medium, or the computer program product, refer to beneficial effects of the corresponding method provided above. Details are not described herein again.


In this application, based on the implementations according to the foregoing aspects, the implementations may be further combined to provide more implementations.





BRIEF DESCRIPTION OF DRAWINGS


FIG. 1A is a schematic diagram of a matrix in a COO compressed format according to an embodiment of this application;



FIG. 1B is a schematic diagram of a matrix in a CSR compressed format according to an embodiment of this application;



FIG. 1C is a schematic diagram of a matrix in a CSC compressed format according to an embodiment of this application;



FIG. 2 is a schematic diagram of a structure of a computing device according to an embodiment of this application;



FIG. 3 is a schematic diagram of a structure of a processor according to an embodiment of this application;



FIG. 4A is a schematic diagram of a structure of a matrix calculation apparatus according to an embodiment of this application;



FIG. 4B is a schematic diagram of a structure of a vector outer product processing engine according to an embodiment of this application;



FIG. 4C is a schematic diagram of a structure of a MAC operation subunit according to an embodiment of this application;



FIG. 4D is a schematic diagram of a structure of an accumulator according to an embodiment of this application;



FIG. 4E is a schematic diagram of a structure of an adder according to an embodiment of this application;



FIG. 5A is a schematic diagram in which a matrix calculation apparatus converts a first matrix into N first column vectors and a second matrix into a plurality of first row vectors according to an embodiment of this application;



FIG. 5B is a schematic diagram in which a matrix calculation apparatus performs outer product calculation on N first column vectors and N first row vectors to obtain N intermediate result matrices according to an embodiment of this application;



FIG. 5C is a schematic diagram of element values and position coordinates in a first intermediate result matrix according to an embodiment of this application;



FIG. 5D is a schematic diagram of element values and position coordinates in a second intermediate result matrix according to an embodiment of this application;



FIG. 5E is a schematic diagram in which a matrix calculation apparatus accumulates third element values with same position coordinates in N intermediate result matrices to obtain a result matrix according to an embodiment of this application;



FIG. 6 is a schematic diagram of an implementation of accumulating third element values with same position coordinates in N intermediate result matrices to obtain a result matrix according to an embodiment of this application;



FIG. 7 is a schematic diagram of a structure of another matrix calculation apparatus according to an embodiment of this application;



FIG. 8 is a schematic diagram of another implementation of accumulating third element values with same position coordinates in N intermediate result matrices to obtain a result matrix according to an embodiment of this application;



FIG. 9 is a schematic diagram in which a format conversion unit converts a matrix in an uncompressed format into a matrix in a compressed format according to an embodiment of this application;



FIG. 10 is a schematic diagram of dividing an integer number of first precision into a plurality of integer numbers of second precision according to an embodiment of this application;



FIG. 11 is a schematic diagram of splitting a first column vector of first precision into a plurality of second column vectors of second precision and a first row vector of the first precision into a plurality of second row vectors of the second precision according to an embodiment of this application;



FIG. 12 is a schematic diagram in which a matrix calculation apparatus performs outer product calculation on a second row vector and a second column vector to obtain a fourth matrix according to an embodiment of this application;



FIG. 13 is a schematic flowchart of steps of an embodiment of a matrix calculation method according to an embodiment of this application; and



FIG. 14 is a schematic flowchart of steps of another embodiment of a matrix calculation method according to an embodiment of this application.





DESCRIPTION OF EMBODIMENTS

The following describes the technical solutions in embodiments of this application with reference to the accompanying drawings in embodiments of this application. In the specification, claims, and accompanying drawings of this application, the terms “first”, “second”, and the like are intended to distinguish between similar objects but do not necessarily indicate a specific order or sequence.


To better understand this application, related terms in this application are first described.


Matrix (matrix): A matrix whose dimension is m×n is a rectangular array obtained by arranging elements of m rows (rows) and n columns (columns). For example, a matrix A is shown in Formula (1), and a matrix B is shown in Formula (2):









A
=

[




a

1

1





a

1

2








a

1

n







a

2

1





a

2

2








a

2

n





















a

m

1





a

m

2








a

m

n





]





Formula



(
1
)













B
=

[




b

1

1





b

1

2








b

1

n







b

2

1





b

2

2








b

2

n





















b

m

1





b

m

2








b

m

n





]





Formula



(
2
)








Matrix addition and subtraction: Mutual addition and subtraction may be performed between matrices of a same dimension, and specifically, addition and subtraction are performed on elements at all positions. For example, both the matrix A and the matrix B are matrices in an m×n dimension, the matrix A and the matrix B are added to obtain a matrix C, and the matrix C is shown in Formula (3):









C
=


A
+
B

=

[





a

1

1


+

b

1

1







a

1

2


+

b

1

2










a

1

n


+

b

1

n









a

2

1


+

b

2

1







a

2

2


+

b

2

2










a

2

n


+

b

2

n























a

m

1


+

b

m

1







a

m

2


+

b

m

2










a

m

n


+

b

m

n






]






Formula



(
3
)








Matrix multiplication: Two matrices can be multiplied only when a quantity of columns (columns) of a first matrix A is equal to a quantity of rows (rows) of the other matrix. For example, the matrix A is a matrix whose dimension is m×n, the matrix B is a matrix whose dimension is n×p, a product of the matrix A and the matrix B is an m×p matrix, and an element in the matrix m×p is shown in Formula (4):












[

a

b

]


i

j


=




a

i
,
1




b

1
,
j



+


a

i
,
2




b

2
,
j



+



+


a

i
,
n




b

n
,
j




=




r
=
1

n



a

i
,
r




b

r
,
j






,




Formula



(
4
)








where 1≤i≤m and 1≤j≤p.


A row vector (row vector) is a matrix whose dimension is 1×m , where m is a positive integer. For example, a row vector is shown in Formula (5):






X=[x
1
x
2
. . . x
m]  Formula (5)


A column vector (column vector) is a matrix whose dimension is m×1, and m is a positive integer. For example, a column vector is shown in Formula (6):









X
=

[




x
1






x
2











x
m




]





Formula



(
6
)








A vector outer product (vector outer product) is a tensor product of two vectors. The tensor product is a matrix. For example, a column vector U whose dimension is m×1 and a row vector V whose dimension is 1×n are given. An outer product U×V of the vector U and the vector V is defined as a matrix D whose dimension is m×n, and the matrix D is shown in Formula (7):










U

V

=



[




ι


ι
1







ι


ι
2







ι


ι
3







ι


ι
4





]



[


v
1




v
2




v
3


]


=

[





v
1


ι


ι
1






v
2


ι


ι
1






v
3


ι


ι
1








v
1


ι


ι
2






v
2


ι


ι
2






v
3


ι


ι
2








v
1


ι


ι
3






v
2


ι


ι
3






v
3


ι


ι
3








v
1


ι


ι
4






v
2


ι


ι
4






v
3


ι


ι
4





]






Formula



(
7
)








Matrix in a compressed format: When a matrix includes zero-element values and non-zero-element values, usually, the non-zero-element values in the matrix may be stored in a specific format and the zero-element values are not stored, to save storage space. In this process, the matrix is compressed, and a matrix obtained after compression and storage is referred to as a matrix in a compressed format. A method for compressing a matrix includes but is not limited to coordinate (coordinate, COO), compressed sparse row (compressed sparse row, CSR), compressed sparse column (compressed sparse column, CSC), and the like.


The following separately uses examples to describe the three compression methods: COO, CSR, and CSC.


COO: A matrix is represented by a triplet. The triplet includes three values: a row number, a column number, and an element value. The row number and the column number are for identifying a position of the element value. For example, the triplet is (row number, column number, element value), or the triplet is (element value, row number, column number). Specifically, an arrangement sequence of the three values in the triplet is not limited. For example, refer to FIG. 1A. FIG. 1A shows a matrix Y whose dimension is 4×4. The matrix includes zero-element values and non-zero-element values. The non-zero-element values are 1, 2, 3, 4, 5, 6, 7, 8, and 9. For example, a position of the non-zero-element value “1” is row 0 column 0, and the triplet is represented as (0, 0, 1). A position of the non-zero-element value “2” is row 0 column 1, and the triplet is represented as (0, 1, 2). A position of the non-zero-element value “3” is row 1 column 1, and the triplet is represented as (1, 1, 3). Not all element values are described in detail herein. For example, triplet forms of a matrix in a compressed format are (0, 0, 1), (0, 1, 2), (1, 1, 3), (1, 2, 4), (2, 0, 5), (2, 2, 6), (2, 3, 7), (3, 1, 8), and (3, 3, 9).


Optionally, the matrix Y in the compressed format may be represented in Formula (8):





Row coordinate=([0, 0,1,1,2,2,2,3,3])





Column coordinate=([0,1,1,2,0,2,3,1,3])





Element value=([1,2,3,4,5,6,7,8,9])   Formula (8)


CSR: A matrix is represented by three types of data: an element value, a column number, and a row offset. The element value and the column number in CSR are represented in manners similar to the element value and the column number in the COO method described above. CSR is different from the COO method in that the row offset indicates a start offset position of the 1st element in a row in all element values. Refer to FIG. 1B. First, non-zero-element values in a matrix Y shown in FIG. 1B are arranged by row, to obtain all element values: 1, 2, 3, 4, 5, 6, 7, 8, and 9. The Pt non-zero-element value in the first row is “1”, and an offset of the element value “1” in all the element values is “0”. Similarly, the 1st non-zero-element value in the second row is “3”, and an offset of the element value “3” in all the element values is “2”. The 1st non-zero-element value in the third row is “5”, and an offset of the element value “5” in all the element values is “4”. The 1st non-zero-element value in the fourth row is “8”, and an offset of the element value “8” in all the element values is “7”. Finally, a total quantity (for example, “9”) of non-zero-element values in the matrix is padded at the end of a row to which the row offset belongs.


The matrix Y in the compressed format may be represented in Formula (9):





Row offset=([0, 2, 4, 7, 9])





Column coordinate=([0,1,1, 2, 0, 2, 3,1, 3])





Element value=([1, 2, 3, 4, 5, 6, 7, 8, 9])   Formula (9)


CSC: A matrix is represented by three types of data: an element value, a row number, and a column offset. The element value and the row number in CSR are represented in manners similar to the element value and the row number in the COO method described above. CSC is different from the COO method in that the column offset indicates a start offset position of the 1st element in a column in all element values. Refer to FIG. 1C. First, non-zero-element values in a matrix Y shown in FIG. 1C are arranged by column, to obtain all element values: 1, 5, 2, 3, 8, 4, 6, 7, and 9. The 1st non-zero-element value in the first column is “1”, and an offset of the element value “1” in all the element values is “0”. Similarly, the Pt non-zero-element value in the second column is “5”, and an offset of the element value “5” in all the element values is “2”. The 1st non-zero-element value in the third column is “2”, and an offset of the element value “2” in all the element values is “5”. The Pt non-zero-element value in the fourth column is “3”, and an offset of the element value “3” in all the element values is “7”. Finally, a total quantity (for example, “9”) of non-zero-element values in the matrix is padded at the end of a row to which the column offset belongs.


The matrix Y in the compressed format may be represented in Formula (10):





Column offset=([0, 2, 5, 7, 9])





Row coordinate=([0, 0,1,1, 2, 2, 2, 3, 3])





Element value=([1, 5, 2, 3, 8, 4, 6,7, 9]) Formula (10)


It can be learned from the descriptions of the foregoing three matrix compression methods that, each element value in the matrix in the COO compressed format has a corresponding row coordinate (row number) and a corresponding column coordinate (column number), each element value in the matrix in the CSR compressed format has a corresponding column coordinate, and each element value in the matrix in the CSC compressed format has a corresponding row coordinate.


Matrix in an uncompressed format: Refer to the matrix Y shown in FIG. 1A. A matrix in an uncompressed format includes zero-element values and non-zero-element values. It should be noted that, usually, the matrix in the compressed format is also referred to as a sparse matrix, and the matrix in the uncompressed format may also be referred to as a dense matrix.


Numeric type: In the computer field, numeric types include an integer and a float. The integer mainly indicates an integer number, and the float mainly indicates a decimal. Integer precision includes int2, int4, int8, int16, int32, and the like. int represents an integer function, and a numerical digit added after int represents a quantity of bits (bits) of a binary value range, and a bit (bit) is 0 or 1. For example, a binary value range for int4 is 4 bits (0000 to 1111), and a converted decimal value range is (−8, 7). Similarly, a binary value range for int8 is (−27, 27−1). A unit of a computer storage capacity is 1 byte, namely, 8 bits. Therefore, correspondingly, there is 1 byte for int8 and 2 bytes for int16. A binary value range for int16 occupies 2 bytes, and a converted decimal value range is (−32768, 32767). A binary value range for int32 occupies 4 bytes, and a converted decimal value range is (−2147483648, 2147483647).


An integer matrix is a matrix that uses integer numbers as elements. For example, an integer matrix with m rows and n columns includes m×n elements, and the m×n elements are integer numbers. The integer number may be of precision such as int2, int4, int8, int16, or int32. For example, the integer matrix may alternatively include matrices in different integer formats, for example, a matrix including integer numbers in an int8 format, a matrix including integer numbers in an int16 format, and a matrix including integer numbers in an int32 format.


A floating-point (floating-point, FP) mainly represents a decimal number, and usually includes three parts: a sign (sign) bit, an exponent (exponent) field, and a mantissa (mantissa) field. The exponent field may also be referred to as an exponent field. The sign bit may be 1 bit (bit), and the exponent field and the mantissa field may be a plurality of bits (bits). The floating-point may usually include a plurality of formats (format), such as a half-precision floating-point, a single-precision floating-point, and a double-precision floating-point in the IEEE 754 standard. The half-precision floating-point (half-precision floating-point) occupies 16 bits (that is, occupies 2 bytes) in a computer memory, and may also be referred to as FP16 for short. An absolute value range of a value that can be represented by the half-precision floating-point is approximately [6.10×10−5, 6.55×104]. The single-precision floating-point (single-precision floating-point) occupies 32 bits (that is, 4 bytes) in the computer memory, and may also be referred to as FP32 for short. An absolute value range of a value that can be represented by the single-precision floating-point is approximately [1.18×10−38, 3.40×1038]. The double-precision floating-point (double precision floating point) occupies 64 bits (that is, occupies 8 bytes) in the computer memory, and may also be referred to as FP64 for short. The double-precision floating-point may represent a 15-digit or 16-digit decimal number. An absolute value range of a value that can be represented by the double-precision floating-point is approximately [2.23×10−308, 1.80'10308].


Table 1 below shows a storage format of the foregoing three types of floating-points. In the 16 bits occupied by FP16, a sign bit occupies 1 bit, an exponent occupies 5 bits, and a mantissa field occupies 10 bits. In the 32 bits occupied by FP32, a sign bit occupies 1 bit, an exponent occupies 8 bits, and a mantissa field occupies 23 bits. In the 64 bits occupied by FP64, a sign bit occupies 1 bit, an exponent occupies 11 bits, and a mantissa field occupies 52 bits.













TABLE 1







Sign bit
Exponent field (exponent)
Mantissa field




















FP16
1 bit
5
bits
10 bits


FP32
1 bit
8
bits
23 bits


FP64
1 bit
11
bits
52 bits









The floating-point matrix may be a matrix that uses floating-points as elements. For example, a floating-point matrix with m rows and n columns includes m×n elements, and the m×n elements may be floating-points. Similar to the floating-point, the floating point matrix may also include matrices in different floating-point formats, for example, a matrix including floating-points in an FP16 format, a matrix including floating-points in an FP32 format, and a matrix including floating-points in an FP64 format.



FIG. 2 is a schematic diagram of a structure of a computing device according to an embodiment. The computing device may be a device having a computation capability, such as a terminal, a network device, or a server. Refer to FIG. 2. The computing device may include a memory 201, a processor 202, a communication interface 203, and a bus 204. The memory 201, the processor 202, and the communication interface 203 are connected to each other by using the bus 204.


The memory 201 may be configured to store data, a software program, and a module, and mainly includes a program storage area and a data storage area. The program storage area may store an operating system, a software application required by at least one function, middleware, and the like. The data storage area may store data created when the device is used, and the like. For example, the operating system may include a Linux operating system, a Unix operating system, a Window operating system, or the like. The software application required by the at least one function may include an application related to artificial intelligence (artificial intelligence), an application related to high-performance computing (high-performance computing, HPC), an application related to deep learning (deep learning), an application related to scientific computing, or the like. The middleware may include a linear algebra library function or the like. In a possible example, the memory 201 includes but is not limited to a static random access memory (static RAM, SRAM), a dynamic random access memory (dynamic RAM, DRAM), a synchronous dynamic random access memory (synchronous DRAM, SDRAM), a high-speed random access memory, or the like. Further, the memory 201 may further include another non-volatile memory, for example, at least one magnetic disk storage device, a flash storage device, or another volatile solid-state storage device.


In addition, the processor 202 is configured to control and manage an operation of the computing device, for example, perform various functions of the computing device and process data by running or executing a software program and/or a module that are/is stored in the memory 201 and invoking data stored in the memory 201. In a possible example, the processor 202 includes but is not limited to a central processing unit (central processing unit, CPU), a network processing unit (network processing unit, NPU), a graphics processing unit (graphics processing unit, GPU), an application-specific integrated circuit (application-specific integrated circuit, ASIC), a field programmable gate array (field programmable gate array, FPGA) or another programmable logic device, a transistor logic device, a logic circuit, or any combination thereof. The processor may implement or execute various example logical blocks, modules, and circuits described with reference to content disclosed in this application. Alternatively, the processor 202 may be a combination for implementing a computation function, for example, a combination of one or more microprocessors, or a combination of a digital signal processor and a microprocessor.


The communication interface 203 is configured to implement communication between the computing device and an external device. The communication interface 203 may include an input interface and an output interface. The input interface may be configured to obtain a first matrix and a second matrix in a compressed format in the following embodiments. In some feasible embodiments, there may be only one input interface, or there may be a plurality of input interfaces. The output interface may be configured to output a result matrix in the following embodiments. In some feasible embodiments, the result matrix may be directly output by the processor, or may be first stored in the memory and then output from the memory. In some other feasible embodiments, there may be only one output interface, or there may be a plurality of output interfaces.


The bus 204 may be a peripheral component interconnect (Peripheral Component Interconnect Express, PCIe) bus, an extended industry standard architecture (extended industry standard architecture, EISA) bus, or the like. The bus 204 may be classified into an address bus, a data bus, a control bus, and the like. For ease of representation, only one thick line is used to represent the bus in FIG. 2, but this does not mean that there is only one bus or only one type of bus.


In this embodiment, the processor 202 may include a matrix calculation apparatus. The matrix calculation apparatus may be an ASIC, an FPGA, a logic circuit, or the like. Certainly, the apparatus may alternatively be implemented by using software. This is not specifically limited in this embodiment of this application. The matrix calculation apparatus may be configured to compute a matrix related to artificial intelligence, scientific computing, graphics computing, and the like.


Further, the processor 202 may alternatively include one or more of other processing units such as a CPU, a GPU, or an NPU. As shown in FIG. 3, an example in which the processor 202 includes a CPU 1 and a matrix calculation apparatus 2 is used. The matrix calculation apparatus 2 may be integrated with the CPU 1 (for example, the matrix calculation apparatus 2 is integrated inside an SoC in which the CPU 1 is located), or may be disposed in parallel with the CPU 1 separately (for example, the matrix calculation apparatus 2 is disposed in a form of a PCIe card), as specifically shown in (a) in FIG. 3 and (b) in FIG. 3. Further, the CPU 1 may further include a controller (controller) 11, one or more arithmetic logic units (arithmetic logic units, ALUs) 12, a cache (cache) 13, a memory management unit (memory management unit, MMU) 14, and the like. In FIG. 3, an example in which the memory 201 is a dynamic random access memory DRAM is used for description.


In this embodiment of this application, the matrix calculation apparatus can calculate a matrix in a compressed format. When performing multiplication calculation on two matrices in a compressed format, the matrix calculation apparatus first obtains N first column vectors converted from one of the matrices, where the first column vector includes first element values and row coordinates of the first element values, and obtains N first row vectors converted from the other matrix, where the first row vector includes second element values and column coordinates of the second element values. The matrix calculation apparatus calculates vector outer products of the N first column vectors and the N first row vectors, to obtain N intermediate result matrices. The intermediate result matrix includes element values and position coordinates corresponding to the element values, and the position coordinates include the row coordinates of the first element values and the column coordinates of the second element values. The matrix calculation apparatus can accumulate, based on indexes of the position coordinates, third element values with same position coordinates in the N intermediate result matrices, and then obtain a result matrix. In this embodiment of this application, the matrix calculation apparatus calculates a first matrix and a second matrix based on vector outer products. In the calculation process, row coordinates of element values in the first column vector are reserved, column coordinates of element values in the second column vector are reserved, and then third element values with same position coordinates are accumulated based on indexes of position coordinates, to obtain the result matrix obtained by calculating the two matrices in the compressed format. Compared with a conventional method in which a matrix in a compressed format needs to be first decompressed and then matrix calculation is performed on a matrix obtained through decompression, the matrix calculation apparatus provided in embodiments of this application can effectively improve calculation efficiency of a matrix in a compressed format.


An embodiment of this application provides a matrix calculation apparatus. Refer to FIG. 4A. The matrix calculation apparatus includes a vector outer product processing engine 401 and an accumulator 402. Optionally, the matrix calculation apparatus includes a first cache 403, and the vector outer product processing engine 401, the accumulator 402, and the first cache 403 are sequentially connected. The first cache 403 may be a first cache 403 (for example, a register) in the matrix calculation apparatus, or the first cache 403 may be a cache 13 in the central processing unit 1 shown in FIG. 3. Optionally, the matrix calculation apparatus further includes a format conversion unit 405 and a second cache 406. A function of the format conversion unit 405 may be implemented by the central processing unit 1 in FIG. 3, or a function of the format conversion unit 405 may be implemented by a logic circuit in the matrix calculation apparatus. The second cache 406 may be a register in the matrix calculation apparatus, or the second cache 406 may be the cache 13 in the central processing unit 1 shown in FIG. 3.


The format conversion unit 405 is configured to: convert a first matrix in a compressed format into N first column vectors, and convert a second matrix in the compressed format into N first row vectors, where the first column vector includes first element values and row coordinates of the first element values, and the first row vector includes second element values and column coordinates of the second element values.


Optionally, the following describes a structure of the vector outer product processing engine 401. Refer to FIG. 4B. FIG. 4B is a schematic diagram of the structure of the vector outer product processing engine 401. The vector outer product processing engine 401 includes a plurality of processing elements (processing elements, PEs) 4011, the plurality of PEs 4011 are a two-dimensional array, and each PE 4011 includes a multiply accumulate (multiply accumulate, MAC) operation subunit 40110 and a coordinate combination subunit 40111. Each PE receives two groups of input data and outputs one group of data. Each group of input data includes one value (value) and one coordinate (index). One group of input data includes a first element value and a row coordinate corresponding to the first element value, and the other group of input data includes a second element value and a column coordinate corresponding to the second element value.


Each PE has two functions. One function is that each PE 4011 receives two groups of input data, and outputs one group of output data based on the groups of input data. The output data includes a third element value and position coordinates of the third element value, where the third element value is obtained by performing a multiplication operation on the first element value and the second element value. The position coordinates are obtained by combining a first row coordinate and a first column coordinate. For example, a PE in row 0 column 0 in a PE array is used as an example. One group of input data is a first element value (for example, a0) in a first row vector and a row coordinate (for example, i0) corresponding to the first element value. The other group of input data is a second element value (for example, b0) in a first column vector and a column coordinate (for example, j0) corresponding to the second element value. The MAC operation subunit 40110 is configured to: receive the first element value and the second element value, perform multiplication calculation on the first element value and the second element value, and then output a product (that is, a third element value) of the first element value and the second element value. The coordinate combination subunit 40111 is configured to: receive the row coordinate of the first element value and the column coordinate of the second element value, and combine the row coordinate and the column coordinate to obtain the position coordinates. Data output by a PE includes a third coordinate value and position coordinates corresponding to the third coordinate value.


The other function of the PE is that the PE needs to transmit a first element value and a column coordinate corresponding to the first element value to a next PE in a row direction, and the PE needs to transmit a second element value and a row coordinate corresponding to the second element value to a next PE in a column direction. For example, after the 1st clock cycle, the Pt PE (that is, the PE in row 0 column 0) in the PE array transmits a0 and i0 to a next PE in the row direction (for example, a PE in row 0 column 1), and transmits b0 and j0 to a next PE in the column direction (that is, a PE in row 1 column 0). Optionally, a data transmission mode of each PE may be: transmitting data to a next PE in each clock cycle; or in a clock cycle, the 1st PE transmits a0 and i0 to a next PE in the row direction (that is, a PE in row 0 column 1) until the data is transmitted to the last PE in the row (that is, row 0), and transmits b0 and jo to the last PE in the column (column 0). In this example, a quantity of levels of PEs for transmission in each clock unit may be designed based on an actual requirement, and is not specifically limited.


Optionally, the following describes a structure of the MAC operation subunit in the PE. Refer to FIG. 4C. FIG. 4C is a schematic diagram of the structure of the MAC operation subunit. The MAC operation subunit in each PE includes a sign subunit 40112, an exponent subunit 40113, an integer subunit 40114, and a precision format conversion subunit 40115. The sign subunit 40112, the exponent subunit 40113, and the integer subunit 40114 all are connected to the precision format conversion subunit 40115. The sign subunit 40112 is configured to process a positive sign and a negative sign of an input value. The exponent subunit 40113 is configured to process decimal point displacement calculation when two floating-points are multiplied. The integer subunit 40114 is configured to perform multiplication calculation on two integers. The precision format conversion subunit 40115 is configured to output a value format (for example, a floating-point format such as FP16, FP32, and FP64, or an integer format such as int32, int16, and int8) that complies with a specification. In this example, the MAC operation subunit is a general-purpose unit that supports the floating-point format and the integer format. This enhances applicability of the matrix calculation apparatus.


Optionally, the following describes a structure of the accumulator 402 in the matrix calculation apparatus. Refer to FIG. 4D. FIG. 4D is a schematic diagram of the structure of the accumulator 402. The accumulator 402 includes a plurality of accumulator (Accumulator, ACC) processing elements 4021. In other words, the accumulator 402 includes an ACC processing element array including the plurality of ACC processing elements 4021. Each ACC processing element 4021 receives a group of data output by the vector outer product processing engine 401, where the group of data includes one third element value and position coordinates corresponding to the third element value. Each ACC processing element 4021 includes an adder 40210 and a data obtaining subunit 40211. The data obtaining subunit 40211 is configured to: receive the position coordinates, and output the position coordinates to the first cache 403, so that the adder 40210 obtains, from the first cache 403, a cached value corresponding to the position coordinates. The adder 40210 is configured to accumulate cached values output by the first cache 403 and the third element values. For example, one ACC processing element 4021 receives one “third element value 0” (for example, c00) in an intermediate result matrix Co output by the vector outer product processing engine 401 and position coordinates (for example, (i0, j0)) corresponding to c00. The ACC processing element 4021 first writes the “third element value 0” into the first cache 403 based on the position coordinates (for example, (1, 1)). Then, when the ACC processing element 4021 receives a “third element value 1” (for example, c10) in an intermediate result matrix C1 that is output by the vector outer product processing engine 401 and position coordinates (for example, (1, 1)) corresponding to c10, the ACC processing element 4021 receives a cached value (for example, c00) corresponding to the position coordinates that is output by the first cache 403. Then, the adder 40210 is configured to add c00 and c10, and output an accumulated value of c00 and c10 to the first cache 403.


Optionally, refer to FIG. 4E. The adder 40210 further specifically includes a sign unit 40215, an exponent unit 40216, an integer unit 40217, and a precision format conversion unit 40218. The sign unit 40215, the exponent unit 40216, and the integer unit 40217 all are connected to the precision format conversion unit 40218. The sign unit 40215 is configured to process a positive or negative sign of an input value. The exponent unit 40216 is configured to process decimal point displacement calculation when two floating-points are multiplied. The integer unit 40217 is configured to perform a multiplication operation on two integers. The precision format conversion unit 40218 is configured to output a value format (for example, a floating-point format such as FP16, FP32, and FP64, or an integer format such as int32, int16, and int8) that complies with a specification. In this example, the adder 40210 is a general-purpose unit that supports the floating-point format and the integer format. This enhances applicability of the matrix calculation apparatus.


The following describes a specific function of the format conversion unit 405. The format conversion unit 405 is configured to: convert the first matrix in the compressed format into the N first column vectors, and convert the second matrix in the compressed format into the N first row vectors, where the first column vector includes the first element values and the row coordinates of the first element values, and the first row vector includes the second element values and the column coordinates of the second element values. A dimension of the first matrix is M×N, and a dimension of the second matrix is N×K, where M, N, and K are integers greater than or equal to 1.


Refer to FIG. 5A. For ease of description, in this example, that values of M, N, and K all are 4 is used as an example for description, that is, the dimensions of the first matrix and the second matrix are both 4×4. An example in which the first matrix is a matrix A and the second matrix is a matrix B is used for description. An example in which the compressed format of the first matrix and the second matrix is COO is used for description.


The format conversion unit 405 is configured to split the matrix A by column into four first column vectors, where the four first column vectors are A0, A1, A2, and A3. The format conversion unit 405 splits the matrix B by row into four first row vectors, where the four first row vectors are B0, B1, B2, and B3. An example in which the first column vector is A0 is used for description, where A0 is [a0, a1, a2, a3]T, and a0, a1, a2, a3 in A0 all are element values. Each element value has a corresponding row coordinate. For example, a row coordinate of a0 is i0, a row coordinate of a1 is i1, a row coordinate of a2 is i2, and a row coordinate of a3 is i3. An example in which the first row vector is B0 is used for description. B0 is [b0,b1,b2,b3], and b0,b1,b2,b3 in B0 all are element values. Each element value in B0 has a corresponding column coordinate. For example, a column coordinate of b0 is jo, a column coordinate of b1 is j1, a column coordinate of b2 is j2, and a column coordinate of b3 is j3. It should be understood that, the matrix A is split based on N columns, and for each element value in a first column vector obtained after splitting, only a row coordinate of the element value is reserved. The matrix B is split based on N rows, and for each element value in a first row vector obtained after splitting, only a column coordinate of the element value is reserved. Similarly, A1 is [c0,c1,c2,c3]T, and c0, c1, c2, c3 in A1 all are element values. Each element value has a corresponding row coordinate. For example, a row coordinate of c0 is k0, a row coordinate of c1 is k1, a row coordinate of c2 is k2, and a row coordinate of c3 is k3. B1 is [d0,d1,d2,d3], and d0,d1,d2,d3 in B1 are element values. Each element value in B1 has a corresponding column coordinate. For example, a column coordinate of d0 is l0, a column coordinate of d1 is l1, a column coordinate of d2 is l2, and a column coordinate of d3 is l3.


For example, the first matrix in the compressed format is represented in the COO format as: (0, 0, 1), (1, 0, 2), (1, 1, 3), (1, 2, 4), (2, 0, 5), (2, 2, 6), (2, 3, 7), (3, 1, 8), (3, 3, 9), and (3, 0, 6). The format conversion unit 405 splits the first matrix in the compressed format by column, where element values in (0, 0, 1), (1, 0, 2), (2, 0, 5) and (3, 0, 6) are element values in a same column, that is, all are element values in column 0. It should be understood that, when the format conversion unit 405 splits the matrix A by column, element values “1”, “2”, “5”, and “6” are used as element values in the A0 vector. The element values “1”, “2”, “5”, and “6” all are element values in a same column. In this case, only row coordinates of the element values need to be reserved. In this example, the vector A0 includes four element values, and the four element values are “1”, “2”, “5”, and “6”. In addition, a row coordinate of the element value “1” is “0”. Similarly, a row coordinate of the element value “2” is “1”, a row coordinate of the element value “5” is “2”, and a row coordinate of the element value “6” is “3”.


For another example, the second matrix in the compressed format is represented in the COO format as: (0, 0, 1), (1, 0, 2), (2, 1, 3), (1, 2, 4), (2, 0, 5), (2, 2, 6), (2, 3, 7), (3, 1, 8), (3, 3, 9), and (3, 0, 6). The format conversion unit 405 splits the second matrix based on row coordinates, that is, uses element values in a same row as element values in a same vector. For example, row coordinates in the four triplets (2, 1, 3), (2, 0, 5), (2, 2, 6) and (2, 3, 7) are the same. In this case, element values in the four triplets are split into one row vector (for example, the row vector B0), and the element values “3”, “5”, “6”, and “7” are used as element values in the first row vector B0, and each element value in the first row vector B0 has a corresponding column coordinate. For example, a column coordinate of the element value “3” is “1”, a column coordinate of the element value “5” is “0”, a column coordinate of the element value “6” is “2”, and a column coordinate of the element value “7” is “3”. It should be noted that a specific value in the COO format is merely an example provided for ease of description, and does not constitute a limitation on this application.


The vector outer product processing engine 401 is configured to calculate vector outer products of the N first column vectors and the N first row vectors, to obtain N intermediate result matrices. The intermediate result matrix includes third element values and position coordinates of the third element values, and the position coordinates include the row coordinates of the first element values and the column coordinates of the second element values. For example, refer to FIG. 5B. The vector outer product processing engine 401 calculates a vector outer product of a first column vector and a first row vector. For example, the vector outer product processing engine 401 calculates a vector outer product of A0 and B0, to obtain an intermediate result matrix C0. Similarly, the vector outer product processing engine 401 calculates a vector outer product of A1 and B1, to obtain an intermediate result matrix C1. The vector outer product processing engine 401 calculates a vector outer product of A2 and B2, to obtain an intermediate result matrix C2. The vector outer product processing engine 401 calculates a vector outer product of A3 and B3, to obtain an intermediate result matrix C3. An example in which the intermediate result matrix calculated by the vector outer product processing engine 401 is C0 is used, as shown in Formula (11):










C
0

=



[




a
0






a
1






a
2






a
3




]



[


b
0




b
1




b
2




b
3


]


=

[





a
0



b
0






a
0



b
1






a
0



b
2






a
0



b
3








a
1



b
0






a
1



b
1






a
1



b
2






a
1



b
3








a
2



b
0






a
2



b
1






a
2



b
2






a
2



b
3








a
3



b
0






a
3



b
1






a
3



b
2






a
3



b
3





]






Formula



(
11
)








The intermediate result matrix C0 includes third element values, and each third element value has corresponding position coordinates. The position coordinates are a combined value of a row coordinate of a first element value and a column coordinate of a second element value. For example, refer to FIG. 5C. If a row coordinate of a0 is i0, and a column coordinate of b0 is j0, position coordinates of a third element value a0b0 are (i0, j0). Similarly, if a row coordinate of a0 is i0, and a column coordinate of b1 is j1, position coordinates of a third element value a0b1 are (i0 j1); and if a row coordinate of a1 is i1, and a column coordinate of b0 is j0, position coordinates of a third element value a1b0 are (i1, j0). The position coordinates of the third element values in the intermediate result matrix C0 are not described by using examples one by one.


An example is as follows: a0 is 1, a1 is 2, a2 is 5, and a3 is 6. In addition, the row coordinate of a0 is “0”. Similarly, the row coordinate of α1 is “1”, the row coordinate of a2 is “2”, and the row coordinate of a3 is “3”. b0 is 3, b1 is 5, b2 is 6, and b3 is 7. The column coordinate of b0 is “1”, the column coordinate of b1 is “0”, the column coordinate of b2 is “2”, and the column coordinate of b3 is “3”. The third element value a0b0=1×3=3, and the position coordinates of a0b0 include the row coordinate of a0 and the column coordinate of b0, that is, the position coordinates of a0b0 are (0, 3). The third element value a0b1=1×5=5, and the position coordinates of a0b1 include the row coordinate of a0 and the column coordinate of b1, that is, the position coordinates of a0b1 are (0, 0). The third element value a1b0=2×3=6, and the position coordinates of a1b0 include the row coordinate of a1 and the column coordinate of b0, that is, the position coordinates of a1b0 are (1, 3).


Refer to FIG. 5D. The vector outer product processing engine 401 obtains the intermediate result matrix according to the vector outer product calculation formula in Formula (7). An example in which the intermediate result matrix calculated by the vector outer product processing engine 401 is C1 is used, as shown in Formula (12):










C
1

=



[




c
0






c
1






c
2






c
3




]



[


d
0




d
1




d
2




d
3


]


=

[





c
0



d
0






c
0



d
1






c
0



d
2






c
0



d
3








c
1



d
0






c
1



d
1






c
1



d
2






c
1



d
3








c
2



d
0






c
2



d
1






c
2



d
2






c
2



d
3








c
3



d
0






c
3



d
1






c
3



d
2






c
3



d
3





]






Formula



(
12
)








Similarly, the intermediate result matrix C1 includes third element values, and each third element value in C1 has corresponding position coordinates. For example, refer to FIG. 5D. If a row coordinate of c0 is k0, and a column coordinate of d0 is l0, position coordinates of a third element value c0d0 are (k0, l0). Similarly, if a row coordinate of c0 is k0, and a column coordinate of d1 is l1, position coordinates of a third element value c0d1 are (k0, l1); and if a row coordinate of c1 is k1, and a column coordinate of d0 is l0, position coordinates of a third element value c1d0 are (k1, l0). The position coordinates of the third element values in the intermediate result matrix C1 are not described by using examples one by one.


The accumulator 402 is configured to accumulate, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix. Still refer to FIG. 5B. In this example, four intermediate result matrices are used as an example for description. Third element values in each intermediate result matrix in the intermediate result matrices C0, C1, C2, and C3 each have position coordinates. The accumulator 402 accumulates the third element values with the same position coordinates, to obtain the result matrix. For example, refer to FIG. 5E. The accumulator 402 adds four third element values whose position coordinates are (0, 0) in C0, C1, C2, and C3, to obtain a fourth element value in the result matrix. Similarly, the accumulator 402 adds four third element values whose position coordinates are (1, 1) in C0, C1, C2, and C3, to obtain a fourth element value in the result matrix. In this example, specific values of position coordinates in a matrix are merely examples for ease of description, and are not for limitation.


Optionally, an implementation in which the accumulator 402 accumulates, based on the indexes of the position coordinates of the third element values, the third element values with the same position coordinates in the N intermediate result matrices includes at least the following two implementations.


In a first possible implementation, when performing vector outer product calculation on the first row vectors and the first column vectors, the vector outer product processing engine 401 generates the intermediate result matrices in a specific sequence. For example, the first row vector and the first column vector that are shown in FIG. 5B are used as an example to calculate a vector outer product. The vector outer product processing engine 401 calculates a vector outer product of A0 and B0, to obtain the intermediate result matrix C0. Then, the vector outer product processing engine 401 calculates a vector outer product of A1 and B1, to obtain the intermediate result matrix C1. Next, the vector outer product processing engine 401 calculates a vector outer product of A2 and B2, to obtain the intermediate result matrix C2. Finally, the vector outer product processing engine 401 calculates a vector outer product of A3 and B3, to obtain the intermediate result matrix C3. For example, a sequence of the four intermediate result matrices is C0, C1, C2, and C3. The accumulator 402 receives the four intermediate result matrices from the vector outer product processing engine 401 according to a sequence of generating the four intermediate result matrices.


Refer to FIG. 6. The four intermediate result matrices include at least a first intermediate result matrix (for example, C0) and a second intermediate result matrix (for example, C0. To distinguish between position coordinates in the first intermediate result matrix and position coordinates in the second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are referred to as “first position coordinates”, and position coordinates of third element values in the second result matrix are referred to as “second position coordinates”.


For example, in the first cache 403, the first cache 403 is divided into a plurality of storage positions by using a row coordinate identifier and a column coordinate identifier. For example, in the first cache 403, storage space of the first cache 403 is divided into q×p storage positions based on p rows and q columns. After receiving C0, the accumulator 402 writes the third element values in C0 into a corresponding position in the first cache 403 based on the first position coordinates. For example, if position coordinates of a third element value a0b0 in C0 are (i0, j0), for example, (i0, j0) are (1, 1), the accumulator 402 writes, based on the position coordinates (i0, j0), a0b0 into a position of row 1 column 1 in the first cache 403. Similarly, if position coordinates of a third element value a0b1 in C0 are (i0, j1), for example, (i0, j1) are (1, 2), the accumulator 402 writes, based on the position coordinates (i0, a0b1 into a position of row 1 column 2 in the first cache 403. If position coordinates of a third element value a0b2 in C0 are (i0, j2), for example, (i0, j2) are (1, 3), the accumulator 402 writes, based on the position coordinates (i0, j2), a0b2 into a position of row 1 column 3 in the first cache 403. If position coordinates of a third element value a0b1 in C0 are (i0, j3), for example, (i0, j3) are (1, 4), the accumulator 402 writes, based on the position coordinates (i0, j3), a0b3 into a position of row 1 column 4 in the first cache 403. A process in which the accumulator 402 writes other third element values in C0 into the first cache 403 is not described in detail. A final result is that the accumulator 402 writes all third element values in C0 into the first cache 403 based on the first position coordinates corresponding to each third element value.


Then, when the accumulator 402 receives C1, the accumulator 402 searches, based on the second position coordinates of a third element value in C1, for a cached value at a position corresponding to the second position coordinates in the first cache 403. If no cached value exists at the position corresponding to the second position coordinates, the accumulator 402 writes the third element value into the position corresponding to the second position coordinates in the first cache 403. If there is a cached value at the position corresponding to the second position coordinates, the accumulator 402 reads the cached value from the first cache 403, and accumulates the cached value and the third element value in C1. For example, position coordinates of a third element value c0d0 in C1 are (k0, l0), for example, (k0, l0) is (1, 0). The accumulator 402 queries the first cache 403 based on the position coordinates (k0, l0). If there is no cached value at the position of row 1 column 0 in the first cache 403, the accumulator 402 writes c0d0 into the position of row 1 column 0 in the first cache 403. Position coordinates of a third element value c0d1 in C1 are (k0, l1), for example, (k0, l1) is (1, 1). The accumulator 402 obtains through query that there is a cached value a0b0 at a position of row 1 column 1 in the first cache 403. The accumulator 402 reads a0b0, adds c0d1 and a0b0, and then writes a result (cod1+a0b0) obtained after the addition into a corresponding position (row 1 column 1) of the first cache 403 based on the position coordinates of c0d1. Similarly, position coordinates of a third element value code in C1 are (k0, l2), for example, (k1, l2) is (1, 2). The accumulator 402 obtains through query that there is a cached value a0b1 at a position of row 1 column 2 in the first cache 403. The accumulator 402 reads the cached value a0b1 from the first cache 403, adds c0d2 and a0b1, and then writes an accumulated value (c0d2+a0b1) obtained after the addition into a corresponding position (row 1 column 2) of the first cache 403 based on the position coordinates c0d2. In this example, the third element values in C1 are not described by using examples one by one. A final result is that the accumulator 402 adds third element values with same position coordinates in C1 and C0, and writes a result obtained after the addition into a corresponding position in the first cache 403 based on the position coordinates. Similarly, when the accumulator 402 receives the intermediate result matrix C2 transmitted by the vector outer product processing engine 401, the accumulator 402 reads a cached value (an accumulated value of third element values with same position coordinates in C0 and C1) from a corresponding position in the first cache 403 based on position coordinates of each third element value in C2. The accumulator 402 accumulates third element values with same position coordinates in C0, C1, and C2, and then writes an accumulated result into a position corresponding to the position coordinates in the first cache 403. It should be understood that processing processes performed by the accumulator 402 for C2 and C3 are similar to a processing process for C1. Details are not described herein again. A final processing result of the accumulator 402 is to accumulate third element values with same position coordinates in C0, C1, C2, and C3, to obtain a fifth element value. The result matrix includes a plurality of fifth element values. Finally, the first cache 403 outputs the result matrix.


It should be noted that the fifth element value is an accumulated value of a third element value in at least one of the intermediate result matrices C0, C1, C2, and C3. For example, still refer to FIG. 6. The position coordinates of a0b0 in C0 are (1, 0), which are included in none of position coordinates of third element values in the other three intermediate result matrices (C1, C2, and C3). In this case, a value written into the position of row 1 column 0 in the first cache 403 is a0b0, and a0b0 is not accumulated with a third element value in another intermediate result matrix. Similarly, position coordinates of a0b1 in C0 and c0d1 in C1 are both (1, 1), which are included in none of position coordinates of third element values in the other two intermediate result matrices (C2 and C3). In this case, a value written into a position of row 1 column 0 in the first cache 403 is an accumulated value of a0b1 and c0d1 . In conclusion, it should be understood that one fifth element value may be one third element value, an accumulated value of two third element values, an accumulated value of three third element values, an accumulated value of four third element values, or the like. In an actual operation, a quantity of third element values accumulated to obtain a fifth element value in the result matrix is determined by a quantity of third element values corresponding to same position coordinates. This is not specifically limited.


Further, if the fifth element value is an accumulated value of third element values, the fifth element value may be a zero-element value. In this example, the result matrix is a matrix in an uncompressed format. Optionally, to save transmission resources or facilitate a next calculation operation, the matrix calculation apparatus may compress the result matrix, to output a matrix in a compressed format. Refer to FIG. 7. FIG. 7 is another schematic diagram of the structure of the matrix calculation apparatus. The matrix calculation apparatus further includes a matrix compression unit 404. The matrix compression unit 404 is connected to the first cache 403, the first cache 403 outputs the result matrix, and the matrix compression unit 404 receives the result matrix, and converts the result matrix into the matrix in the compressed format based on the row coordinate identifier and the column coordinate identifier in the first cache 403. The result matrix may be in a COO format, a CSR format, a CSC format, or the like. This is not specifically limited.


It should be understood that, in a process of performing accumulation calculation by the accumulator 402, the first cache 403 is configured to store the third element values and the result obtained after the third element values are accumulated. The storage space of the first cache 403 needs to be greater than or equal to a value. For example, if a dimension of each intermediate result matrix is M×P, in other words, each intermediate result matrix includes M×P third element values, a quantity of cache positions included in the first cache 403 is greater than or equal to M×P. In this case, the storage space of the first cache 403 needs to be capable of storing at least M×P values.


In a first possible implementation, the matrix calculation apparatus may output a matrix in an uncompressed format or a matrix in a compressed format based on different application scenarios, thereby expanding an applicable scenario of the matrix calculation apparatus.


In a second possible implementation, refer to FIG. 8. The accumulator 402 sorts the third element values in the N intermediate result matrices based on the position coordinates of the third element values, then compares position coordinates in N intermediate result matrices obtained through sorting, adds up third element values with same position coordinates, and deletes position coordinates of zero-element values, to obtain the result matrix in the compressed format. For example, when the accumulator 402 receives Co, the accumulator 402 first sorts the third element values in Co based on the position coordinates of the third element values, and then writes the sorted third element values into the first cache 403. The accumulator 402 may sort the third element values based on row coordinates of the third element values. Optionally, the accumulator 402 may sort the third element values based on column coordinates of the third element values. In this example, an example in which the accumulator 402 sorts the third element values based on the row coordinates is used for description. The accumulator 402 receives C1, and sorts third element values in C1 based on row coordinates of the third element values. The accumulator 402 compares the position coordinates of the third element values in C1 and the position coordinates of the third element values in C0 according to a sequence of the position coordinates of the third element values in C1. For example, the accumulator 402 first compares the row coordinates, for example, compares k0 and i0, and if k0 is consistent with i0, continues to compare the column coordinates. If k0 is inconsistent with i0, the accumulator 402 continues to compare k0 and i1. After comparing the row coordinates, the accumulator 402 compares the column coordinates l0 and j0, l0 and j1, and the like in a sequence of the position coordinates. When position coordinates (k0, l0) are consistent with position coordinates (i0, j1), the accumulator 402 adds a third element value c0d0 corresponding to the position coordinates (k0, l0) and a third element value a0b0 corresponding to the position coordinates (i0, j1), and then writes an addition result into the first cache 403.


Similarly, when the accumulator 402 receives C2, the accumulator 402 sorts third element values in C2 based on position coordinates. The accumulator 402 reads position coordinates in the first cache 403, compares the position coordinates of the third element values in C2 and the position coordinates in the first cache 403, adds third element values with same position coordinates to cached values, to obtain accumulated values, and then stores the accumulated values in the first cache 403. Calculation performed by the accumulator 402 on the third element values in C3 is similar to calculation performed on the third element values in C2. Details are not described by using examples herein. A final result is that the accumulator 402 accumulates third element values with same position coordinates in the intermediate result matrices C3, C2, C1, and C0, to obtain fifth element values, and the accumulator 402 deletes a zero-element value and position coordinates of the zero-element value from the plurality of fifth element values. The first cache 403 outputs the result matrix, where the result matrix includes the fifth element values and position coordinates of the fifth element values. In addition, the fifth element value included in the result matrix is a non-zero-element value. Therefore, the result matrix output by the first cache 403 is a matrix in a compressed format.


In the second implementation, the matrix in the compressed format may be directly obtained, and the matrix in the compressed format output by the matrix calculation apparatus may be used in some application scenarios in which a matrix in a compressed format needs to be subsequently calculated. In addition, because the matrix calculation apparatus outputs the matrix in the compressed format, transmission resources of a subsequent transmission matrix can be reduced. In addition, in the second implementation, because the third element values cached in the first cache 403 are cached according to the sequence of the position coordinates, the implementation in this example can be implemented by using smaller cache space, thereby saving cache space of the first cache 403.


Optionally, in this example, the format conversion unit 405 is further configured to: obtain a fifth matrix and a sixth matrix, and perform format conversion on the fifth matrix and the sixth matrix in an uncompressed format to obtain the first matrix and the second matrix in the compressed format; and output the first matrix and the second matrix to the second cache 406, where at least one of the fifth matrix or the sixth matrix is a matrix in the uncompressed format. The vector outer product processing engine 401 obtains the first matrix and the second matrix in the compressed format from the second cache 406. In this example, the matrix calculation apparatus may receive the matrix in the uncompressed format, and then convert the matrix in the uncompressed format into the matrix in the compressed format, so that the matrix calculation apparatus can support calculation of matrices in a plurality of formats. In this example, the fifth matrix and the sixth matrix may include the following several cases.


In a first case, refer to FIG. 9. Both the fifth matrix and the sixth matrix are matrices in the uncompressed format. In this case, the format conversion unit 405 converts the fifth matrix into the first matrix in the compressed format, and converts the sixth matrix into the second matrix in the compressed format. Optionally, the format conversion unit 405 may convert the fifth matrix and the sixth matrix into compressed matrices in the COO format. Optionally, because a row coordinate of an element value is reserved in a CSC compressed format, and a column coordinate of an element value is reserved in a CSR compressed format, the format conversion unit may convert the fifth matrix in the uncompressed format into the first matrix in the CSC compressed format, and convert the sixth matrix in the uncompressed format into the CSR compressed format. Further, the format conversion unit 405 converts the first matrix in the compressed format into the N first column vectors, converts the second matrix in the compressed format into the N first row vectors, and writes the N first column vectors and the N first row vectors into the second cache 406. The vector outer product processing engine 401 obtains the first row vectors and the second row vectors from the second cache 406.


In a second case, one of the fifth matrix and the sixth matrix is a matrix in the uncompressed format, and the other matrix is a matrix in the compressed format. An example in which the fifth matrix is a matrix in the uncompressed format, and the sixth matrix is a matrix in the compressed format is used for description. If the sixth matrix is a compressed matrix in a CSC or CSR format, the format conversion unit converts both the fifth matrix and the sixth matrix into compressed matrices in a COO format.


Optionally, the format conversion unit 405 is further configured to convert a matrix in the compressed format into a matrix in a target compressed format. For example, when both the first matrix and the second matrix are in the CSC format or the CSR format, the format conversion unit converts both the first matrix and the second matrix into the COO format.


In this example, the matrix calculation apparatus may convert the matrix in the uncompressed format into the matrix in the compressed format by using the format conversion unit, so that the matrix calculation apparatus can support both calculation of the matrix in the compressed format and calculation of the matrix in the uncompressed format. Optionally, the format conversion unit may convert a matrix in a non-target compressed format into a matrix in the target compressed format (for example, the COO format). In this example, the matrix calculation apparatus may convert a matrix in another compressed format into a matrix in the target compressed format, and perform matrix calculation on the matrix in the target compressed format. The matrix calculation apparatus provided in this application may support calculation of matrices in various formats.


Optionally, to improve applicability of the matrix calculation unit, high-precision matrix calculation may be implemented based on a low-precision matrix calculation apparatus. Precision of element values included in the first column vector and the first row vector is first precision. The format conversion unit 405 splits the first column vector into X second column vectors, and splits the first row vector into X second row vectors, where the second column vector and the second row vector include element values of second precision, and the first precision is higher than the second precision. Then, the vector outer product processing engine 401 calculates vector outer products of the X second column vectors and the X second row vectors to obtain X2 fourth matrices, where the fourth matrix includes fourth element values and position coordinates of the fourth element values. Finally, the accumulator 402 accumulates, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices, to obtain the intermediate result matrix, where precision of the third element values in the intermediate result matrix is the first precision.


The following uses an example to describe how to split a vector of the first precision into a vector of the second precision in this example. After converting the first matrix in the compressed format into the N first column vectors and converting the second matrix in the compressed format into the N first row vectors, the format conversion unit 405 may further split a first column vector into X second column vectors and split a first row vector into X second row vectors based on precision of element values. The precision of the element values included in the first column vector and the first row vector is the first precision. Both the second column vector and the second row vector include an element value of the second precision. For ease of description, a vector including the element value of the second precision is referred to as a “vector of the second precision”, and a vector including the element value of the first precision is referred to as a “vector of the first precision”. X is an integer greater than or equal to 2. The first precision and the second precision may be integer precision, or the first precision and the second precision may be floating point precision. This is not specifically limited. Specifically, the format conversion unit 405 splits each element value in the vector of the first precision into a plurality of values of the second precision, and splits the vector of the first precision into X vectors of the second precision. The following provides descriptions based on cases in which the vector of the first precision and the vector of the second precision are integers, and the vector of the first precision and the vector of the second precision are floating-point vectors.


In a first case, the vector of the first precision and the vector of the second precision are integers. The first precision is higher than the second precision. For example, the first precision is int32, and the second precision may be int2, int4, int8, or int16. Alternatively, the first precision is int16, and the second precision may be int4 or int8. Alternatively, the first precision is int8, and the second precision is int2 or int4. Alternatively, the first precision is int4, and the second precision is int2.


The format conversion unit 405 splits a high-precision integer number into a plurality of low-precision integer numbers, and splits the high-precision integer number from a most significant bit to a least significant bit. For example, refer to FIG. 10. An example in which the first precision is int32 and the second precision is int16 is used for description. int32 includes a 32-bit value range. The 32 bits in int32 are split into two 16 bits from the most significant bit to the least significant bit. That is, int32 is split into two int16. Alternatively, the 32 bits in int32 are split into four 8 bits from the most significant bit to the least significant bit. That is, int32 is split into four int8. Similarly, int16 includes a 16-bit value range. The 16 bits in int16 are split into two 8 bits from the most significant bit to the least significant bit. That is, int16 is split into two int8.


In a second case, the vector of the first precision and the vector of the second precision are floats. For example, a first-precision floating-point is FP32 or FP64, and a second-precision floating-point is FP16. Alternatively, a first-precision floating-point is FP64, and a second-precision floating-point may be FP32, or may be FP16. The following uses a case in which the first-precision floating-point is FP32 and the second-precision floating-point is FP16 as an example for description.


1. One FP32 is split to obtain three FP16.


Currently, composition of FP32 in a standard format is shown in Table 1. FP32 includes a 1-bit (bit) sign, an 8-bit exponent, and a 23-bit mantissa. In addition, there is an omitted 1-bit integer, and the omitted integer is 1. For FP32 in the standard format, the integer and the mantissa are 24 bits in total. FP16 in a standard format includes a 1-bit sign, a 5-bit exponent, a 10-bit mantissa. In addition, there is an omitted 1-bit integer, and the omitted integer is 1. For FP16 in the standard format, the integer and the mantissa are 11 bits in total. To split FP32 in the standard format to obtain FP16 in the standard format, three FP16 in the standard format are required.


The integer and the mantissa of FP32 in the standard format may be divided into three parts. A first part is the integer and the first 10 bits of the mantissa, a second part is the 11th bit to the 21st bit of the mantissa, and a third part is the 22nd bit and the 23rd bit of the mantissa. The three parts are separately represented by FP16 in the standard format. It should be noted herein that, when the 22nd bit and the 23rd bit of the mantissa in the third part is represented by FP16 in the standard format, nine 0s may be padded after the 23rd bit of the mantissa, that is, the 22nd bit and the 23rd bit of the mantissa and the padded 0s are represented by FP16 in the standard format.


In addition, the exponent range of FP16 is −15 to 15, that is, may indicate that a decimal point is shifted leftward by 15 bits to rightward by 15 bits. When FP16 in the standard format is for representing the first part of FP32, the fixed exponent bias is 0; when FP16 in the standard format is for representing the second part of FP32, the fixed exponent bias is −11; and when FP16 in the standard format is for representing the third part of FP32, the fixed exponent bias is −22. It can be learned that, when the third part is represented, only the corresponding fixed exponent bias has exceeded an exponent range of FP16. Therefore, a corresponding fixed exponent bias may be extracted for the exponent of each FP16 in the standard format.


Therefore, FP32 in the standard format may be represented as:

    • A1=2EA1 (a0+2S1 a1+2−2S1 a2) , where A1 is FP32 in the standard format, EA1 is an exponent of A1, a0, a1, and a2 are three FP16 in the standard format obtained through splitting, and S1 is a minimum fixed exponent bias. For FP16 in the standard format, S1=11.


In addition, a common exponent bias may be extracted for each exponent in FP16 in the standard format. Similarly, FP32 in the standard format may be represented as:

    • A12EA1−s1 (a0′+a1′+a2′), where a0′, a1′, and a2′ are three FP16 in the standard format obtained through splitting. In the foregoing two representation methods, FP16 obtained through splitting has the following relationship: a0=2S1a0′, a1=a1′, and a2=2S1a2′.


      2. One FP32 is split to obtain two FP16.


To reduce an amount of FP16 obtained through splitting, current FP16 in a standard format may be adjusted, and a mantissa of FP16 is adjusted to 13 bits, and a quantity of bits of a sign and a quantity of bits of an exponent remain unchanged. Adjusted FP16 may be referred to as FP16 in a non-standard format. For FP16 in the non-standard format, an integer and the mantissa are 14 bits in total. In this case, if a mantissa of FP32 in a standard format is to be represented by FP16 in the non-standard format, only two FP16 in the non-standard format are required.


An integer and a mantissa of FP32 in the standard format are divided into two parts. A first part is the integer and the first 13 bits of the mantissa, and a second part is the 14th bit to the 23rd bit. The two parts are separately represented by FP16 in the non-standard format.


It should be further noted herein that, when the second part is represented by non-standard FP16, four 0s may be padded after the 23rd bit of the mantissa, that is, the 14th bit to the 23rd bit of the mantissa and the padded 0s are represented by FP16 in the non-standard format. Same as the foregoing first case, herein, a corresponding fixed exponent bias may also be extracted for an exponent of each FP16 in the standard format.


Similarly, FP32 in the standard format may be represented as:

    • A2=2EA2 (a3+2−S2a4), where A2 is FP32 in the standard format, EA2 is an exponent of A2, a3 and a4 are two FP16 in the non-standard format obtained through splitting, and S2 is a fixed exponent bias. For FP16 in the non-standard format, S2=14 .


In addition, a common exponent bias may be extracted for each exponent in FP16 in the standard format. Similarly, FP32 in the standard format may be represented as:

    • A2=2EA2−s2 (a3′+a4′), where a3′ and a4′ are two FP16 in the non-standard format obtained through splitting. In the foregoing two representation methods, FP16 obtained through splitting has the following relationship: a3=2S2a3′ and a4=a4′.


Certainly, for a case in which the first-precision floating-point is FP64 and the second-precision floating-point is FP32, there may be the following cases in which for FP64 is split to obtain a plurality of FP32: One FP64 floating-point is split to obtain three FP32 floating-points; or one FP64 floating-point is split to obtain two FP32 floating-points. Optionally, in a case in which the first-precision floating-point is FP64 and the second-precision floating-point is FP16, there may be the following cases in which FP64 is split to obtain a plurality of FP16: One FP64 floating-point is split to obtain five FP16 floating-points; or one FP64 floating-point is split to obtain four FP16 floating-points. A splitting principle is similar to the foregoing described case in which the first-precision floating-point is FP32 and the second-precision floating-point is FP16. Details are not described herein.


For example, for ease of description, an example in which the first row vector is split into two second row vectors and the first column vector is split into two second column vectors is used for description. For example, the first row vector A0 is a column vector [a0, a1, a2, a3]T whose precision is FP32. For example, in the foregoing method for splitting a value of the first precision into two values of the second precision, the format conversion unit 405 splits a floating-point a0 whose precision is FP32 into two floating-points (for example, a0M and a0L) whose precision is FP16. Similarly, a1 is split into a1M and a1L. a2 is split into a2M and a2L. a3 is split into a3M and a3L. To be specific, as shown in FIG. 11, the format conversion unit 405 splits a column vector [a0 a1 a2 a3]T whose precision is FP32 into two column vectors [a0M a1M a2M a3M]T (denoted as a “second column vector 1”) and [a0L a1L a2L a3L]T (denoted as a “second column vector 2”) whose precision is FP16. Similarly, B0 is a row vector [b0 b1 b2 b3] whose precision is FP32, and [b0 b1 b2 b3] is also split into two row vectors [b0M b1M b2M b3M] (denoted as a “second row vector 1”) and [b0L b1L b2L b3L] (denoted as a “second row vector 2”) corresponding to FP16. It should be noted that row coordinates of all element values in the first column vector and the second column vector are the same. To be specific, row coordinates of a0M, and a0L are i0, row coordinates of a1M and a1L are i1, row coordinates of a2M and a2L are i2, and row coordinates of a3M and a3L are i3. Column coordinates of all element values in the first row vector and the second row vector are the same. To be specific, column coordinates of b0M and b0L are k0, column coordinates of b1M and b1L are j1, column coordinates of b2M and b2L are j2, and column coordinates of b3M and b3L are j3.


Further, the vector outer product processing engine 401 calculates vector outer products of the X second column vectors and the X second row vectors to obtain X2 fourth matrices, where the fourth matrix includes fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values include the row coordinates of the element values in the first column vector and the column coordinates of the element values in the first row vector, and precision of the fourth element values is the first precision (for example, FP32). In this example, refer to FIG. 12. An example in which X is 2 is used for description. The vector outer product processing engine 401 calculates vector outer products of two second column vectors [a0M a1M a2M a3M]T and [a0L a1L a2L a3L]T and two second row vectors [b0M b1M b2M b3M]T and [b0L b1L b2L b2L]. To be specific, the vector outer product processing engine 401 calculates a vector outer product of [a0M a1M a2M a3M]T and [b0M b1M b2M b3M]T to obtain a fourth matrix 1; calculates a vector outer product of [a0M a1M a2M a3M]T and [b0L b1L b2L b3L] to obtain a fourth matrix 2; calculates a vector outer product of [a0L a1L a2L a3]T and [b0M b1M b2M b3M] to obtain a fourth matrix 3; and calculates a vector outer product of [a0L a1L a2L a3L]T and [b0L b1L b2L b3L] to obtain a fourth matrix 4. The vector outer product processing engine 401 obtains four fourth matrices through vector outer product calculation. The accumulator 402 obtains the four fourth matrices, and accumulates, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the four fourth matrices, to obtain an intermediate result matrix, where precision of the third element values in the intermediate result matrix is the first precision. For example, a fourth element value a0Mb0L in the fourth matrix 1, a fourth element value a0Mb1M in the fourth matrix 2, a fourth element value a1Lb0L in the fourth matrix 3, and a fourth element value a1Lb1M in the fourth matrix 4 have same position coordinates. The accumulator 402 accumulates the four fourth element values a0Mb0L, a0Mb1M, a1Lb0L, and a1Lb1M, where an obtained accumulated value is a third element value in the intermediate result matrix C0. In this example, a specific method for accumulating the fourth element values with the same position coordinates in the X2 fourth matrices is similar to the methods for accumulating the values performed by the accumulator 402 in the examples corresponding to FIG. 6 and FIG. 8. For details, refer to the foregoing methods performed by the accumulator 402 in the examples corresponding to FIG. 5E, FIG. 6, and FIG. 8. Details are not described herein again. In this example, when a vector outer product of two vectors is calculated, a first-precision (high-precision) vector may be further split into a plurality of second-precision (low-precision) vectors, and then vector outer product calculation is performed on the low-precision vectors, so that an outer product of the first-precision vector may be obtained by accumulating outer product results of the plurality of second-precision vectors, without losing precision.


In this example, if the precision of the element values in the first row vector and the first column vector is high, the matrix calculation apparatus may split both the first row vector and the first column vector into a plurality of low-precision vectors. The matrix calculation apparatus performs vector outer product calculation on the low-precision second column vectors and the low-precision second row vectors, to obtain a result matrix, so that calculation can be performed on a matrix in a compressed format, and high-precision matrix calculation can be implemented based on a low-precision matrix calculation apparatus, thereby improving applicability of the matrix calculation apparatus. In addition, in a matrix calculation process, an upper-layer software application (such as AI and HPC) based on the matrix calculation apparatus does not sense a specific matrix calculation process, so that software adaptation costs can be greatly reduced.


Based on the matrix calculation apparatus provided in this application, a great benefit can be obtained in a plurality of matrix calculation scenarios. For example, when the matrix calculation apparatus is used in an AI training and inference scenario, calculation of a matrix in a compressed format and matrix in an uncompressed format can be completely supported. In AI calculation, a sparseness characteristic of a weight and feature data is more than 50% on average (that is, more than 50% matrices are in the compressed format). The matrix calculation format in this application may directly calculate the matrix in the compressed format, without splitting the matrix in the compressed format. In this way, calculation efficiency can be improved by more than four times. In addition, for an HPC scenario such as scientific computing, regardless of calculation of a matrix in an uncompressed format that requires high computing power or a matrix calculation scenario in which memory bandwidth is limited, the matrix calculation apparatus in this application can directly access a matrix in a compressed format from a memory, thereby improving a calculation benefit. The matrix calculation apparatus supports full-precision numerical calculation, and can also effectively cover calculation with various different precision requirements. For example, floating-point calculation such as FP32 and FP16 usually required in an AI training scenario, and some scenarios such as an AI training scenario that requires FP64 and HPC scientific computing can be fully supported by the matrix calculation apparatus. In addition, the MAC in the matrix calculation apparatus can also support calculation of integer formats with medium and low precision such as INT1, INT2, INT4, and INT8. For a calculation scenario of AI inference, computing power can be improved and inference computation time can be reduced. In addition, various scenarios in which different precision can be mixed in inference computing, thereby greatly enhancing applicability of the matrix calculation apparatus.


The foregoing describes the embodiment of the matrix calculation apparatus, and the following describes a method performed by the matrix calculation apparatus. Refer to FIG. 13. An embodiment of this application provides a matrix calculation method. The method may be performed by the computing device shown in FIG. 2. Optionally, the method is performed by the matrix calculation apparatus shown in FIG. 3. Optionally, the method may be performed by the matrix calculation apparatus shown in FIG. 4A. Optionally, the method may be performed by the matrix calculation apparatus shown in FIG. 7.


Step 1301: Obtain a first calculation instruction, where the first calculation instruction includes N first column vectors and N first row vectors.


The N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1.


Step 1302: Calculate vector outer products of the N first column vectors and the N first row vectors, to obtain N intermediate result matrices, where the first column vector includes first element values and row coordinates of the first element values, the first row vector includes second element values and column coordinates of the second element values, the intermediate result matrix includes third element values and position coordinates of the third element values, and the position coordinates include the row coordinates of the first element values and the column coordinates of the second element values.


For this step, refer to specific descriptions of functions performed by the vector outer product processing engine 401 in FIG. 5B, FIG. 5C, and FIG. 5D in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


Step 1303: Accumulate, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix.


The N intermediate result matrices include at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates.


In a first possible implementation, the matrix calculation apparatus writes, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates; and then reads, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache, and accumulate the third element values in the second intermediate result matrix and the cached values, to obtain a result matrix in an uncompressed format. Optionally, the matrix calculation apparatus compresses the result matrix in the uncompressed format to obtain a result matrix in a compressed format.


In the first possible implementation, refer to specific descriptions of functions performed by the accumulator 402 in the examples corresponding to FIG. 5E and FIG. 6 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


In a second possible implementation, the matrix calculation apparatus sorts the third element values in the N intermediate result matrices based on the position coordinates of the third element values. The matrix calculation apparatus compares position coordinates in N intermediate result matrices obtained through sorting, adds up third element values with same position coordinates, and deletes position coordinates of a zero-element value, to obtain a result matrix in a compressed format.


In the second possible implementation, refer to specific descriptions of functions performed by the accumulator 402 in the examples corresponding to FIG. 5E and FIG. 8 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


In this embodiment of this application, the matrix calculation apparatus can directly calculate the matrix in the compressed format, and does not need to perform operations such as decompressing the matrix in the compressed format and performing matrix calculation on a decompressed matrix in a conventional method. The matrix calculation apparatus in this embodiment of this application can improve calculation efficiency of the matrix in the compressed format.


Optionally, refer to FIG. 14. To support matrix calculation in a plurality of formats, the matrix calculation apparatus can convert a matrix in an uncompressed format into a matrix in a compressed format, so that the matrix in the uncompressed format can be calculated, and format conversion performed on the matrix in the uncompressed format is shown in the following step 1401 and step 1402.


Step 1401: Obtain a third calculation instruction, where the third calculation instruction includes a fifth matrix and a sixth matrix, and at least one of the fifth matrix and the sixth matrix is a matrix in an uncompressed format.


Step 1402: Perform format conversion on the fifth matrix to obtain the first matrix in the compressed format, and perform format conversion on the sixth matrix to obtain the second matrix.


For step 1401 and step 1402, refer to specific descriptions of functions performed by the format conversion unit 405 in the example corresponding to FIG. 9 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


In this embodiment of this application, the matrix calculation apparatus may convert the matrix in the uncompressed format into the matrix in the compressed format by using the format conversion unit, so that the matrix calculation apparatus can support both calculation of the matrix in the compressed format and calculation of the matrix in the uncompressed format. Optionally, the format conversion unit may convert a matrix in a non-target compressed format into a matrix in the target compressed format (for example, the COO format). In this example, the matrix calculation apparatus may convert a matrix in another compressed format into a matrix in the target compressed format, and perform matrix calculation on the matrix in the target compressed format. The matrix calculation apparatus provided in this application may support calculation of matrices in various formats.


Step 1403: Obtain a second calculation instruction, where the second calculation instruction includes the first matrix and the second matrix.


Step 1404: Convert the first matrix into the N first column vectors, and convert the second matrix into the N first row vectors.


For step 1403 and step 1404, refer to descriptions of functions performed by the format conversion unit 405 in the example corresponding to FIG. 5A in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


Optionally, to improve applicability of the matrix calculation unit, high-precision matrix calculation may be implemented based on a low-precision matrix calculation apparatus.


A vector of first precision is split into a plurality of vectors of second precision, and then vector outer product calculation is performed on the vectors of the second precision. Refer to the following step 1405 to step 1407.


Step 1405: Split the first column vector into X second column vectors, and split the first row vector into X second row vectors. Precision of element values included in the first column vector and the first row vector is the first precision, precision of element values included in the second column vector and the second row vector is the second precision, the first precision is higher than the second precision, and X is an integer greater than or equal to 2.


For this step, refer to the specific descriptions in which the format conversion unit 405 splits the integer number and the format conversion unit 405 splits the floating point value in the examples corresponding to FIG. 10 and FIG. 11 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


Step 1406: Calculate vector outer products of the X second column vectors and the X second row vectors to obtain X2 fourth matrices, where the fourth matrix includes fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values include the row coordinates of the first element values and the column coordinates of the second element values, and precision of the fourth element values is the first precision.


For this step, refer to the descriptions of the function performed by the vector outer product processing engine 401 in the example corresponding to FIG. 12 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


Step 1407: Accumulate, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices, to obtain the intermediate result matrix, where precision of the third element values in the intermediate result matrix is the first precision.


Step 1408: Accumulate, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices, to obtain a result matrix.


For step 1407 and step 1408, refer to functions performed by the accumulator 402 in the examples corresponding to FIG. 5E, FIG. 6, and FIG. 8 in the foregoing embodiment of the matrix calculation apparatus. Details are not described herein again.


In this embodiment of this application, when a vector outer product of two vectors is calculated, a first-precision (high-precision) vector may be further split into a plurality of second-precision (low-precision) vectors, and then vector outer product calculation is performed on the low-precision vectors, so that an outer product of the first-precision vector may be obtained by accumulating outer product results of the plurality of second-precision vectors, without losing precision.


An embodiment of this application provides a matrix calculation circuit. The matrix calculation circuit is configured to perform one or more steps in step 1301 to step 1303 or one or more steps in step 1401 to step 1408 in the foregoing method embodiments. In actual application, the matrix calculation circuit may be an ASIC, an FPGA, a logic circuit, or the like.


Another embodiment of this application provides a matrix calculation system or a chip. A structure of the system or the chip may be shown in FIG. 3, and includes a processor (using a central processing unit as an example) 1 and a matrix calculation apparatus 1. The processor 1 is configured to send a calculation instruction to the matrix calculation apparatus 1. The matrix calculation apparatus 1 is configured to perform one or more steps in step 1301 to step 1303 or one or more steps in step 1401 to step 1408 in the foregoing method embodiments.


Still another embodiment of this application provides a matrix calculation device. A structure of the device may be shown in FIG. 2. The device may be specifically a PCIe card, an SoC, a processor, a server including the foregoing hardware, and the like. Refer to FIG. 2. The device includes a memory 201, a processor 202, a communication interface 203, and a bus 204. The communication interface 203 may include an input interface and an output interface.


The processor 202 may be configured to perform one or more steps in step 1301 to step 1303 or one or more steps in step 1401 to step 1408 in the foregoing method embodiments. In some feasible embodiments, the processor 202 may include a matrix calculation unit, and the matrix calculation unit may be configured to support the processor in performing one or more steps in the foregoing method embodiments. In actual application, the matrix calculation unit may be an ASIC, an FPGA, a logic circuit, or the like. Certainly, the matrix calculation unit may alternatively be implemented by using software. This is not specifically limited in this embodiment of this application.


It should be noted that the components of the matrix calculation circuit, the matrix calculation system, the matrix calculation device, and the like provided in embodiments of this application are separately configured to implement functions of corresponding steps in the foregoing method embodiments. Because the steps are described in detail in the foregoing method embodiments, details are not described herein again.


All or some of the foregoing embodiments may be implemented using software, hardware, firmware, or any combination thereof. When software is used to implement embodiments, the foregoing embodiments may be implemented completely or partially in a form of a computer program product. The computer program product includes one or more computer instructions. When the computer program instructions are loaded or executed on a computer, all or some of the processes or the functions according to embodiments of this application are generated. The computer may be a general-purpose computer, a dedicated computer, a computer network, or other programmable apparatuses. The computer instructions may be stored in a computer-readable storage medium or may be transmitted from a computer-readable storage medium to another computer-readable storage medium. For example, the computer instructions may be transmitted from a website, computer, server, or data center to another website, computer, server, or data center in a wired (for example, a coaxial cable, an optical fiber, or a digital subscriber line (DSL)) or wireless (for example, infrared, radio, or microwave) manner. The computer-readable storage medium may be any usable medium accessible by a computer, or a data storage device, such as a server or a data center, integrating one or more usable media. The usable medium may be a magnetic medium (for example, a floppy disk, a hard disk, or a magnetic tape), an optical medium (for example, a DVD), or a semiconductor medium. The semiconductor medium may be a solid-state drive (solid-state drive, SSD).


In conclusion, the foregoing embodiments are merely intended for describing the technical solutions of this application, but not for limiting this application. Although this application is described in detail with reference to the foregoing embodiments, persons of ordinary skill in the art should understand that they may still make modifications to the technical solutions described in the foregoing embodiments or make equivalent replacements to some technical features thereof, without departing from the scope of the technical solutions of embodiments of this application.

Claims
  • 1. A matrix calculation apparatus, comprising: at least one processor; andone or more memories coupled to the at least one processor and storing programming instructions for execution by the at least one processor to: calculate vector outer products of N first column vectors and N first row vectors to obtain N intermediate result matrices, wherein each first column vector comprises first element values and row coordinates of the first element values, each first row vector comprises second element values and column coordinates of the second element values, each intermediate result matrix comprises third element values and position coordinates of the third element values, the position coordinates of the third element values comprise row coordinates of respective first element values and column coordinates of respective second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1; andaccumulate, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix.
  • 2. The matrix calculation apparatus according to claim 1, wherein the N intermediate result matrices comprise at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates; and the one or more memories store the programming instructions for execution by the at least one processor to: write, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates;read, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache; andaccumulate the third element values in the second intermediate result matrix and the cached values to obtain a result matrix in an uncompressed format.
  • 3. The matrix calculation apparatus according to claim 2, wherein the one or more memories store the programming instructions for execution by the at least one processor to: compress the result matrix in the uncompressed format to obtain a result matrix in a compressed format.
  • 4. The matrix calculation apparatus according to claim 1, wherein the one or more memories store the programming instructions for execution by the at least one processor to: sort third element values in the N intermediate result matrices based on position coordinates of the third element values;compare position coordinates in N intermediate result matrices obtained through sorting;aggregate third element values, in the N intermediate result matrices obtained through sorting, with same position coordinates; and delete position coordinates of a zero-element value to obtain a result matrix in a compressed format.
  • 5. The matrix calculation apparatus according to claim 1, wherein the one or more memories store the programming instructions for execution by the at least one processor to: obtain the first matrix and the second matrix;convert the first matrix into the N first column vectors; andconvert the second matrix into the N first row vectors.
  • 6. The matrix calculation apparatus according to claim 5, wherein the one or more memories store the programming instructions for execution by the at least one processor to: obtain a fifth matrix and a sixth matrix;perform format conversion on the fifth matrix to obtain the first matrix; andperform format conversion on the sixth matrix to obtain the second matrix,wherein at least one of the fifth matrix or the sixth matrix is a matrix in an uncompressed format.
  • 7. The matrix calculation apparatus according to claim 1, wherein precision of element values comprised in each first column vector and each first row vector is first precision; the one or more memories store the programming instructions for execution by the at least one processor to: calculate vector outer products of X second column vectors and X second row vectors to obtain X2 fourth matrices, wherein each fourth matrix comprises fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values comprise row coordinates of respective first element values and column coordinates of respective second element values, and precision of the fourth element values is the first precision, wherein the X second column vectors are obtained by splitting the first column vector, the X second row vectors are obtained by splitting the first row vector, precision of element values comprised in each second column vector and each second row vector is second precision, the first precision is higher than the second precision, and X is an integer greater than or equal to 2; andaccumulate, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices to obtain the intermediate result matrix, wherein precision of the third element values in the intermediate result matrix is the first precision.
  • 8. A matrix calculation method, wherein the method comprises: obtaining a first calculation instruction, wherein the first calculation instruction comprises N first column vectors and N first row vectors;calculating vector outer products of the N first column vectors and the N first row vectors to obtain N intermediate result matrices, wherein each first column vector comprises first element values and row coordinates of the first element values, each first row vector comprises second element values and column coordinates of the second element values, each intermediate result matrix comprises third element values and position coordinates of the third element values, the position coordinates of the third element values comprise row coordinates of respective first element values and column coordinates of respective second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1; andaccumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix.
  • 9. The matrix calculation method according to claim 8, wherein the N intermediate result matrices comprise at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates; and the accumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix comprises: writing, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates;reading, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache; andaccumulating the third element values in the second intermediate result matrix and the cached values to obtain a result matrix in an uncompressed format.
  • 10. The matrix calculation method according to claim 9, wherein the matrix calculation method further comprises: compressing the result matrix in the uncompressed format to obtain a result matrix in a compressed format.
  • 11. The matrix calculation method according to claim 8, wherein the accumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix comprises: sorting third element values in the N intermediate result matrices based on position coordinates of the third element values;comparing position coordinates in N intermediate result matrices obtained through sorting,aggregating third element values, in the N intermediate result matrices obtained through sorting, with same position coordinates; anddeleting position coordinates of a zero-element value to obtain a result matrix in a compressed format.
  • 12. The matrix calculation method according to claim 8, wherein before the obtaining a first calculation instruction, the matrix calculation method further comprises: obtaining a second calculation instruction, wherein the second calculation instruction comprises the first matrix and the second matrix;converting the first matrix into the N first column vectors; andconverting the second matrix into the N first row vectors.
  • 13. The matrix calculation method according to claim 12, wherein before the obtaining a second calculation instruction, the matrix calculation method further comprises: obtaining a third calculation instruction, wherein the third calculation instruction comprises a fifth matrix and a sixth matrix, and at least one of the fifth matrix or the sixth matrix is a matrix in an uncompressed format;performing format conversion on the fifth matrix to obtain the first matrix in the compressed format; andperforming format conversion on the sixth matrix to obtain the second matrix.
  • 14. The matrix calculation method according to claim 8, wherein precision of element values comprised in each first column vector and each first row vector is first precision; the calculating vector outer products of the N first column vectors and the N first row vectors to obtain N intermediate result matrices comprises: calculating vector outer products of X second column vectors and X second row vectors to obtain X2 fourth matrices, wherein each fourth matrix comprises fourth element values and position coordinates of the fourth element values, the position coordinates of the fourth element values comprise row coordinates of respective first element values and column coordinates of respective second element values, and precision of the fourth element values is the first precision, wherein the X second column vectors are obtained by splitting the first column vector, the X second row vectors are obtained by splitting the first row vector, precision of element values comprised in each second column vector and each second row vector is second precision, the first precision is higher than the second precision, and X is an integer greater than or equal to 2; andthe matrix calculation method further comprises: accumulating, based on indexes of the position coordinates of the fourth element values, fourth element values with same position coordinates in the X2 fourth matrices to obtain the intermediate result matrix, wherein precision of the third element values in the intermediate result matrix is the first precision.
  • 15. A non-transitory computer-readable storage medium, wherein the non-transitory computer-readable storage medium store programming instructions for execution by at least one processor to perform operations comprising: obtaining a first calculation instruction, wherein the first calculation instruction comprises N first column vectors and N first row vectors;calculating vector outer products of the N first column vectors and the N first row vectors to obtain N intermediate result matrices, wherein each first column vector comprises first element values and row coordinates of the first element values, each first row vector comprises second element values and column coordinates of the second element values, each intermediate result matrix comprises third element values and position coordinates of the third element values, the position coordinates of the third element values comprise row coordinates of respective first element values and column coordinates of respective second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1; andaccumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix.
  • 16. A chip, wherein the chip comprises a matrix calculation apparatus, and the matrix calculation apparatus comprises: at least one processor; andone or more memories coupled to the at least one processor and storing programming instructions for execution by the at least one processor to perform operations comprising: obtaining a first calculation instruction, wherein the first calculation instruction comprises N first column vectors and N first row vectors;calculating vector outer products of the N first column vectors and the N first row vectors to obtain N intermediate result matrices, wherein each first column vector comprises first element values and row coordinates of the first element values, each first row vector comprises second element values and column coordinates of the second element values, each intermediate result matrix comprises third element values and position coordinates of the third element values, the position coordinates of the third element values comprise row coordinates of respective first element values and column coordinates of respective second element values, the N first column vectors are obtained by converting a first matrix in a compressed format, the N first row vectors are obtained by converting a second matrix in the compressed format, and N is an integer greater than or equal to 1; andaccumulating, based on indexes of the position coordinates of the third element values, third element values with same position coordinates in the N intermediate result matrices to obtain a result matrix.
  • 17. The chip according to claim 16, wherein the N intermediate result matrices comprise at least a first intermediate result matrix and a second intermediate result matrix, position coordinates of third element values in the first intermediate result matrix are first position coordinates, and position coordinates of third element values in the second intermediate result matrix are second position coordinates; and the one or more memories store the programming instructions for execution by the at least one processor to: write, in a generation sequence of the N intermediate result matrices, the third element values in the first intermediate result matrix into corresponding positions in a cache based on the first position coordinates;read, based on the second position coordinates of the third element values in the second intermediate result matrix, cached values that are at positions corresponding to the second position coordinates in the cache; andaccumulate the third element values in the second intermediate result matrix and the cached values to obtain a result matrix in an uncompressed format.
  • 18. The chip according to claim 17, wherein the one or more memories store the programming instructions for execution by the at least one processor to: compress the result matrix in the uncompressed format to obtain a result matrix in a compressed format.
  • 19. The chip according to claim 16, wherein the one or more memories store the programming instructions for execution by the at least one processor to: sort third element values in the N intermediate result matrices based on position coordinates of the third element values;compare position coordinates in N intermediate result matrices obtained through sorting;aggregate third element values, in the N intermediate result matrices obtained through sorting, with same position coordinates; anddelete position coordinates of a zero-element value to obtain a result matrix in a compressed format.
  • 20. The chip according to claim 16, wherein the one or more memories store the programming instructions for execution by the at least one processor to: obtain the first matrix and the second matrix;convert the first matrix into the N first column vectors; andconvert the second matrix into the N first row vectors.
Priority Claims (2)
Number Date Country Kind
202011617575.X Dec 2020 CN national
202110181498.6 Feb 2021 CN national
Continuations (1)
Number Date Country
Parent PCT/CN2021/141000 Dec 2021 US
Child 18343622 US