frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-30 19:54:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source 
[ghstack-poisoned]
This commit is contained in:
Nikita Vedeneev
2025-10-29 10:50:25 +00:00
parent c1e7268182 c2eb709432
commit 02fa1ad97a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/native/cuda/Blas.cpp
View File

@ -174,7 +174,7 @@ static bool isInputCompliesAddmmCudaLt(Tensor& result, const Tensor& self, const
// Conditions for bias to be fusable
&& (
self.is_contiguous() &&
// NOTE: fine to have 1-len dims to the left from the leading one
// NOTE: fine to have 1-len dims to the left from the right-most one
(self.dim() == 1 || self.squeeze().dim() == 1) &&
self.sizes().back() == mat2_sizes[1]
)