Skip to content

Commit

Permalink
Fix uninitialized data and broken broadcasting with sparse.mm and spa…
Browse files Browse the repository at this point in the history
…… (#16572)

Summary:
…rse.addmm.

Fixes pytorch/pytorch#16543.
Pull Request resolved: pytorch/pytorch#16572

Differential Revision: D13884235

Pulled By: gchanan

fbshipit-source-id: 308916051364d72f72ec56f0495c6c7c09845131
  • Loading branch information
gchanan authored and facebook-github-bot committed Jan 31, 2019
1 parent 2facdac commit 261cead
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions aten/src/ATen/native/sparse/SparseTensorMath.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -593,14 +593,16 @@ Tensor _sparse_addmm(
Scalar beta,
Scalar alpha
) {
return at::s_native_addmm(t, sparse, dense, beta, alpha);
Tensor b_t;
std::tie(b_t) = expand_size(t, {sparse.size(0), dense.size(1)}, "addmm");
return at::s_native_addmm(b_t, sparse, dense, beta, alpha);
}

Tensor _sparse_mm(
const SparseTensor& sparse,
const Tensor& dense
) {
Tensor t = at::empty({sparse.size(0), dense.size(1)}, dense.options());
Tensor t = at::zeros({}, dense.options());
return at::_sparse_addmm(t, sparse, dense, 0, 1);
}

Expand Down

0 comments on commit 261cead

Please sign in to comment.