pytorch/benchmarks/sparse
Aaron Gokaslan bd10fea79a [BE]: Enable F821 and fix bugs (#116579)
Fixes #112371

I tried to fix as many of the bugs as I could, a few I could not figure out what the proper fix for them was though and so I left them with noqas.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/116579
Approved by: https://github.com/ezyang
2024-01-01 08:40:46 +00:00
..
dlmc Apply UFMT to all files in benchmarks/ (#105928) 2023-07-26 01:18:48 +00:00
__init__.py Apply UFMT to all files in benchmarks/ (#105928) 2023-07-26 01:18:48 +00:00
benchmark_semi_structured_sparsity.py [sparse] Add padding for dense matrices in semi-structured sparse (#110583) 2023-10-13 20:04:23 +00:00
README.md
spmm.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
spmv.py Apply UFMT to all files in benchmarks/ (#105928) 2023-07-26 01:18:48 +00:00
test_csr.sh
triton_ops.py Use weakref in storing tensors as keys (follow-up to #111470) (#112076) 2023-10-30 19:16:05 +00:00
utils.py Apply UFMT to all files in benchmarks/ (#105928) 2023-07-26 01:18:48 +00:00

#Sparse benchmarks

These sets of benchmarks are for the sparse matrix functionality. They exist for comparing the performance of sparse matrix routines such as SpMV between various sparse matrix formats and with other frameworks such as TensorFlow.