commit efa16c20c38c6eebeef337a600db904e81b3ffd4
author mikey dagitses <mikeyd@meta.com> Wed Apr 19 17:48:54 2023 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed Apr 19 17:48:54 2023 +0000
tree 147f09b20a297dcab3881c2a3732b38b22504ee9
parent 5cb788a9a5c00a20116e8f6141d2a8146eb0f887

make ATen/native/cuda/UpSampleNearest3d.cu data_ptr-correct (#99328)

make ATen/native/cuda/UpSampleNearest3d.cu data_ptr-correct

Test Plan: Rely on CI.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/99328
Approved by: https://github.com/ezyang

aten/src/ATen/native/cuda/UpSampleNearest3d.cu

diff --git a/aten/src/ATen/native/cuda/UpSampleNearest3d.cu b/aten/src/ATen/native/cuda/UpSampleNearest3d.cu
index 8dde1c1..f9c1dfd 100644
--- a/aten/src/ATen/native/cuda/UpSampleNearest3d.cu
+++ b/aten/src/ATen/native/cuda/UpSampleNearest3d.cu
@@ -187,8 +187,8 @@
   AT_DISPATCH_FLOATING_TYPES_AND3(ScalarType::Half, ScalarType::BFloat16, ScalarType::Byte,input.scalar_type(), "upsample_nearest3d_out_frame", [&] {
         using accscalar_t = at::acc_type<scalar_t, true>;
 
-        auto idata = input.data_ptr<scalar_t>();
-        auto odata = output_c.data_ptr<scalar_t>();
+        auto idata = input.const_data_ptr<scalar_t>();
+        auto odata = output_c.mutable_data_ptr<scalar_t>();
 
         const float depth_scale = compute_scales_value<float>(scales_d, input_depth, output_depth);
         const float height_scale = compute_scales_value<float>(scales_h, input_height, output_height);
@@ -260,8 +260,8 @@
   AT_DISPATCH_FLOATING_TYPES_AND3(ScalarType::Half, ScalarType::BFloat16, ScalarType::Byte, grad_output.scalar_type(), "upsample_nearest3d_backward_out_frame", [&] {
         using accscalar_t = at::acc_type<scalar_t, true>;
 
-        auto idata = grad_input.data_ptr<scalar_t>();
-        auto odata = grad_output.data_ptr<scalar_t>();
+        auto idata = grad_input.mutable_data_ptr<scalar_t>();
+        auto odata = grad_output.const_data_ptr<scalar_t>();
 
         float depth_scale = compute_scales_value_backwards<float>(scales_d, output_depth, input_depth);
         float height_scale = compute_scales_value_backwards<float>(scales_h, output_height, input_height);