commit 79891abf40c44f661a26cbc6d4d4fc28bdacb7e3
author anjali411 <chourdiaanjali123@gmail.com> Fri Apr 22 18:18:16 2022 +0000
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> Fri Apr 22 20:02:31 2022 +0000
tree b2c8eede92afc15e265f9e03de59eaf3e8b795d8
parent 44bbb247a6658cd3a97d952badb1bc0a46961695

fix UndefinedBehaviorSanitizer

Pull Request resolved: https://github.com/pytorch/pytorch/pull/76235

Approved by: https://github.com/albanD

torch/csrc/utils/tensor_list.cpp

diff --git a/torch/csrc/utils/tensor_list.cpp b/torch/csrc/utils/tensor_list.cpp
index a34b17e..80ac318 100644
--- a/torch/csrc/utils/tensor_list.cpp
+++ b/torch/csrc/utils/tensor_list.cpp
@@ -24,7 +24,9 @@
     PyObject* obj = recursive_to_list(data, sizes, strides, dim + 1, scalarType, elementSize);
     if (!obj) throw python_error();
     PyList_SET_ITEM(list.get(), i, obj);
-    data += strides[dim] * elementSize;
+    auto advance_data_ptr = strides[dim] * elementSize;
+    TORCH_INTERNAL_ASSERT(data || advance_data_ptr == 0);
+    data += advance_data_ptr;
   }
   return list.release();
 }