Skip to content

Commit 311c781

Browse files
committed
quick fix
1 parent a509212 commit 311c781

File tree

3 files changed

+19
-18
lines changed

3 files changed

+19
-18
lines changed

docs/src/docs/internals/parallel.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ get_device
2323
get_parent_task
2424
```
2525

26-
## Data Movement
26+
## Data Transfer
2727

2828
Before entering a parallel loop, a tensor may reside on a single task, or
2929
represent a single view of data distributed across multiple tasks, or represent
@@ -59,4 +59,4 @@ scatter_recv
5959
gather
6060
gather_send
6161
gather_recv
62-
```
62+
```

src/abstract_tensor.jl

-16
Original file line numberDiff line numberDiff line change
@@ -93,22 +93,6 @@ function similar in spirit to `Base.resize!`.
9393
"""
9494
function virtual_resize! end
9595

96-
"""
97-
moveto(arr, device)
98-
99-
If the array is not on the given device, it creates a new version of this array on that device
100-
and copies the data in to it, according to the `device` trait.
101-
"""
102-
function moveto end
103-
104-
"""
105-
virtual_moveto(device, arr)
106-
107-
If the virtual array is not on the given device, copy the array to that device. This
108-
function may modify underlying data arrays, but cannot change the virtual itself. This
109-
function is used to move data to the device before a kernel is launched.
110-
"""
111-
function virtual_moveto end
11296

11397
struct LabelledTree
11498
key

src/architecture.jl

+17
Original file line numberDiff line numberDiff line change
@@ -319,3 +319,20 @@ function virtual_parallel_region(f, ctx, device::VirtualCPU)
319319
end
320320
end
321321
end
322+
323+
"""
324+
moveto(arr, device)
325+
326+
If the array is not on the given device, it creates a new version of this array on that device
327+
and copies the data in to it, according to the `device` trait.
328+
"""
329+
function moveto end
330+
331+
"""
332+
virtual_moveto(device, arr)
333+
334+
If the virtual array is not on the given device, copy the array to that device. This
335+
function may modify underlying data arrays, but cannot change the virtual itself. This
336+
function is used to move data to the device before a kernel is launched.
337+
"""
338+
function virtual_moveto end

0 commit comments

Comments
 (0)