Skip to content

Commit b000c5a

Browse files
committed
remove trailing spaces
quick-checks
1 parent 7dce9d3 commit b000c5a

3 files changed

Lines changed: 9 additions & 18 deletions

File tree

benchmarks/sparse/dlmc/README.md

Lines changed: 5 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,15 @@
11
# Sparse benchmarks
22

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].
54

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.
86

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.
148

159
- `matmul_bench.py` with `--operation sparse@vector` is for Sparse matrix-vector multiplication (SPMV) performance test.
1610

1711
References:
1812

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
2214

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

benchmarks/sparse/dlmc/matmul_bench.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
# This benchmark is for sparse matmul performance test.
44
# They exist for comparing the performance of sparse matrix routines
55
# `sparse @ vector`, `sparse @ sparse` and `sparse @ dense` with different backends (CPU/CUDA)
6-
# and with other frameworks such as scipy.
6+
# and with other frameworks such as scipy.
77

88
import sys
99
import argparse

benchmarks/sparse/dlmc/test.sh

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,9 @@
33
DATASET_ROOT_DIR=$HOME/datasets/
44

55
# wget https://storage.googleapis.com/sgk-sc2020/dlmc.tar.gz -P $DATASET_ROOT_DIR
6-
# tar -xvf $DATASET_ROOT_DIR/dlmc.tar.gz
6+
# tar -xvf $DATASET_ROOT_DIR/dlmc.tar.gz
77

8-
echo "!! SPARSE SPMS TIME BENCHMARK!! "
8+
echo "!! SPARSE SPMS TIME BENCHMARK!! "
99

1010
# cpu
1111
python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnitude_pruning --operation sparse@sparse
@@ -14,7 +14,7 @@ python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnit
1414
python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnitude_pruning --operation sparse@dense
1515
python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnitude_pruning --operation sparse@dense --backward_test
1616

17-
python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnitude_pruning --operation sparse@vector
17+
python -m dlmc.matmul_bench --path $DATASET_ROOT_DIR/dlmc/rn50 --dataset magnitude_pruning --operation sparse@vector
1818

1919

2020
# cuda

0 commit comments

Comments
 (0)