From 34262d904cc9976939321191d4c2b848042f5ccc Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Thu, 17 Jun 2021 16:46:57 +0800
Subject: [PATCH] fix(imperative): fix the size of blob with offset

GitOrigin-RevId: bb1736dfe11f0891834645461b15d5fb68b5a6fa
---
 imperative/python/test/unit/jit/test_tracing.py | 2 +-
 imperative/src/impl/physical_tensor.cpp         | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/imperative/python/test/unit/jit/test_tracing.py b/imperative/python/test/unit/jit/test_tracing.py
index 6bc9e339..7fee2ab0 100644
--- a/imperative/python/test/unit/jit/test_tracing.py
+++ b/imperative/python/test/unit/jit/test_tracing.py
@@ -583,7 +583,7 @@ def test_trace_advance_indexing(shape_mode):
 
     inputs = {
         "x": np.random.randn(5, 5, 5, 5, 5).astype("float32"),
-        "i": 0,
+        "i": 4,
         "j": 2,
         "start": 1,
         "end": 3,
diff --git a/imperative/src/impl/physical_tensor.cpp b/imperative/src/impl/physical_tensor.cpp
index c6831de2..ba0b5ae2 100644
--- a/imperative/src/impl/physical_tensor.cpp
+++ b/imperative/src/impl/physical_tensor.cpp
@@ -232,7 +232,7 @@ namespace {
 
 Blob::Blob(const DeviceTensorStorage& s):
     m_comp_node{s.comp_node()}, m_storage{s.raw_storage()},
-    m_size{s.size()} {
+    m_size{s.size() + s.offset()} {
     m_id = next_blob_id++;
     BlobManager::inst()->register_blob(this);
 }