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
__init__.py
benchmark_semi_structured_sparsity.py
README.md
spmm.py [BE]: Enable F821 and fix bugs (#116579) 2024-01-01 08:40:46 +00:00
spmv.py
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

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