pytorch/aten
Nikita Vedeneev 467c21ad9a nn.Linear: nD contiguous input + bias -- dispatch to addmm also when weight is sparse (#166071)
As per title.

It seems safe to be able to generalize to arbitrary contiguous inputs since `at::matmul` is likely to do the flattening to avoid `baddmm`.

Additionally, we guard for bias to be 1D and contiguous which is guaranteed to be fused with no copies.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166071
Approved by: https://github.com/ngimel
2025-10-29 13:13:40 +00:00
..
conda
src nn.Linear: nD contiguous input + bias -- dispatch to addmm also when weight is sparse (#166071) 2025-10-29 13:13:40 +00:00
tools Adds Issue#153109 as a test for CUDAPluggableAllocator (#163575) 2025-10-01 09:07:48 +00:00
CMakeLists.txt Revert "Use official CUDAToolkit module in CMake (#154595)" 2025-06-23 21:15:31 +00:00