fix (#4319)

tinygrad · Apr 28, 2024 · bc36940 · bc36940
1 parent 8d1649d
commit bc36940
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/extra/onnx.py b/extra/onnx.py
@@ -170,7 +170,7 @@ def fetch_tensor(x: str):
           starts, ends = inp[1:3]
           axes = safe_numpy(Tensor.arange(inp[0].ndim) if len(inp) <= 3 else inp[3].cast(dtypes.int32)).tolist()
           steps = safe_numpy(inp[4].cast(dtypes.int32)).tolist() if len(inp) > 4 else [1]*inp[0].ndim
-          starts, ends = safe_numpy(starts.ceil().cast(dtypes.int32)).tolist(), safe_numpy(ends.ceil().cast(dtypes.int32)).tolist()
+          starts, ends = safe_numpy(starts).tolist(), safe_numpy(ends).tolist()
         arg = [(0,x,1) for x in inp[0].shape]
         for i, axis in enumerate(axes):
           axis = int(axis) + inp[0].ndim if axis < 0 else int(axis)