Jagrit Digani 5ad133f8bb No copy gems (#801) ...

* Enable collapsing batch dims in gemm
* Update gemm to only make copies when neither of the last 2 axes are contiguous
* Update addmm to support gemv shapes
* Update addmm to support irregular batch strides
* Update tests

2024-03-12 13:13:41 -07:00

26 KiB

Raw Blame History

View Raw