xref: /libCEED/rust/libceed-sys/c-src/backends/sycl-ref/ceed-sycl-vector.sycl.cpp (revision f5d1e50421556545666f89e18ad21fef6dcea5ba)
1bd882c8aSJames Wright // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2bd882c8aSJames Wright // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3bd882c8aSJames Wright //
4bd882c8aSJames Wright // SPDX-License-Identifier: BSD-2-Clause
5bd882c8aSJames Wright //
6bd882c8aSJames Wright // This file is part of CEED:  http://github.com/ceed
7bd882c8aSJames Wright 
8bd882c8aSJames Wright #include <ceed/backend.h>
9bd882c8aSJames Wright #include <ceed/ceed.h>
10bd882c8aSJames Wright 
11bd882c8aSJames Wright #include <cmath>
12bd882c8aSJames Wright #include <string>
13bd882c8aSJames Wright #include <sycl/sycl.hpp>
14bd882c8aSJames Wright 
15bd882c8aSJames Wright #include "ceed-sycl-ref.hpp"
16bd882c8aSJames Wright 
17bd882c8aSJames Wright //------------------------------------------------------------------------------
18bd882c8aSJames Wright // Check if host/device sync is needed
19bd882c8aSJames Wright //------------------------------------------------------------------------------
20bd882c8aSJames Wright static inline int CeedVectorNeedSync_Sycl(const CeedVector vec, CeedMemType mem_type, bool *need_sync) {
21bd882c8aSJames Wright   bool             has_valid_array = false;
22dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
23dd64fc84SJeremy L Thompson 
24dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
25bd882c8aSJames Wright   CeedCallBackend(CeedVectorHasValidArray(vec, &has_valid_array));
26bd882c8aSJames Wright   switch (mem_type) {
27bd882c8aSJames Wright     case CEED_MEM_HOST:
28bd882c8aSJames Wright       *need_sync = has_valid_array && !impl->h_array;
29bd882c8aSJames Wright       break;
30bd882c8aSJames Wright     case CEED_MEM_DEVICE:
31bd882c8aSJames Wright       *need_sync = has_valid_array && !impl->d_array;
32bd882c8aSJames Wright       break;
33bd882c8aSJames Wright   }
34bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
35bd882c8aSJames Wright }
36bd882c8aSJames Wright 
37bd882c8aSJames Wright //------------------------------------------------------------------------------
38bd882c8aSJames Wright // Sync host to device
39bd882c8aSJames Wright //------------------------------------------------------------------------------
40bd882c8aSJames Wright static inline int CeedVectorSyncH2D_Sycl(const CeedVector vec) {
41bd882c8aSJames Wright   Ceed             ceed;
42bd882c8aSJames Wright   Ceed_Sycl       *data;
43dd64fc84SJeremy L Thompson   CeedSize         length;
44dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
45dd64fc84SJeremy L Thompson 
46dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
47dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
48bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
494e3038a5SJeremy L Thompson   CeedCheck(impl->h_array, ceed, CEED_ERROR_BACKEND, "No valid host data to sync to device");
50bd882c8aSJames Wright 
51bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetLength(vec, &length));
52bd882c8aSJames Wright   if (impl->d_array_borrowed) {
53bd882c8aSJames Wright     impl->d_array = impl->d_array_borrowed;
54bd882c8aSJames Wright   } else if (impl->d_array_owned) {
55bd882c8aSJames Wright     impl->d_array = impl->d_array_owned;
56bd882c8aSJames Wright   } else {
57bd882c8aSJames Wright     CeedCallSycl(ceed, impl->d_array_owned = sycl::malloc_device<CeedScalar>(length, data->sycl_device, data->sycl_context));
58bd882c8aSJames Wright     impl->d_array = impl->d_array_owned;
59bd882c8aSJames Wright   }
60bd882c8aSJames Wright 
61bd882c8aSJames Wright   sycl::event e = data->sycl_queue.ext_oneapi_submit_barrier();
62bd882c8aSJames Wright   // Copy from host to device
63bd882c8aSJames Wright   sycl::event copy_event = data->sycl_queue.copy<CeedScalar>(impl->h_array, impl->d_array, length, {e});
64bd882c8aSJames Wright   // Wait for copy to finish and handle exceptions.
65bd882c8aSJames Wright   CeedCallSycl(ceed, copy_event.wait_and_throw());
66bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
67bd882c8aSJames Wright }
68bd882c8aSJames Wright 
69bd882c8aSJames Wright //------------------------------------------------------------------------------
70bd882c8aSJames Wright // Sync device to host
71bd882c8aSJames Wright //------------------------------------------------------------------------------
72bd882c8aSJames Wright static inline int CeedVectorSyncD2H_Sycl(const CeedVector vec) {
73bd882c8aSJames Wright   Ceed             ceed;
74bd882c8aSJames Wright   Ceed_Sycl       *data;
75dd64fc84SJeremy L Thompson   CeedSize         length;
76dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
77dd64fc84SJeremy L Thompson 
78dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
79dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
80bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
81bd882c8aSJames Wright 
82bd882c8aSJames Wright   CeedCheck(impl->d_array, ceed, CEED_ERROR_BACKEND, "No valid device data to sync to host");
83bd882c8aSJames Wright 
84bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetLength(vec, &length));
85bd882c8aSJames Wright   if (impl->h_array_borrowed) {
86bd882c8aSJames Wright     impl->h_array = impl->h_array_borrowed;
87bd882c8aSJames Wright   } else if (impl->h_array_owned) {
88bd882c8aSJames Wright     impl->h_array = impl->h_array_owned;
89bd882c8aSJames Wright   } else {
90bd882c8aSJames Wright     CeedCallBackend(CeedCalloc(length, &impl->h_array_owned));
91bd882c8aSJames Wright     impl->h_array = impl->h_array_owned;
92bd882c8aSJames Wright   }
93bd882c8aSJames Wright 
94bd882c8aSJames Wright   // Order queue
95bd882c8aSJames Wright   sycl::event e = data->sycl_queue.ext_oneapi_submit_barrier();
96bd882c8aSJames Wright   // Copy from device to host
97bd882c8aSJames Wright   sycl::event copy_event = data->sycl_queue.copy<CeedScalar>(impl->d_array, impl->h_array, length, {e});
98bd882c8aSJames Wright   // Wait for copy to finish and handle exceptions.
99bd882c8aSJames Wright   CeedCallSycl(ceed, copy_event.wait_and_throw());
100bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
101bd882c8aSJames Wright }
102bd882c8aSJames Wright 
103bd882c8aSJames Wright //------------------------------------------------------------------------------
104bd882c8aSJames Wright // Sync arrays
105bd882c8aSJames Wright //------------------------------------------------------------------------------
106bd882c8aSJames Wright static int CeedVectorSyncArray_Sycl(const CeedVector vec, CeedMemType mem_type) {
107bd882c8aSJames Wright   bool need_sync = false;
108dd64fc84SJeremy L Thompson 
109dd64fc84SJeremy L Thompson   // Check whether device/host sync is needed
110bd882c8aSJames Wright   CeedCallBackend(CeedVectorNeedSync_Sycl(vec, mem_type, &need_sync));
111bd882c8aSJames Wright   if (!need_sync) return CEED_ERROR_SUCCESS;
112bd882c8aSJames Wright 
113bd882c8aSJames Wright   switch (mem_type) {
114bd882c8aSJames Wright     case CEED_MEM_HOST:
115bd882c8aSJames Wright       return CeedVectorSyncD2H_Sycl(vec);
116bd882c8aSJames Wright     case CEED_MEM_DEVICE:
117bd882c8aSJames Wright       return CeedVectorSyncH2D_Sycl(vec);
118bd882c8aSJames Wright   }
119bd882c8aSJames Wright   return CEED_ERROR_UNSUPPORTED;
120bd882c8aSJames Wright }
121bd882c8aSJames Wright 
122bd882c8aSJames Wright //------------------------------------------------------------------------------
123bd882c8aSJames Wright // Set all pointers as invalid
124bd882c8aSJames Wright //------------------------------------------------------------------------------
125bd882c8aSJames Wright static inline int CeedVectorSetAllInvalid_Sycl(const CeedVector vec) {
126bd882c8aSJames Wright   CeedVector_Sycl *impl;
127bd882c8aSJames Wright 
128dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
129bd882c8aSJames Wright   impl->h_array = NULL;
130bd882c8aSJames Wright   impl->d_array = NULL;
131bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
132bd882c8aSJames Wright }
133bd882c8aSJames Wright 
134bd882c8aSJames Wright //------------------------------------------------------------------------------
135bd882c8aSJames Wright // Check if CeedVector has any valid pointer
136bd882c8aSJames Wright //------------------------------------------------------------------------------
137bd882c8aSJames Wright static inline int CeedVectorHasValidArray_Sycl(const CeedVector vec, bool *has_valid_array) {
138bd882c8aSJames Wright   CeedVector_Sycl *impl;
139dd64fc84SJeremy L Thompson 
140bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(vec, &impl));
1411c66c397SJeremy L Thompson   *has_valid_array = impl->h_array || impl->d_array;
142bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
143bd882c8aSJames Wright }
144bd882c8aSJames Wright 
145bd882c8aSJames Wright //------------------------------------------------------------------------------
146bd882c8aSJames Wright // Check if has array of given type
147bd882c8aSJames Wright //------------------------------------------------------------------------------
148bd882c8aSJames Wright static inline int CeedVectorHasArrayOfType_Sycl(const CeedVector vec, CeedMemType mem_type, bool *has_array_of_type) {
149bd882c8aSJames Wright   CeedVector_Sycl *impl;
150bd882c8aSJames Wright 
151dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
152bd882c8aSJames Wright   switch (mem_type) {
153bd882c8aSJames Wright     case CEED_MEM_HOST:
1541c66c397SJeremy L Thompson       *has_array_of_type = impl->h_array_borrowed || impl->h_array_owned;
155bd882c8aSJames Wright       break;
156bd882c8aSJames Wright     case CEED_MEM_DEVICE:
1571c66c397SJeremy L Thompson       *has_array_of_type = impl->d_array_borrowed || impl->d_array_owned;
158bd882c8aSJames Wright       break;
159bd882c8aSJames Wright   }
160bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
161bd882c8aSJames Wright }
162bd882c8aSJames Wright 
163bd882c8aSJames Wright //------------------------------------------------------------------------------
164bd882c8aSJames Wright // Check if has borrowed array of given type
165bd882c8aSJames Wright //------------------------------------------------------------------------------
166bd882c8aSJames Wright static inline int CeedVectorHasBorrowedArrayOfType_Sycl(const CeedVector vec, CeedMemType mem_type, bool *has_borrowed_array_of_type) {
167bd882c8aSJames Wright   CeedVector_Sycl *impl;
168bd882c8aSJames Wright 
169dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
170bd882c8aSJames Wright   switch (mem_type) {
171bd882c8aSJames Wright     case CEED_MEM_HOST:
1721c66c397SJeremy L Thompson       *has_borrowed_array_of_type = impl->h_array_borrowed;
173bd882c8aSJames Wright       break;
174bd882c8aSJames Wright     case CEED_MEM_DEVICE:
1751c66c397SJeremy L Thompson       *has_borrowed_array_of_type = impl->d_array_borrowed;
176bd882c8aSJames Wright       break;
177bd882c8aSJames Wright   }
178bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
179bd882c8aSJames Wright }
180bd882c8aSJames Wright 
181bd882c8aSJames Wright //------------------------------------------------------------------------------
182bd882c8aSJames Wright // Set array from host
183bd882c8aSJames Wright //------------------------------------------------------------------------------
184bd882c8aSJames Wright static int CeedVectorSetArrayHost_Sycl(const CeedVector vec, const CeedCopyMode copy_mode, CeedScalar *array) {
185f59ebe5eSJeremy L Thompson   CeedSize         length;
186bd882c8aSJames Wright   CeedVector_Sycl *impl;
187bd882c8aSJames Wright 
188dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
189f59ebe5eSJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(vec, &length));
190f59ebe5eSJeremy L Thompson 
191*f5d1e504SJeremy L Thompson   CeedCallBackend(CeedSetHostCeedScalarArray(array, copy_mode, length, (const CeedScalar **)&impl->h_array_owned,
192*f5d1e504SJeremy L Thompson                                              (const CeedScalar **)&impl->h_array_borrowed, (const CeedScalar **)&impl->h_array));
193bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
194bd882c8aSJames Wright }
195bd882c8aSJames Wright 
196bd882c8aSJames Wright //------------------------------------------------------------------------------
197bd882c8aSJames Wright // Set array from device
198bd882c8aSJames Wright //------------------------------------------------------------------------------
199bd882c8aSJames Wright static int CeedVectorSetArrayDevice_Sycl(const CeedVector vec, const CeedCopyMode copy_mode, CeedScalar *array) {
200f59ebe5eSJeremy L Thompson   CeedSize         length;
201bd882c8aSJames Wright   Ceed             ceed;
202bd882c8aSJames Wright   Ceed_Sycl       *data;
203dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
204dd64fc84SJeremy L Thompson 
205dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
206dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
207bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
208f59ebe5eSJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(vec, &length));
209bd882c8aSJames Wright 
210bd882c8aSJames Wright   // Order queue
211bd882c8aSJames Wright   sycl::event e = data->sycl_queue.ext_oneapi_submit_barrier();
212bd882c8aSJames Wright 
213bd882c8aSJames Wright   switch (copy_mode) {
214bd882c8aSJames Wright     case CEED_COPY_VALUES: {
215f59ebe5eSJeremy L Thompson       if (!impl->d_array_owned)
216bd882c8aSJames Wright         CeedCallSycl(ceed, impl->d_array_owned = sycl::malloc_device<CeedScalar>(length, data->sycl_device, data->sycl_context));
217bd882c8aSJames Wright       if (array) {
218bd882c8aSJames Wright         sycl::event copy_event = data->sycl_queue.copy<CeedScalar>(array, impl->d_array, length, {e});
219bd882c8aSJames Wright         // Wait for copy to finish and handle exceptions.
220bd882c8aSJames Wright         CeedCallSycl(ceed, copy_event.wait_and_throw());
221bd882c8aSJames Wright       }
222f59ebe5eSJeremy L Thompson       impl->d_array_borrowed = NULL;
223f59ebe5eSJeremy L Thompson       impl->d_array          = impl->d_array_owned;
224bd882c8aSJames Wright     } break;
225bd882c8aSJames Wright     case CEED_OWN_POINTER:
226bd882c8aSJames Wright       if (impl->d_array_owned) {
227bd882c8aSJames Wright         // Wait for all work to finish before freeing memory
228bd882c8aSJames Wright         CeedCallSycl(ceed, data->sycl_queue.wait_and_throw());
229bd882c8aSJames Wright         CeedCallSycl(ceed, sycl::free(impl->d_array_owned, data->sycl_context));
230bd882c8aSJames Wright       }
231bd882c8aSJames Wright       impl->d_array_owned    = array;
232bd882c8aSJames Wright       impl->d_array_borrowed = NULL;
233f59ebe5eSJeremy L Thompson       impl->d_array          = impl->d_array_owned;
234bd882c8aSJames Wright       break;
235bd882c8aSJames Wright     case CEED_USE_POINTER:
236bd882c8aSJames Wright       if (impl->d_array_owned) {
237bd882c8aSJames Wright         // Wait for all work to finish before freeing memory
238bd882c8aSJames Wright         CeedCallSycl(ceed, data->sycl_queue.wait_and_throw());
239bd882c8aSJames Wright         CeedCallSycl(ceed, sycl::free(impl->d_array_owned, data->sycl_context));
240bd882c8aSJames Wright       }
241bd882c8aSJames Wright       impl->d_array_owned    = NULL;
242bd882c8aSJames Wright       impl->d_array_borrowed = array;
243f59ebe5eSJeremy L Thompson       impl->d_array          = impl->d_array_borrowed;
244bd882c8aSJames Wright       break;
245bd882c8aSJames Wright   }
246bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
247bd882c8aSJames Wright }
248bd882c8aSJames Wright 
249bd882c8aSJames Wright //------------------------------------------------------------------------------
250bd882c8aSJames Wright // Set the array used by a vector,
251bd882c8aSJames Wright //   freeing any previously allocated array if applicable
252bd882c8aSJames Wright //------------------------------------------------------------------------------
253bd882c8aSJames Wright static int CeedVectorSetArray_Sycl(const CeedVector vec, const CeedMemType mem_type, const CeedCopyMode copy_mode, CeedScalar *array) {
254bd882c8aSJames Wright   CeedVector_Sycl *impl;
255dd64fc84SJeremy L Thompson 
256bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(vec, &impl));
257bd882c8aSJames Wright 
258bd882c8aSJames Wright   CeedCallBackend(CeedVectorSetAllInvalid_Sycl(vec));
259bd882c8aSJames Wright   switch (mem_type) {
260bd882c8aSJames Wright     case CEED_MEM_HOST:
261bd882c8aSJames Wright       return CeedVectorSetArrayHost_Sycl(vec, copy_mode, array);
262bd882c8aSJames Wright     case CEED_MEM_DEVICE:
263bd882c8aSJames Wright       return CeedVectorSetArrayDevice_Sycl(vec, copy_mode, array);
264bd882c8aSJames Wright   }
265bd882c8aSJames Wright   return CEED_ERROR_UNSUPPORTED;
266bd882c8aSJames Wright }
267bd882c8aSJames Wright 
268bd882c8aSJames Wright //------------------------------------------------------------------------------
269bd882c8aSJames Wright // Set host array to value
270bd882c8aSJames Wright //------------------------------------------------------------------------------
2716ca0f394SUmesh Unnikrishnan static int CeedHostSetValue_Sycl(CeedScalar *h_array, CeedSize length, CeedScalar val) {
2726ca0f394SUmesh Unnikrishnan   for (CeedSize i = 0; i < length; i++) h_array[i] = val;
273bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
274bd882c8aSJames Wright }
275bd882c8aSJames Wright 
276bd882c8aSJames Wright //------------------------------------------------------------------------------
277bd882c8aSJames Wright // Set device array to value
278bd882c8aSJames Wright //------------------------------------------------------------------------------
2796ca0f394SUmesh Unnikrishnan static int CeedDeviceSetValue_Sycl(sycl::queue &sycl_queue, CeedScalar *d_array, CeedSize length, CeedScalar val) {
280bd882c8aSJames Wright   // Order queue
281bd882c8aSJames Wright   sycl::event e = sycl_queue.ext_oneapi_submit_barrier();
282bd882c8aSJames Wright   sycl_queue.fill(d_array, val, length, {e});
283bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
284bd882c8aSJames Wright }
285bd882c8aSJames Wright 
286bd882c8aSJames Wright //------------------------------------------------------------------------------
287bd882c8aSJames Wright // Set a vector to a value,
288bd882c8aSJames Wright //------------------------------------------------------------------------------
289bd882c8aSJames Wright static int CeedVectorSetValue_Sycl(CeedVector vec, CeedScalar val) {
290bd882c8aSJames Wright   Ceed             ceed;
291bd882c8aSJames Wright   Ceed_Sycl       *data;
292dd64fc84SJeremy L Thompson   CeedSize         length;
293dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
294dd64fc84SJeremy L Thompson 
295dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
296dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
297dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(vec, &length));
298bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
299bd882c8aSJames Wright 
300bd882c8aSJames Wright   // Set value for synced device/host array
301bd882c8aSJames Wright   if (!impl->d_array && !impl->h_array) {
302bd882c8aSJames Wright     if (impl->d_array_borrowed) {
303bd882c8aSJames Wright       impl->d_array = impl->d_array_borrowed;
304bd882c8aSJames Wright     } else if (impl->h_array_borrowed) {
305bd882c8aSJames Wright       impl->h_array = impl->h_array_borrowed;
306bd882c8aSJames Wright     } else if (impl->d_array_owned) {
307bd882c8aSJames Wright       impl->d_array = impl->d_array_owned;
308bd882c8aSJames Wright     } else if (impl->h_array_owned) {
309bd882c8aSJames Wright       impl->h_array = impl->h_array_owned;
310bd882c8aSJames Wright     } else {
311bd882c8aSJames Wright       CeedCallBackend(CeedVectorSetArray(vec, CEED_MEM_DEVICE, CEED_COPY_VALUES, NULL));
312bd882c8aSJames Wright     }
313bd882c8aSJames Wright   }
314bd882c8aSJames Wright   if (impl->d_array) {
315bd882c8aSJames Wright     CeedCallBackend(CeedDeviceSetValue_Sycl(data->sycl_queue, impl->d_array, length, val));
316bd882c8aSJames Wright     impl->h_array = NULL;
317bd882c8aSJames Wright   }
318bd882c8aSJames Wright   if (impl->h_array) {
319bd882c8aSJames Wright     CeedCallBackend(CeedHostSetValue_Sycl(impl->h_array, length, val));
320bd882c8aSJames Wright     impl->d_array = NULL;
321bd882c8aSJames Wright   }
322bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
323bd882c8aSJames Wright }
324bd882c8aSJames Wright 
325bd882c8aSJames Wright //------------------------------------------------------------------------------
326bd882c8aSJames Wright // Vector Take Array
327bd882c8aSJames Wright //------------------------------------------------------------------------------
328bd882c8aSJames Wright static int CeedVectorTakeArray_Sycl(CeedVector vec, CeedMemType mem_type, CeedScalar **array) {
329bd882c8aSJames Wright   Ceed             ceed;
330bd882c8aSJames Wright   Ceed_Sycl       *data;
331dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
332dd64fc84SJeremy L Thompson 
333dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
334dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
335bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
336bd882c8aSJames Wright 
337bd882c8aSJames Wright   // Order queue
338bd882c8aSJames Wright   data->sycl_queue.ext_oneapi_submit_barrier();
339bd882c8aSJames Wright 
340bd882c8aSJames Wright   // Sync array to requested mem_type
341bd882c8aSJames Wright   CeedCallBackend(CeedVectorSyncArray(vec, mem_type));
342bd882c8aSJames Wright 
343bd882c8aSJames Wright   // Update pointer
344bd882c8aSJames Wright   switch (mem_type) {
345bd882c8aSJames Wright     case CEED_MEM_HOST:
346bd882c8aSJames Wright       (*array)               = impl->h_array_borrowed;
347bd882c8aSJames Wright       impl->h_array_borrowed = NULL;
348bd882c8aSJames Wright       impl->h_array          = NULL;
349bd882c8aSJames Wright       break;
350bd882c8aSJames Wright     case CEED_MEM_DEVICE:
351bd882c8aSJames Wright       (*array)               = impl->d_array_borrowed;
352bd882c8aSJames Wright       impl->d_array_borrowed = NULL;
353bd882c8aSJames Wright       impl->d_array          = NULL;
354bd882c8aSJames Wright       break;
355bd882c8aSJames Wright   }
356bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
357bd882c8aSJames Wright }
358bd882c8aSJames Wright 
359bd882c8aSJames Wright //------------------------------------------------------------------------------
360bd882c8aSJames Wright // Core logic for array syncronization for GetArray.
361bd882c8aSJames Wright //   If a different memory type is most up to date, this will perform a copy
362bd882c8aSJames Wright //------------------------------------------------------------------------------
363bd882c8aSJames Wright static int CeedVectorGetArrayCore_Sycl(const CeedVector vec, const CeedMemType mem_type, CeedScalar **array) {
364bd882c8aSJames Wright   CeedVector_Sycl *impl;
365dd64fc84SJeremy L Thompson 
366bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(vec, &impl));
367bd882c8aSJames Wright 
368bd882c8aSJames Wright   // Sync array to requested mem_type
369bd882c8aSJames Wright   CeedCallBackend(CeedVectorSyncArray(vec, mem_type));
370bd882c8aSJames Wright 
371bd882c8aSJames Wright   // Update pointer
372bd882c8aSJames Wright   switch (mem_type) {
373bd882c8aSJames Wright     case CEED_MEM_HOST:
374bd882c8aSJames Wright       *array = impl->h_array;
375bd882c8aSJames Wright       break;
376bd882c8aSJames Wright     case CEED_MEM_DEVICE:
377bd882c8aSJames Wright       *array = impl->d_array;
378bd882c8aSJames Wright       break;
379bd882c8aSJames Wright   }
380bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
381bd882c8aSJames Wright }
382ff1e7120SSebastian Grimberg 
383bd882c8aSJames Wright //------------------------------------------------------------------------------
384bd882c8aSJames Wright // Get read-only access to a vector via the specified mem_type
385bd882c8aSJames Wright //------------------------------------------------------------------------------
386bd882c8aSJames Wright static int CeedVectorGetArrayRead_Sycl(const CeedVector vec, const CeedMemType mem_type, const CeedScalar **array) {
387bd882c8aSJames Wright   return CeedVectorGetArrayCore_Sycl(vec, mem_type, (CeedScalar **)array);
388bd882c8aSJames Wright }
389bd882c8aSJames Wright 
390bd882c8aSJames Wright //------------------------------------------------------------------------------
391bd882c8aSJames Wright // Get read/write access to a vector via the specified mem_type
392bd882c8aSJames Wright //------------------------------------------------------------------------------
393bd882c8aSJames Wright static int CeedVectorGetArray_Sycl(const CeedVector vec, const CeedMemType mem_type, CeedScalar **array) {
394bd882c8aSJames Wright   CeedVector_Sycl *impl;
395dd64fc84SJeremy L Thompson 
396bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(vec, &impl));
397bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayCore_Sycl(vec, mem_type, array));
398bd882c8aSJames Wright   CeedCallBackend(CeedVectorSetAllInvalid_Sycl(vec));
399bd882c8aSJames Wright   switch (mem_type) {
400bd882c8aSJames Wright     case CEED_MEM_HOST:
401bd882c8aSJames Wright       impl->h_array = *array;
402bd882c8aSJames Wright       break;
403bd882c8aSJames Wright     case CEED_MEM_DEVICE:
404bd882c8aSJames Wright       impl->d_array = *array;
405bd882c8aSJames Wright       break;
406bd882c8aSJames Wright   }
407bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
408bd882c8aSJames Wright }
409bd882c8aSJames Wright 
410bd882c8aSJames Wright //------------------------------------------------------------------------------
411bd882c8aSJames Wright // Get write access to a vector via the specified mem_type
412bd882c8aSJames Wright //------------------------------------------------------------------------------
413bd882c8aSJames Wright static int CeedVectorGetArrayWrite_Sycl(const CeedVector vec, const CeedMemType mem_type, CeedScalar **array) {
414bd882c8aSJames Wright   bool             has_array_of_type = true;
415dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
416dd64fc84SJeremy L Thompson 
417dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
418bd882c8aSJames Wright   CeedCallBackend(CeedVectorHasArrayOfType_Sycl(vec, mem_type, &has_array_of_type));
419bd882c8aSJames Wright   if (!has_array_of_type) {
420bd882c8aSJames Wright     // Allocate if array is not yet allocated
421bd882c8aSJames Wright     CeedCallBackend(CeedVectorSetArray(vec, mem_type, CEED_COPY_VALUES, NULL));
422bd882c8aSJames Wright   } else {
423bd882c8aSJames Wright     // Select dirty array
424bd882c8aSJames Wright     switch (mem_type) {
425bd882c8aSJames Wright       case CEED_MEM_HOST:
426bd882c8aSJames Wright         if (impl->h_array_borrowed) impl->h_array = impl->h_array_borrowed;
427bd882c8aSJames Wright         else impl->h_array = impl->h_array_owned;
428bd882c8aSJames Wright         break;
429bd882c8aSJames Wright       case CEED_MEM_DEVICE:
430bd882c8aSJames Wright         if (impl->d_array_borrowed) impl->d_array = impl->d_array_borrowed;
431bd882c8aSJames Wright         else impl->d_array = impl->d_array_owned;
432bd882c8aSJames Wright     }
433bd882c8aSJames Wright   }
434bd882c8aSJames Wright   return CeedVectorGetArray_Sycl(vec, mem_type, array);
435bd882c8aSJames Wright }
436bd882c8aSJames Wright 
437bd882c8aSJames Wright //------------------------------------------------------------------------------
438bd882c8aSJames Wright // Get the norm of a CeedVector
439bd882c8aSJames Wright //------------------------------------------------------------------------------
440bd882c8aSJames Wright static int CeedVectorNorm_Sycl(CeedVector vec, CeedNormType type, CeedScalar *norm) {
441bd882c8aSJames Wright   Ceed              ceed;
442bd882c8aSJames Wright   Ceed_Sycl        *data;
443dd64fc84SJeremy L Thompson   CeedSize          length;
444dd64fc84SJeremy L Thompson   const CeedScalar *d_array;
445dd64fc84SJeremy L Thompson   CeedVector_Sycl  *impl;
446dd64fc84SJeremy L Thompson 
447dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
448dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
449dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(vec, &length));
450bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
451bd882c8aSJames Wright 
452bd882c8aSJames Wright   // Compute norm
453bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetArrayRead(vec, CEED_MEM_DEVICE, &d_array));
454bd882c8aSJames Wright   switch (type) {
455bd882c8aSJames Wright     case CEED_NORM_1: {
456bd882c8aSJames Wright       // Order queue
457bd882c8aSJames Wright       sycl::event e            = data->sycl_queue.ext_oneapi_submit_barrier();
458bd882c8aSJames Wright       auto        sumReduction = sycl::reduction(impl->reduction_norm, sycl::plus<>(), {sycl::property::reduction::initialize_to_identity{}});
459bd882c8aSJames Wright       data->sycl_queue.parallel_for(length, {e}, sumReduction, [=](sycl::id<1> i, auto &sum) { sum += abs(d_array[i]); }).wait_and_throw();
460bd882c8aSJames Wright     } break;
461bd882c8aSJames Wright     case CEED_NORM_2: {
462bd882c8aSJames Wright       // Order queue
463bd882c8aSJames Wright       sycl::event e            = data->sycl_queue.ext_oneapi_submit_barrier();
464bd882c8aSJames Wright       auto        sumReduction = sycl::reduction(impl->reduction_norm, sycl::plus<>(), {sycl::property::reduction::initialize_to_identity{}});
465bd882c8aSJames Wright       data->sycl_queue.parallel_for(length, {e}, sumReduction, [=](sycl::id<1> i, auto &sum) { sum += (d_array[i] * d_array[i]); }).wait_and_throw();
466bd882c8aSJames Wright     } break;
467bd882c8aSJames Wright     case CEED_NORM_MAX: {
468bd882c8aSJames Wright       // Order queue
469bd882c8aSJames Wright       sycl::event e            = data->sycl_queue.ext_oneapi_submit_barrier();
470bd882c8aSJames Wright       auto        maxReduction = sycl::reduction(impl->reduction_norm, sycl::maximum<>(), {sycl::property::reduction::initialize_to_identity{}});
471bd882c8aSJames Wright       data->sycl_queue.parallel_for(length, {e}, maxReduction, [=](sycl::id<1> i, auto &max) { max.combine(abs(d_array[i])); }).wait_and_throw();
472bd882c8aSJames Wright     } break;
473bd882c8aSJames Wright   }
474bd882c8aSJames Wright   // L2 norm - square root over reduced value
475bd882c8aSJames Wright   if (type == CEED_NORM_2) *norm = sqrt(*impl->reduction_norm);
476bd882c8aSJames Wright   else *norm = *impl->reduction_norm;
477bd882c8aSJames Wright   CeedCallBackend(CeedVectorRestoreArrayRead(vec, &d_array));
478bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
479bd882c8aSJames Wright }
480bd882c8aSJames Wright 
481bd882c8aSJames Wright //------------------------------------------------------------------------------
482bd882c8aSJames Wright // Take reciprocal of a vector on host
483bd882c8aSJames Wright //------------------------------------------------------------------------------
4846ca0f394SUmesh Unnikrishnan static int CeedHostReciprocal_Sycl(CeedScalar *h_array, CeedSize length) {
4856ca0f394SUmesh Unnikrishnan   for (CeedSize i = 0; i < length; i++) {
486bd882c8aSJames Wright     if (std::fabs(h_array[i]) > CEED_EPSILON) h_array[i] = 1. / h_array[i];
487bd882c8aSJames Wright   }
488bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
489bd882c8aSJames Wright }
490bd882c8aSJames Wright 
491bd882c8aSJames Wright //------------------------------------------------------------------------------
492bd882c8aSJames Wright // Take reciprocal of a vector on device
493bd882c8aSJames Wright //------------------------------------------------------------------------------
4946ca0f394SUmesh Unnikrishnan static int CeedDeviceReciprocal_Sycl(sycl::queue &sycl_queue, CeedScalar *d_array, CeedSize length) {
495bd882c8aSJames Wright   // Order queue
496bd882c8aSJames Wright   sycl::event e = sycl_queue.ext_oneapi_submit_barrier();
497bd882c8aSJames Wright   sycl_queue.parallel_for(length, {e}, [=](sycl::id<1> i) {
498bd882c8aSJames Wright     if (std::fabs(d_array[i]) > CEED_EPSILON) d_array[i] = 1. / d_array[i];
499bd882c8aSJames Wright   });
500bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
501bd882c8aSJames Wright }
502bd882c8aSJames Wright 
503bd882c8aSJames Wright //------------------------------------------------------------------------------
504bd882c8aSJames Wright // Take reciprocal of a vector
505bd882c8aSJames Wright //------------------------------------------------------------------------------
506bd882c8aSJames Wright static int CeedVectorReciprocal_Sycl(CeedVector vec) {
507bd882c8aSJames Wright   Ceed             ceed;
508bd882c8aSJames Wright   Ceed_Sycl       *data;
509dd64fc84SJeremy L Thompson   CeedSize         length;
510dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
511dd64fc84SJeremy L Thompson 
512dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
513dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
514dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(vec, &length));
515bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
516bd882c8aSJames Wright 
517bd882c8aSJames Wright   // Set value for synced device/host array
518bd882c8aSJames Wright   if (impl->d_array) CeedCallBackend(CeedDeviceReciprocal_Sycl(data->sycl_queue, impl->d_array, length));
519bd882c8aSJames Wright   if (impl->h_array) CeedCallBackend(CeedHostReciprocal_Sycl(impl->h_array, length));
520bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
521bd882c8aSJames Wright }
522bd882c8aSJames Wright 
523bd882c8aSJames Wright //------------------------------------------------------------------------------
524bd882c8aSJames Wright // Compute x = alpha x on the host
525bd882c8aSJames Wright //------------------------------------------------------------------------------
5266ca0f394SUmesh Unnikrishnan static int CeedHostScale_Sycl(CeedScalar *x_array, CeedScalar alpha, CeedSize length) {
5276ca0f394SUmesh Unnikrishnan   for (CeedSize i = 0; i < length; i++) x_array[i] *= alpha;
528bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
529bd882c8aSJames Wright }
530bd882c8aSJames Wright 
531bd882c8aSJames Wright //------------------------------------------------------------------------------
532bd882c8aSJames Wright // Compute x = alpha x on device
533bd882c8aSJames Wright //------------------------------------------------------------------------------
5346ca0f394SUmesh Unnikrishnan static int CeedDeviceScale_Sycl(sycl::queue &sycl_queue, CeedScalar *x_array, CeedScalar alpha, CeedSize length) {
535bd882c8aSJames Wright   // Order queue
536bd882c8aSJames Wright   sycl::event e = sycl_queue.ext_oneapi_submit_barrier();
537bd882c8aSJames Wright   sycl_queue.parallel_for(length, {e}, [=](sycl::id<1> i) { x_array[i] *= alpha; });
538bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
539bd882c8aSJames Wright }
540bd882c8aSJames Wright 
541bd882c8aSJames Wright //------------------------------------------------------------------------------
542bd882c8aSJames Wright // Compute x = alpha x
543bd882c8aSJames Wright //------------------------------------------------------------------------------
544bd882c8aSJames Wright static int CeedVectorScale_Sycl(CeedVector x, CeedScalar alpha) {
545bd882c8aSJames Wright   Ceed             ceed;
546bd882c8aSJames Wright   Ceed_Sycl       *data;
547dd64fc84SJeremy L Thompson   CeedSize         length;
548dd64fc84SJeremy L Thompson   CeedVector_Sycl *x_impl;
549dd64fc84SJeremy L Thompson 
550dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(x, &ceed));
551dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(x, &x_impl));
552dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetLength(x, &length));
553bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
554bd882c8aSJames Wright 
555bd882c8aSJames Wright   // Set value for synced device/host array
556bd882c8aSJames Wright   if (x_impl->d_array) CeedCallBackend(CeedDeviceScale_Sycl(data->sycl_queue, x_impl->d_array, alpha, length));
557bd882c8aSJames Wright   if (x_impl->h_array) CeedCallBackend(CeedHostScale_Sycl(x_impl->h_array, alpha, length));
558bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
559bd882c8aSJames Wright }
560bd882c8aSJames Wright 
561bd882c8aSJames Wright //------------------------------------------------------------------------------
562bd882c8aSJames Wright // Compute y = alpha x + y on the host
563bd882c8aSJames Wright //------------------------------------------------------------------------------
5646ca0f394SUmesh Unnikrishnan static int CeedHostAXPY_Sycl(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedSize length) {
5656ca0f394SUmesh Unnikrishnan   for (CeedSize i = 0; i < length; i++) y_array[i] += alpha * x_array[i];
566bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
567bd882c8aSJames Wright }
568bd882c8aSJames Wright 
569bd882c8aSJames Wright //------------------------------------------------------------------------------
570bd882c8aSJames Wright // Compute y = alpha x + y on device
571bd882c8aSJames Wright //------------------------------------------------------------------------------
5726ca0f394SUmesh Unnikrishnan static int CeedDeviceAXPY_Sycl(sycl::queue &sycl_queue, CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedSize length) {
573bd882c8aSJames Wright   // Order queue
574bd882c8aSJames Wright   sycl::event e = sycl_queue.ext_oneapi_submit_barrier();
575bd882c8aSJames Wright   sycl_queue.parallel_for(length, {e}, [=](sycl::id<1> i) { y_array[i] += alpha * x_array[i]; });
576bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
577bd882c8aSJames Wright }
578bd882c8aSJames Wright 
579bd882c8aSJames Wright //------------------------------------------------------------------------------
580bd882c8aSJames Wright // Compute y = alpha x + y
581bd882c8aSJames Wright //------------------------------------------------------------------------------
582bd882c8aSJames Wright static int CeedVectorAXPY_Sycl(CeedVector y, CeedScalar alpha, CeedVector x) {
583bd882c8aSJames Wright   Ceed             ceed;
584dd64fc84SJeremy L Thompson   Ceed_Sycl       *data;
585dd64fc84SJeremy L Thompson   CeedSize         length;
586bd882c8aSJames Wright   CeedVector_Sycl *y_impl, *x_impl;
587dd64fc84SJeremy L Thompson 
588dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(y, &ceed));
589bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(y, &y_impl));
590bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(x, &x_impl));
591bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetLength(y, &length));
592bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
593bd882c8aSJames Wright 
594bd882c8aSJames Wright   // Set value for synced device/host array
595bd882c8aSJames Wright   if (y_impl->d_array) {
596bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(x, CEED_MEM_DEVICE));
597bd882c8aSJames Wright     CeedCallBackend(CeedDeviceAXPY_Sycl(data->sycl_queue, y_impl->d_array, alpha, x_impl->d_array, length));
598bd882c8aSJames Wright   }
599bd882c8aSJames Wright   if (y_impl->h_array) {
600bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(x, CEED_MEM_HOST));
601bd882c8aSJames Wright     CeedCallBackend(CeedHostAXPY_Sycl(y_impl->h_array, alpha, x_impl->h_array, length));
602bd882c8aSJames Wright   }
603bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
604bd882c8aSJames Wright }
605bd882c8aSJames Wright 
606bd882c8aSJames Wright //------------------------------------------------------------------------------
607bd882c8aSJames Wright // Compute the pointwise multiplication w = x .* y on the host
608bd882c8aSJames Wright //------------------------------------------------------------------------------
6096ca0f394SUmesh Unnikrishnan static int CeedHostPointwiseMult_Sycl(CeedScalar *w_array, CeedScalar *x_array, CeedScalar *y_array, CeedSize length) {
6106ca0f394SUmesh Unnikrishnan   for (CeedSize i = 0; i < length; i++) w_array[i] = x_array[i] * y_array[i];
611bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
612bd882c8aSJames Wright }
613bd882c8aSJames Wright 
614bd882c8aSJames Wright //------------------------------------------------------------------------------
615bd882c8aSJames Wright // Compute the pointwise multiplication w = x .* y on device (impl in .cu file)
616bd882c8aSJames Wright //------------------------------------------------------------------------------
6176ca0f394SUmesh Unnikrishnan static int CeedDevicePointwiseMult_Sycl(sycl::queue &sycl_queue, CeedScalar *w_array, CeedScalar *x_array, CeedScalar *y_array, CeedSize length) {
618bd882c8aSJames Wright   // Order queue
619bd882c8aSJames Wright   sycl::event e = sycl_queue.ext_oneapi_submit_barrier();
620bd882c8aSJames Wright   sycl_queue.parallel_for(length, {e}, [=](sycl::id<1> i) { w_array[i] = x_array[i] * y_array[i]; });
621bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
622bd882c8aSJames Wright }
623bd882c8aSJames Wright 
624bd882c8aSJames Wright //------------------------------------------------------------------------------
625bd882c8aSJames Wright // Compute the pointwise multiplication w = x .* y
626bd882c8aSJames Wright //------------------------------------------------------------------------------
627bd882c8aSJames Wright static int CeedVectorPointwiseMult_Sycl(CeedVector w, CeedVector x, CeedVector y) {
628bd882c8aSJames Wright   Ceed             ceed;
629dd64fc84SJeremy L Thompson   Ceed_Sycl       *data;
630dd64fc84SJeremy L Thompson   CeedSize         length;
631bd882c8aSJames Wright   CeedVector_Sycl *w_impl, *x_impl, *y_impl;
632dd64fc84SJeremy L Thompson 
633dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(w, &ceed));
634bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(w, &w_impl));
635bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(x, &x_impl));
636bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetData(y, &y_impl));
637bd882c8aSJames Wright   CeedCallBackend(CeedVectorGetLength(w, &length));
638bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
639bd882c8aSJames Wright 
640bd882c8aSJames Wright   // Set value for synced device/host array
641bd882c8aSJames Wright   if (!w_impl->d_array && !w_impl->h_array) {
642bd882c8aSJames Wright     CeedCallBackend(CeedVectorSetValue(w, 0.0));
643bd882c8aSJames Wright   }
644bd882c8aSJames Wright   if (w_impl->d_array) {
645bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(x, CEED_MEM_DEVICE));
646bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(y, CEED_MEM_DEVICE));
647bd882c8aSJames Wright     CeedCallBackend(CeedDevicePointwiseMult_Sycl(data->sycl_queue, w_impl->d_array, x_impl->d_array, y_impl->d_array, length));
648bd882c8aSJames Wright   }
649bd882c8aSJames Wright   if (w_impl->h_array) {
650bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(x, CEED_MEM_HOST));
651bd882c8aSJames Wright     CeedCallBackend(CeedVectorSyncArray(y, CEED_MEM_HOST));
652bd882c8aSJames Wright     CeedCallBackend(CeedHostPointwiseMult_Sycl(w_impl->h_array, x_impl->h_array, y_impl->h_array, length));
653bd882c8aSJames Wright   }
654bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
655bd882c8aSJames Wright }
656bd882c8aSJames Wright 
657bd882c8aSJames Wright //------------------------------------------------------------------------------
658bd882c8aSJames Wright // Destroy the vector
659bd882c8aSJames Wright //------------------------------------------------------------------------------
660bd882c8aSJames Wright static int CeedVectorDestroy_Sycl(const CeedVector vec) {
661bd882c8aSJames Wright   Ceed             ceed;
662bd882c8aSJames Wright   Ceed_Sycl       *data;
663dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
664dd64fc84SJeremy L Thompson 
665dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
666dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetData(vec, &impl));
667bd882c8aSJames Wright   CeedCallBackend(CeedGetData(ceed, &data));
668bd882c8aSJames Wright 
669bd882c8aSJames Wright   // Wait for all work to finish before freeing memory
670bd882c8aSJames Wright   CeedCallSycl(ceed, data->sycl_queue.wait_and_throw());
671bd882c8aSJames Wright   CeedCallSycl(ceed, sycl::free(impl->d_array_owned, data->sycl_context));
672bd882c8aSJames Wright   CeedCallSycl(ceed, sycl::free(impl->reduction_norm, data->sycl_context));
673bd882c8aSJames Wright 
674bd882c8aSJames Wright   CeedCallBackend(CeedFree(&impl->h_array_owned));
675bd882c8aSJames Wright   CeedCallBackend(CeedFree(&impl));
676bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
677bd882c8aSJames Wright }
678bd882c8aSJames Wright 
679bd882c8aSJames Wright //------------------------------------------------------------------------------
680bd882c8aSJames Wright // Create a vector of the specified length (does not allocate memory)
681bd882c8aSJames Wright //------------------------------------------------------------------------------
682bd882c8aSJames Wright int CeedVectorCreate_Sycl(CeedSize n, CeedVector vec) {
683bd882c8aSJames Wright   Ceed             ceed;
684bd882c8aSJames Wright   Ceed_Sycl       *data;
685dd64fc84SJeremy L Thompson   CeedVector_Sycl *impl;
686bd882c8aSJames Wright 
687dd64fc84SJeremy L Thompson   CeedCallBackend(CeedVectorGetCeed(vec, &ceed));
688dd64fc84SJeremy L Thompson   CeedCallBackend(CeedGetData(ceed, &data));
689bd882c8aSJames Wright   CeedCallBackend(CeedCalloc(1, &impl));
690bd882c8aSJames Wright   CeedCallSycl(ceed, impl->reduction_norm = sycl::malloc_host<CeedScalar>(1, data->sycl_context));
691bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "HasValidArray", CeedVectorHasValidArray_Sycl));
692bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "HasBorrowedArrayOfType", CeedVectorHasBorrowedArrayOfType_Sycl));
693bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "SetArray", CeedVectorSetArray_Sycl));
694bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "TakeArray", CeedVectorTakeArray_Sycl));
695bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "SetValue", CeedVectorSetValue_Sycl));
696bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "SyncArray", CeedVectorSyncArray_Sycl));
697bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "GetArray", CeedVectorGetArray_Sycl));
698bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "GetArrayRead", CeedVectorGetArrayRead_Sycl));
699bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "GetArrayWrite", CeedVectorGetArrayWrite_Sycl));
700bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "Norm", CeedVectorNorm_Sycl));
701bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "Reciprocal", CeedVectorReciprocal_Sycl));
702bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "AXPY", CeedVectorAXPY_Sycl));
703bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "Scale", CeedVectorScale_Sycl));
704bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "PointwiseMult", CeedVectorPointwiseMult_Sycl));
705bd882c8aSJames Wright   CeedCallBackend(CeedSetBackendFunctionCpp(ceed, "Vector", vec, "Destroy", CeedVectorDestroy_Sycl));
706bd882c8aSJames Wright   CeedCallBackend(CeedVectorSetData(vec, impl));
707bd882c8aSJames Wright   return CEED_ERROR_SUCCESS;
708bd882c8aSJames Wright }
709