|
1 | 1 | # Sparse benchmarks |
2 | 2 |
|
3 | | -These sets of benchmarks are for the sparse matrix functionality using a popular real dataset collection called |
4 | | -the Deep Learning Matrix Collection (DLMC), which were used in recent studies [1, 2]. |
| 3 | +These sets of benchmarks are for the sparse matrix functionality using a popular real dataset collection called the Deep Learning Matrix Collection (DLMC), which were used in recent studies [1, 2]. |
5 | 4 |
|
6 | | -Performance benchmarks scripts for matrix-matrix and matrix-vector ops |
7 | | -(dense-sparse, sparse-sparse, and compare to dense-dense) are implemented here. |
| 5 | +Performance benchmarks scripts for matrix-matrix and matrix-vector ops (dense-sparse, sparse-sparse, and compare to dense-dense) are implemented here. |
8 | 6 |
|
9 | | -- `matmul_bench.py` with `--operation sparse@sparse|sparse@dense` is for |
10 | | - Sparse matrix-matrix multiplication (SPMM) performance test. |
11 | | - It can run in forward and backward mode with `--backward_test`, on CPU or CUDA with `--with_cuda`, |
12 | | - using different datasets from the dataset collection DLMC. |
13 | | - For more details see `test.sh` file. |
| 7 | +- `matmul_bench.py` with `--operation sparse@sparse|sparse@dense` is for Sparse matrix-matrix multiplication (SPMM) performance test. It can run in forward and backward mode with `--backward_test`, on CPU or CUDA with `--with_cuda`, using different datasets from the dataset collection DLMC. For more details see `test.sh` file. |
14 | 8 |
|
15 | 9 | - `matmul_bench.py` with `--operation sparse@vector` is for Sparse matrix-vector multiplication (SPMV) performance test. |
16 | 10 |
|
17 | 11 | References: |
18 | 12 |
|
19 | | -1. Trevor Gale, Matei Zaharia, Cliff Young, Erich Elsen. Sparse GPU Kernels for Deep Learning. |
20 | | -Proceedings of the International Conference for High Performance Computing, 2020. |
21 | | -https://github.com/google-research/google-research/tree/master/sgk |
| 13 | +1. Trevor Gale, Matei Zaharia, Cliff Young, Erich Elsen. Sparse GPU Kernels for Deep Learning. Proceedings of the International Conference for High Performance Computing, 2020. https://github.com/google-research/google-research/tree/master/sgk |
22 | 14 |
|
23 | | -2. Trevor Gale, Erich Elsen, Sara Hooker. The State of Sparsity in Deep Neural Networks. |
24 | | -https://github.com/google-research/google-research/tree/master/state_of_sparsity |
| 15 | +2. Trevor Gale, Erich Elsen, Sara Hooker. The State of Sparsity in Deep Neural Networks. https://github.com/google-research/google-research/tree/master/state_of_sparsity |
0 commit comments