We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 662d3ae commit 6a948ffCopy full SHA for 6a948ff
1 file changed
array_api_compat/torch/_aliases.py
@@ -619,6 +619,7 @@ def arange(start: float,
619
return torch.empty(0, device=device, **kwargs).to(dtype)
620
try:
621
return torch.arange(start, stop, step, dtype=dtype, device=device, **kwargs)
622
+ # torch 2.7 raises RuntimeError, 2.9 emits NotImplementedError
623
except (NotImplementedError, RuntimeError):
624
return torch.arange(start, stop, step, device=device, **kwargs).to(dtype)
625
0 commit comments