Jesse Cai 5accae4197 [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-27 05:32:45 +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-27 05:32:45 +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