Nikita Vedeneev 39a22e2791 softmax: Triton kernel for BSR inputs (#102095) ... Implements `softmax` Triton kernel for BSR inputs. So far, only over `dim=-1`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/102095 Approved by: https://github.com/cpuhrsch		2023-06-21 01:23:27 +00:00
..
__init__.py	Fix a bug in nesting check_sparse_tensor_invariants context managers (#95372)	2023-02-23 18:22:13 +00:00
_triton_ops.py	softmax: Triton kernel for BSR inputs (#102095)	2023-06-21 01:23:27 +00:00