pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Jesse Cai f1451163ec [sparse] add extra options to _cslt_spare_mm (#137427 )

Summary:

Splitting this PR into two, one for the cuSPARSELt improvements, and one
for the inductor lowering.

This PR adds in the additional cuSPARSELt bindings into pytorch.

* `torch._cslt_sparse_mm_search` will be deprecated in a future PR,
  so a warning has been added

* Added a header file for cuSPARSELtOps.cpp

* max_id is now available in `torch.backends.cusparselt` via
  `torch.backends.cusparselt.get_max_alg_id()`

* fixed meta registrations for float8

Test Plan:

python test/test_sparse_semi_structured.py

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/137427
Approved by: https://github.com/cpuhrsch, https://github.com/eqy

2024-11-25 23:45:41 +00:00

dlmc

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126 )

2024-05-27 14:49:57 +00:00

__init__.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

README.md

Add CSR (compressed sparse row) layout for sparse tensors (#50937 )

2021-04-12 10:09:12 -07:00

spmm.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126 )

2024-05-27 14:49:57 +00:00

spmv.py

Apply UFMT to all files in benchmarks/ (#105928 )

2023-07-26 01:18:48 +00:00

test_csr.sh

[BE] Prefer dash over underscore in command-line options (#94505 )

2023-02-09 20:16:49 +00:00

triton_ops.py

remove fast_flush arguments (#135387 )

2024-09-13 08:13:46 +00:00

utils.py

[5/N][Easy] fix typo for usort config in pyproject.toml (kown -> known): sort torch (#127126 )

2024-05-27 14:49:57 +00:00

README.md

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