Skip to content

Commit d1d2398

Browse files
authored
fix f class on OrangePi (#2216)
1 parent 8809765 commit d1d2398

File tree

3 files changed

+5
-1
lines changed

3 files changed

+5
-1
lines changed

mindtorch/ops/array.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -765,7 +765,10 @@ def _slice_helper(tensor, slice_spec, do_update=False, updates=None):
765765
begin.append(0)
766766
begin_mask |= (1 << index)
767767
if s.stop is not None:
768-
end.append(s.stop)
768+
stop = s.stop
769+
if stop == -1:
770+
stop = tensor.shape[index]
771+
end.append(stop)
769772
else:
770773
end.append(0)
771774
end_mask |= (1 << index)

mindtorch/ops/creation.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -114,6 +114,7 @@ def arange(start=0, end=None, step=1, *, out=None, dtype=None, layout=None, devi
114114
start = start.item() if isinstance(start, (mindtorch.Tensor, np.integer)) else start
115115
end = end.item() if isinstance(end, (mindtorch.Tensor, np.integer)) else end
116116
step = step.item() if isinstance(step, (mindtorch.Tensor, np.integer)) else step
117+
step = type(start)(step)
117118

118119
output = execute('arange', start, end, step, dtype, device=device)
119120
if out is None:

torch4ms/__init__.py

Whitespace-only changes.

0 commit comments

Comments
 (0)