pytorch/torch/sparse
2024-11-26 08:01:24 +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 Revert "[sparse] add extra options to _cslt_spare_mm (#137427)" 2024-11-26 08:01:24 +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