Jesse Cai 255cd75a97 [sparse] Add cuSPARSELt as a backend (#128534) ... Summary: This PR adds in cuSPARSELt as a backend to PyTorch. It is now possible to see if cuSPARSELt is available and the version if it is with ``` torch.backends.cusparselt.is_available() torch.backends.cusparselt.version() ``` Test Plan: ``` python test/test_sparse_semi_structured.py -k test_cusparselt_backend ``` Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/128534 Approved by: https://github.com/cpuhrsch, https://github.com/eqy, https://github.com/syed-ahmed		2024-08-21 22:06:07 +00:00
..
shared	[sparse] Add cuSPARSELt as a backend (#128534)	2024-08-21 22:06:07 +00:00
comm.cpp	[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)	2024-07-08 07:03:53 +00:00
comm.h	[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)	2024-07-08 07:03:53 +00:00
CUDAPluggableAllocator.cpp	Move handle-creation logic into cudacaching allocator. (#130888)	2024-07-18 21:34:38 +00:00
CUDAPluggableAllocator.h	Fix typo in assignment operators (#131890)	2024-07-27 11:13:42 +00:00
device_set.h
Event.cpp	[2/N] Change #include <c10/util/Optional.h> to #include <optional> (#130236)	2024-07-09 03:17:24 +00:00
Event.h
GdsFile.cpp	[Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489)	2024-08-15 17:11:52 +00:00
GdsFile.h	[Reland] Add wrappers for synchronous GPUDirect Storage APIs (#133489)	2024-08-15 17:11:52 +00:00
Graph.cpp
memory_snapshot.cpp	[Memory Snapshot] Skip C++ warmup unwind() call if context is not set (#133038)	2024-08-13 17:25:24 +00:00
memory_snapshot.h	[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)	2024-07-08 07:03:53 +00:00
MemPool.cpp	Implements torch.cuda.MemPool() API (#131152)	2024-08-01 01:29:30 +00:00
Module.cpp	[sparse] Add cuSPARSELt as a backend (#128534)	2024-08-21 22:06:07 +00:00
Module.h
nccl.cpp	[nccl] Wrap nccl code update with version check (#130419)	2024-08-02 01:22:07 +00:00
nccl.h	[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)	2024-07-08 07:03:53 +00:00
python_comm.cpp
python_comm.h
python_nccl.cpp	[1/N] Change #include <c10/util/Optional.h> to #include <optional> (#128301)	2024-07-08 07:03:53 +00:00
python_nccl.h
Stream.cpp
Stream.h
Tensor.cpp
THCP.h
utils.cpp