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