torch.sparse.mm benchmark script

[rusty1s]

Aggregator: sum, Layout: torch.sparse_coo
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.0291s   | 1.1011s    | 1.1302s |
+--------+-----------+------------+---------+
Aggregator: sum, Layout: torch.sparse_csr
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.0234s   | 1.0986s    | 1.1220s |
+--------+-----------+------------+---------+
Aggregator: sum, Layout: torch.sparse_csc
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 1.0967s   | 0.0306s    | 1.1273s |
+--------+-----------+------------+---------+
Aggregator: mean, Layout: torch.sparse_coo
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.0320s   | 0.9957s    | 1.0277s |
+--------+-----------+------------+---------+
Aggregator: mean, Layout: torch.sparse_csr
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.0395s   | 1.0311s    | 1.0706s |
+--------+-----------+------------+---------+
Aggregator: mean, Layout: torch.sparse_csc
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 1.1745s   | 0.0431s    | 1.2176s |
+--------+-----------+------------+---------+
Aggregator: min, Layout: torch.sparse_coo
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.1950s   | 1.0798s    | 1.2748s |
+--------+-----------+------------+---------+
Aggregator: min, Layout: torch.sparse_csr
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.2009s   | 1.0845s    | 1.2855s |
+--------+-----------+------------+---------+
Aggregator: min, Layout: torch.sparse_csc
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 1.2554s   | 1.0785s    | 2.3339s |
+--------+-----------+------------+---------+
Aggregator: max, Layout: torch.sparse_coo
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.1663s   | 1.0461s    | 1.2124s |
+--------+-----------+------------+---------+
Aggregator: max, Layout: torch.sparse_csr
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 0.1613s   | 1.0435s    | 1.2048s |
+--------+-----------+------------+---------+
Aggregator: max, Layout: torch.sparse_csc
+--------+-----------+------------+---------+
| Name   | Forward   | Backward   | Total   |
|--------+-----------+------------+---------|
| spmm   | 1.2571s   | 1.0804s    | 2.3375s |
+--------+-----------+------------+---------+

[codecov]

Codecov Report

Merging #6950 (a58ba76) into master (b5ecfd9) will decrease coverage by 0.29%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##           master    #6950      +/-   ##
==========================================
- Coverage   91.76%   91.47%   -0.29%     
==========================================
  Files         429      429              
  Lines       23350    23348       -2     
==========================================
- Hits        21427    21358      -69     
- Misses       1923     1990      +67     

Impacted Files	Coverage Δ
torch_geometric/profile/benchmark.py	95.00% <100.00%> (+0.12%)	⬆️
torch_geometric/utils/spmm.py	70.83% <100.00%> (ø)

... and 16 files with indirect coverage changes

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more