@@ -25,25 +25,25 @@ void inline sparse_indices_to_result_dtype_inplace(
25
25
const at::Tensor& input) {
26
26
auto [compressed_indices, plain_indices] =
27
27
at::sparse_csr::getCompressedPlainIndices (input);
28
- static_cast <at::SparseCsrTensorImpl*>(input.unsafeGetTensorImpl ())
29
- ->set_member_tensors (
30
- compressed_indices.to (dtype),
31
- plain_indices.to (dtype),
32
- input.values (),
33
- input.sizes ());
28
+ static_cast <at::SparseCsrTensorImpl*>(input.unsafeGetTensorImpl ())
29
+ ->set_member_tensors (
30
+ compressed_indices.to (dtype),
31
+ plain_indices.to (dtype),
32
+ input.values (),
33
+ input.sizes ());
34
34
}
35
35
36
36
void inline sparse_indices_and_values_resize (
37
37
const at::Tensor& input,
38
38
int64_t nnz) {
39
39
auto [compressed_indices, plain_indices] =
40
40
at::sparse_csr::getCompressedPlainIndices (input);
41
- static_cast <SparseCsrTensorImpl*>(input.unsafeGetTensorImpl ())
42
- ->set_member_tensors (
43
- compressed_indices,
44
- plain_indices.resize_ ({nnz}),
45
- input.values ().resize_ ({nnz}),
46
- input.sizes ());
41
+ static_cast <SparseCsrTensorImpl*>(input.unsafeGetTensorImpl ())
42
+ ->set_member_tensors (
43
+ compressed_indices,
44
+ plain_indices.resize_ ({nnz}),
45
+ input.values ().resize_ ({nnz}),
46
+ input.sizes ());
47
47
}
48
48
49
49
template <typename scalar_t , int eigen_options, typename index_t >
@@ -245,7 +245,7 @@ void addmm_out_sparse_eigen(
245
245
});
246
246
}
247
247
248
- } // anonymus namespace
248
+ } // anonymous namespace
249
249
250
250
void addmm_out_sparse (
251
251
const at::Tensor& mat1,
0 commit comments