Nikita Vedeneev 05dda7ff65 bsr_dense_mm Triton kernel: fix out kwarg (#96648) ... As per title. The kernel did not handle `out=` correctly and returned a different tensor which only shared storage with `out`. Pull Request resolved: https://github.com/pytorch/pytorch/pull/96648 Approved by: https://github.com/cpuhrsch		2023-03-14 18:01:22 +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	bsr_dense_mm Triton kernel: fix out kwarg (#96648)	2023-03-14 18:01:22 +00:00