fix: negative dimensions are not allowed (DeepLink-org#766)

* fix: negative dimensions are not allowed * fix: cover the scalar tensor
xuq7410 · Dec 21, 2023 · f60cad1 · f60cad1
1 parent 24ca8d0
commit f60cad1
Showing 1 changed file with 6 additions and 14 deletions.
diff --git a/diopi_test/python/conformance/diopi_runtime.py b/diopi_test/python/conformance/diopi_runtime.py
@@ -310,22 +310,14 @@ def from_numpy(cls, darray, context=None, device=Device.AIChip):
         return tr
 
     def numpy(self) -> np.ndarray:
+        if all(x == 0 for x in self.size().data) and self.numel == 0:
+            # cases when shape all 0, but not include the scalar tensor
+            return np.empty(self.size().data, to_numpy_dtype(self.get_dtype()))
         data = np.empty((1,), to_numpy_dtype(self.get_dtype()))
         element_size = data.itemsize
-        sumsize = int(
-            sum(
-                [
-                    (s - 1) * st
-                    for s, st in zip(
-                        list(self.size().data),
-                        [
-                            int(stride * element_size)
-                            for stride in self.get_stride().data
-                        ],
-                    )
-                ]
-            ) / element_size + 1
-        )
+        stride_scaled = [int(stride * element_size) for stride in self.get_stride().data]
+        sum_of_products = sum((s - 1) * st for s, st in zip(self.size().data, stride_scaled))
+        sumsize = int(sum_of_products / element_size) + 1
         darray = np.empty(sumsize, to_numpy_dtype(self.get_dtype()))
         PyCapsule_Destructor = ctypes.CFUNCTYPE(None, ctypes.py_object)
         PyCapsule_New = ctypes.pythonapi.PyCapsule_New