Home
last modified time | relevance | path

Searched refs:y_impl (Results 1 – 3 of 3) sorted by relevance

/libCEED/backends/cuda-ref/
H A Dceed-cuda-ref-vector.c728 CeedVector_Cuda *y_impl, *x_impl; in CeedVectorAXPY_Cuda() local
730 CeedCallBackend(CeedVectorGetData(y, &y_impl)); in CeedVectorAXPY_Cuda()
734 if (y_impl->d_array) { in CeedVectorAXPY_Cuda()
741 …nCeed(y), cublasSaxpy_64(handle, (int64_t)length, &alpha, x_impl->d_array, 1, y_impl->d_array, 1)); in CeedVectorAXPY_Cuda()
743 …nCeed(y), cublasDaxpy_64(handle, (int64_t)length, &alpha, x_impl->d_array, 1, y_impl->d_array, 1)); in CeedVectorAXPY_Cuda()
746 CeedCallBackend(CeedDeviceAXPY_Cuda(y_impl->d_array, alpha, x_impl->d_array, length)); in CeedVectorAXPY_Cuda()
748 y_impl->h_array = NULL; in CeedVectorAXPY_Cuda()
749 } else if (y_impl->h_array) { in CeedVectorAXPY_Cuda()
751 CeedCallBackend(CeedHostAXPY_Cuda(y_impl->h_array, alpha, x_impl->h_array, length)); in CeedVectorAXPY_Cuda()
752 y_impl->d_array = NULL; in CeedVectorAXPY_Cuda()
[all …]
/libCEED/backends/hip-ref/
H A Dceed-hip-ref-vector.c838 CeedVector_Hip *y_impl, *x_impl; in CeedVectorAXPY_Hip() local
840 CeedCallBackend(CeedVectorGetData(y, &y_impl)); in CeedVectorAXPY_Hip()
844 if (y_impl->d_array) { in CeedVectorAXPY_Hip()
853 …Ceed(y), hipblasSaxpy_64(handle, (int64_t)length, &alpha, x_impl->d_array, 1, y_impl->d_array, 1)); in CeedVectorAXPY_Hip()
855 …Ceed(y), hipblasDaxpy_64(handle, (int64_t)length, &alpha, x_impl->d_array, 1, y_impl->d_array, 1)); in CeedVectorAXPY_Hip()
859 CeedCallBackend(CeedDeviceAXPY_Hip(y_impl->d_array, alpha, x_impl->d_array, length)); in CeedVectorAXPY_Hip()
861 y_impl->h_array = NULL; in CeedVectorAXPY_Hip()
862 } else if (y_impl->h_array) { in CeedVectorAXPY_Hip()
864 CeedCallBackend(CeedHostAXPY_Hip(y_impl->h_array, alpha, x_impl->h_array, length)); in CeedVectorAXPY_Hip()
865 y_impl->d_array = NULL; in CeedVectorAXPY_Hip()
[all …]
/libCEED/backends/sycl-ref/
H A Dceed-sycl-vector.sycl.cpp603 CeedVector_Sycl *y_impl, *x_impl; in CeedVectorAXPY_Sycl() local
608 CeedCallBackend(CeedVectorGetData(y, &y_impl)); in CeedVectorAXPY_Sycl()
613 if (y_impl->d_array) { in CeedVectorAXPY_Sycl()
615 …CeedCallBackend(CeedDeviceAXPY_Sycl(data->sycl_queue, y_impl->d_array, alpha, x_impl->d_array, len… in CeedVectorAXPY_Sycl()
617 if (y_impl->h_array) { in CeedVectorAXPY_Sycl()
619 CeedCallBackend(CeedHostAXPY_Sycl(y_impl->h_array, alpha, x_impl->h_array, length)); in CeedVectorAXPY_Sycl()
650 CeedVector_Sycl *w_impl, *x_impl, *y_impl; in CeedVectorPointwiseMult_Sycl() local
657 CeedCallBackend(CeedVectorGetData(y, &y_impl)); in CeedVectorPointwiseMult_Sycl()
667 …cePointwiseMult_Sycl(data->sycl_queue, w_impl->d_array, x_impl->d_array, y_impl->d_array, length)); in CeedVectorPointwiseMult_Sycl()
672 …CeedCallBackend(CeedHostPointwiseMult_Sycl(w_impl->h_array, x_impl->h_array, y_impl->h_array, leng… in CeedVectorPointwiseMult_Sycl()