File tree Expand file tree Collapse file tree 1 file changed +15
-1
lines changed Expand file tree Collapse file tree 1 file changed +15
-1
lines changed Original file line number Diff line number Diff line change @@ -640,7 +640,21 @@ cdef class usm_ndarray:
640
640
641
641
def to_device (self , target_device ):
642
642
"""
643
- Transfer array to target device
643
+ Transfers this array to specified target device.
644
+
645
+ Args:
646
+ target_device: array API concept of target device.
647
+ target_device can be a oneAPI filter selector string,
648
+ an instance of :class:`dpctl.SyclDevice` corresponding to a
649
+ non-partitioned SYCL device, an instance of
650
+ :class:`dpctl.SyclQueue`, or a :class:`dpctl.tensor.Device`
651
+ object returned by :attr:`dpctl.tensor.usm_array.device`.
652
+
653
+ Returns:
654
+ A view if data copy is not required, and a copy otherwise.
655
+ If copying is required, it is done by copying from original
656
+ allocation device to the host, followed by copying from host
657
+ to the target device.
644
658
"""
645
659
cdef c_dpctl.DPCTLSyclQueueRef QRef = NULL
646
660
cdef c_dpmem._Memory arr_buf
You can’t perform that action at this time.
0 commit comments