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