pytorch
diff --git a/‎aten/src/ATen/templates/Functions.cpp‎
Lines changed: 1 addition & 1 deletion b/‎aten/src/ATen/templates/Functions.cpp‎
Lines changed: 1 addition & 1 deletion
diff --git a/‎aten/src/ATen/test/basic.cpp‎
Lines changed: 9 additions & 0 deletions b/‎aten/src/ATen/test/basic.cpp‎
Lines changed: 9 additions & 0 deletions
@@ -75,7 +75,7 @@ Tensor TensorMaker::make_tensor() {
    }
    auto storage_size = size * itemsize;
    if (storage_offset_) {
-     storage_size += storage_offset_.value();
+     storage_size += storage_offset_.value() * itemsize;
    }
    return storage_size;
  }
 
@@ -519,6 +519,15 @@ TEST(BasicTest, BasicStdTestCPU) {
 }
 
 TEST(BasicTest, TestForBlobResizeCPU) {
+  // Checks that for_blob can correctly create tensors with non-empty offset and resize them
+  std::array<int32_t, 6> storage;
+  std::iota(storage.begin(), storage.end(), 1);
+  auto t = at::for_blob(storage.data(), {3,}).storage_offset(3).options(c10::TensorOptions(kInt)).make_tensor();
+  auto te = *at::expand_size(t, {3, 3});
+  ASSERT_EQ(te[1][1].item<int32_t>(), 5);
+}
+
+TEST(BasicTest, TestForBlobStridesResizeCPU) {
   // Checks that for_blob can correctly create tensors with non-empty offset and resize them
   std::array<int32_t, 6> storage;
   std::iota(storage.begin(), storage.end(), 1);
Original file line number	Diff line number	Diff line change
`@@ -75,7 +75,7 @@ Tensor TensorMaker::make_tensor() {`
`75`	`75`	`}`
`76`	`76`	`auto storage_size = size * itemsize;`
`77`	`77`	`if (storage_offset_) {`
`78`		`- storage_size += storage_offset_.value();`
	`78`	`+ storage_size += storage_offset_.value() * itemsize;`
`79`	`79`	`}`
`80`	`80`	`return storage_size;`
`81`	`81`	`}`