commit 09458a2bf13064d38db6a6ac15c852af642201bb
author mikey dagitses <mikeyd@meta.com> Wed Apr 12 04:12:18 2023 -0700
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Wed Apr 12 18:15:18 2023 +0000
tree c36a2c7db1fafdd57b9cbe3381641d1770237194
parent be8a4eb8e359d08853a821d0d204077222936374

introduce TensorBase::mutable_data_ptr() (#98163)

See D44409928 for motivation.

Note that we keep the const-ness of the existing data_ptr() member so
that we don't have to change all references atomically. We just change
the ones here that we have higher confidence with.

Differential Revision: [D44611466](https://our.internmc.facebook.com/intern/diff/D44611466/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/98163
Approved by: https://github.com/ezyang

aten/src/ATen/core/TensorBase.h

diff --git a/aten/src/ATen/core/TensorBase.h b/aten/src/ATen/core/TensorBase.h
index afb72ff..aab1001 100644
--- a/aten/src/ATen/core/TensorBase.h
+++ b/aten/src/ATen/core/TensorBase.h
@@ -560,10 +560,22 @@
                           .layout(layout());
   }
 
-  void* data_ptr() const {
+  const void* const_data_ptr() const {
+    return this->unsafeGetTensorImpl()->data();
+  }
+
+  void* mutable_data_ptr() const {
     return this->unsafeGetTensorImpl()->mutable_data();
   }
 
+  // TODO(#97856) Make this return a const pointer. This currently
+  //              returns a non-const pointer because of the large
+  //              number of clients that we still want to audit before
+  //              migrating to mutable_data_ptr().
+  void* data_ptr() const {
+    return mutable_data_ptr();
+  }
+
   template <typename T>
   const T* const_data_ptr() const;