Fix overflow issue

Author: Abhinav-Khot

pytensor/tensor/basic.py

@@ -3224,6 +3224,8 @@ def __init__(self, dtype):
         self.dtype = dtype
 
     def make_node(self, start, stop, step):
+        from math import ceil
+
         types = TensorConstant
         shape = (None,)
         # if it is possible to directly determine the shape i.e static shape is present, we find it.
@@ -3232,11 +3234,9 @@ def make_node(self, start, stop, step):
             and isinstance(stop, types)
             and isinstance(step, types)
         ):
+            # Convert to int32 or int64 before calculation
             length = max(
-                np.max(np.ceil((stop.value - start.value) / step.value))
-                .astype(int)
-                .item(),
-                0,
+                ceil((float(stop.value) - float(start.value)) / float(step.value)), 0
             )
             shape = (length,)
 

tests/tensor/test_basic.py

@@ -2866,6 +2866,7 @@ def test_static_shape(self):
         assert np.arange(10, 1, -1).shape == arange(10, 1, -1).type.shape
         assert np.arange(1, -9, 2).shape == arange(1, -9, 2).type.shape
         assert np.arange(1.3, 17.48, 2.67).shape == arange(1.3, 17.48, 2.67).type.shape
+        assert np.arange(-64, 64).shape == arange(-64, 64).type.shape
 
 
 class TestNdGrid: