pytorch/torch/sparse
2024-04-13 07:47:32 +00:00
..
__init__.py Update DimOrDims typing in torch.sparse (#122471) 2024-03-25 16:25:56 +00:00
_semi_structured_conversions.py Revert "[sparse] Add fast semi-structured spasification kernels (#122350)" 2024-04-12 13:26:10 +00:00
_semi_structured_ops.py Revert "[sparse] Add fast semi-structured spasification kernels (#122350)" 2024-04-12 13:26:10 +00:00
_triton_ops_meta.py Update bsr_dense_addmm kernel parameters for sizes 3 x 2 ^ N (#122506) 2024-03-23 11:54:33 +00:00
_triton_ops.py Update bsr_dense_addmm kernel parameters for sizes 3 x 2 ^ N (#122506) 2024-03-23 11:54:33 +00:00
semi_structured.py Revert "Simplify ATen sparse semi-structured operators based on CUTLASS (#123473)" 2024-04-13 07:47:32 +00:00