Skip to content

Commit 66106cb

Browse files
authored
Merge pull request #252 from crusaderky/ndonnx_device
ndonnx: Revert change to `to_device`
2 parents 84390f5 + a7dfb2e commit 66106cb

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

array_api_compat/common/_helpers.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -776,7 +776,7 @@ def to_device(x: Array, device: Device, /, *, stream: Optional[Union[int, Any]]
776776
device : Hardware device the array data resides on.
777777
778778
"""
779-
if is_numpy_array(x) or is_ndonnx_array(x):
779+
if is_numpy_array(x):
780780
if stream is not None:
781781
raise ValueError("The stream argument to to_device() is not supported")
782782
if device == 'cpu':

0 commit comments

Comments
 (0)