Use native::resize_output for deprecation warning

pytorch · Apr 4, 2022 · 6ec1929 · 6ec1929
1 parent ca596ca
commit 6ec1929
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/aten/src/ATen/native/sparse/SparseCsrTensorMath.cpp b/aten/src/ATen/native/sparse/SparseCsrTensorMath.cpp
@@ -461,7 +461,7 @@ Tensor& addmm_out_sparse_csr_cpu(
 
   if (&result != &self) {
     if (result.layout() == kStrided) {
-      result.resize_as_(*self_);
+      at::native::resize_output(result, self_->sizes());
     } else {
       result.resize_as_sparse_(*self_);
     }

diff --git a/aten/src/ATen/native/sparse/cuda/SparseBlas.cpp b/aten/src/ATen/native/sparse/cuda/SparseBlas.cpp
@@ -156,7 +156,7 @@ Tensor& addmm_out_sparse_csr_cuda(
 
   if (&result != &self) {
     if (result.layout() == kStrided) {
-      result.resize_as_(*self_);
+      at::native::resize_output(result, self_->sizes());
     } else {
       result.resize_as_sparse_(*self_);
     }