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); }