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

..

__init__.py

SparseCsrCUDA: cuDSS backend for linalg.solve (#129856)

2024-08-22 07:57:30 +00:00

_semi_structured_conversions.py

Remove unused Python variables in torch/[b-z]* (#136963)

2024-10-19 16:45:22 +00:00

_semi_structured_ops.py

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

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

_triton_ops_meta.py

[BE]: Apply PERF401 autofixes from ruff (#140980)

2024-11-20 17:52:07 +00:00

_triton_ops.py

Add out_dtype kw argument to optimize_bsr_dense_addmm (#136626)

2024-10-22 09:52:25 +00:00

semi_structured.py

[sparse][semi-structured] Add float8 dtype support to 24 sparsity (#136397)

2024-09-27 21:37:34 +00:00