Chinese Academy of Sciences

China Institute of Atomic Energy

Zhongxiao Cao

Wen Wen Yang

Jue Wang

Yue Gao

Xiaoguang Wang

Yangang Wang

Ningming Nie

Fuhai Gao

Yangde Feng

Tiancai Liu

The disclosure relates to a method and a system for predicting the operation time of sparse matrix vector multiplication. The method comprises constructing a convolutional neural network comprising an input layer, a feature processing layer, a data splicing layer and an output layer for outputting prediction results. The method further comprises acquiring a plurality of groups of sparse matrices with known sparse matrix vector multiplication operation time as sample data, inputting the sample data into the convolutional neural network to train the convolutional neural network, and inputting the sparse matrix to be classified into the trained convolutional neural network to realize the prediction of the operation time of sparse matrix vector multiplication.

Method and system for predicting operation time of sparse matrix vector multiplication

Growing community of inventors

Beijing, China

Zhongxiao Cao