fix a bug about offsets of the tensor, test=develop (#31)

Co-authored-by: shixiaowei02 <39303645+Shixiaowei02@users.noreply.github.com>
PaddlePaddle · Oct 27, 2021 · be9df70 · be9df70 · paddle-bot-old · Oct 27, 2021
1 parent 8f100da
commit be9df70
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/paddle/pten/hapi/lib/utils/storage.h b/paddle/pten/hapi/lib/utils/storage.h
@@ -47,10 +47,14 @@ class ExternalStorage : public pten::Storage {
 class SharedStorage : public pten::Storage {
  public:
   explicit SharedStorage(
-      const std::shared_ptr<paddle::memory::Allocation>& allocation)
+      const std::shared_ptr<paddle::memory::Allocation>& allocation,
+      size_t offset)
       : allocation_(allocation) {
     CHECK(allocation);
-    data_ = pten::Allocation(allocation->ptr(), allocation->place());
+    data_ = pten::Allocation(
+        reinterpret_cast<void*>(reinterpret_cast<uintptr_t>(allocation->ptr()) +
+                                offset),
+        allocation->place());
     size_ = allocation->size();
   }
 

diff --git a/paddle/pten/hapi/lib/utils/tensor_utils.cc b/paddle/pten/hapi/lib/utils/tensor_utils.cc
@@ -31,7 +31,8 @@ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensor(
   pten::DenseTensorMeta meta{pten::TransToPtenDataType(src.type()),
                              src.dims(),
                              pten::TransToPtenDataLayout(src.layout())};
-  auto shared_storage = pten::make_intrusive<SharedStorage>(src.Holder());
+  auto shared_storage =
+      pten::make_intrusive<SharedStorage>(src.Holder(), src.offset());
   return std::make_unique<pten::DenseTensor>(std::move(shared_storage),
                                              std::move(meta));
 }
@@ -42,7 +43,8 @@ std::unique_ptr<pten::DenseTensor> MakePtenDenseTensor(
                              src.dims(),
                              pten::TransToPtenDataLayout(src.layout())};
   SetLoD(&meta.lod, src.lod());
-  auto shared_storage = pten::make_intrusive<SharedStorage>(src.Holder());
+  auto shared_storage =
+      pten::make_intrusive<SharedStorage>(src.Holder(), src.offset());
   return std::make_unique<pten::DenseTensor>(std::move(shared_storage),
                                              std::move(meta));
 }