Skip to content

Commit b326450

Browse files
committed
Add memory copy to device memory
1 parent 1d82a62 commit b326450

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

dpnp/backend/kernels/dpnp_krnl_indexing.cpp

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,19 +60,19 @@ DPCTLSyclEventRef dpnp_choose_c(DPCTLSyclQueueRef q_ref,
6060

6161
sycl::queue q = *(reinterpret_cast<sycl::queue *>(q_ref));
6262

63-
DPNPC_ptr_adapter<_DataType1> input1_ptr(q_ref, array1_in, size);
63+
DPNPC_ptr_adapter<_DataType1> input1_ptr(q_ref, array1_in, size, true);
6464
_DataType1 *array_in = input1_ptr.get_ptr();
6565

66-
DPNPC_ptr_adapter<_DataType2 *> choices_ptr(q_ref, choices1, choices_size);
66+
DPNPC_ptr_adapter<_DataType2 *> choices_ptr(q_ref, choices1, choices_size, true);
6767
_DataType2 **choices = choices_ptr.get_ptr();
6868

6969
for (size_t i = 0; i < choices_size; ++i) {
7070
DPNPC_ptr_adapter<_DataType2> choice_ptr(q_ref, choices[i],
71-
choice_size);
71+
choice_size, true);
7272
choices[i] = choice_ptr.get_ptr();
7373
}
7474

75-
DPNPC_ptr_adapter<_DataType2> result1_ptr(q_ref, result1, size, false,
75+
DPNPC_ptr_adapter<_DataType2> result1_ptr(q_ref, result1, size, true,
7676
true);
7777
_DataType2 *result = result1_ptr.get_ptr();
7878

@@ -88,6 +88,7 @@ DPCTLSyclEventRef dpnp_choose_c(DPCTLSyclQueueRef q_ref,
8888
};
8989

9090
sycl::event event = q.submit(kernel_func);
91+
result1_ptr.depends_on(event);
9192

9293
event_ref = reinterpret_cast<DPCTLSyclEventRef>(&event);
9394

0 commit comments

Comments
 (0)