Home
last modified time | relevance | path

Searched refs:alpha (Results 1 – 23 of 23) sorted by relevance

/libCEED/backends/magma/
H A Dceed-magma-gemm-nontensor.cpp24 …ns_t trans_A, magma_trans_t trans_B, magma_int_t m, magma_int_t n, magma_int_t k, CeedScalar alpha, in magmablas_gemm() argument
28 …magmablas_sgemm(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda, (const float *)… in magmablas_gemm()
31 …magmablas_dgemm(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldda, (const double… in magmablas_gemm()
39 … CeedScalar alpha, const CeedScalar *d_A, magma_int_t ldda, magma_int_t strideA, in magmablas_gemm_batched_strided() argument
43 …magmablas_sgemm_batched_strided(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda,… in magmablas_gemm_batched_strided()
46 …magmablas_dgemm_batched_strided(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldd… in magmablas_gemm_batched_strided()
53 …ns_t trans_A, magma_trans_t trans_B, magma_int_t m, magma_int_t n, magma_int_t k, CeedScalar alpha, in devblas_gemm() argument
57 …magma_sgemm(trans_A, trans_B, m, n, k, (float)alpha, (const float *)d_A, ldda, (const float *)d_B,… in devblas_gemm()
59 …magma_dgemm(trans_A, trans_B, m, n, k, (double)alpha, (const double *)d_A, ldda, (const double *)d… in devblas_gemm()
67 …CeedScalar alpha, const CeedScalar *d_A, magma_int_t ldda, magma_int_t strideA, const CeedScalar *… in devblas_gemm_batched_strided() argument
[all …]
H A Dceed-magma-gemm-nontensor.h12 …ns_t trans_A, magma_trans_t trans_B, magma_int_t m, magma_int_t n, magma_int_t k, CeedScalar alpha,
/libCEED/backends/sycl-ref/kernels/
H A Dsycl-ref-vector.cpp61 __global__ static void scaleValueK(CeedScalar *__restrict__ x, CeedScalar alpha, CeedInt size) { in scaleValueK() argument
65 x[index] *= alpha; in scaleValueK()
71 extern "C" int CeedDeviceScale_Sycl(CeedScalar *x_array, CeedScalar alpha, CeedInt length) { in CeedDeviceScale_Sycl() argument
77 scaleValueK<<<grid_size, block_size>>>(x_array, alpha, length); in CeedDeviceScale_Sycl()
84 __global__ static void axpyValueK(CeedScalar *__restrict__ y, CeedScalar alpha, CeedScalar *__restr… in axpyValueK() argument
87 y[index] += alpha * x[index]; in axpyValueK()
93 extern "C" int CeedDeviceAXPY_Sycl(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, Ceed… in CeedDeviceAXPY_Sycl() argument
99 axpyValueK<<<grid_size, block_size>>>(y_array, alpha, x_array, length); in CeedDeviceAXPY_Sycl()
/libCEED/backends/cuda-ref/kernels/
H A Dcuda-ref-vector.cu108 __global__ static void scaleValueK(CeedScalar *__restrict__ x, CeedScalar alpha, CeedSize size) { in scaleValueK() argument
111 if (index < size) x[index] *= alpha; in scaleValueK()
117 extern "C" int CeedDeviceScale_Cuda(CeedScalar *x_array, CeedScalar alpha, CeedSize length) { in CeedDeviceScale_Cuda() argument
123 scaleValueK<<<grid_size, block_size>>>(x_array, alpha, length); in CeedDeviceScale_Cuda()
130 __global__ static void axpyValueK(CeedScalar *__restrict__ y, CeedScalar alpha, CeedScalar *__restr… in axpyValueK() argument
133 if (index < size) y[index] += alpha * x[index]; in axpyValueK()
139 extern "C" int CeedDeviceAXPY_Cuda(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, Ceed… in CeedDeviceAXPY_Cuda() argument
145 axpyValueK<<<grid_size, block_size>>>(y_array, alpha, x_array, length); in CeedDeviceAXPY_Cuda()
152 __global__ static void axpbyValueK(CeedScalar *__restrict__ y, CeedScalar alpha, CeedScalar beta, C… in axpbyValueK() argument
157 y[index] += alpha * x[index]; in axpbyValueK()
[all …]
/libCEED/backends/hip-ref/kernels/
H A Dhip-ref-vector.hip.cpp108 __global__ static void scaleValueK(CeedScalar *__restrict__ x, CeedScalar alpha, CeedSize size) { in scaleValueK() argument
111 if (index < size) x[index] *= alpha; in scaleValueK()
117 extern "C" int CeedDeviceScale_Hip(CeedScalar *x_array, CeedScalar alpha, CeedSize length) { in CeedDeviceScale_Hip() argument
123 hipLaunchKernelGGL(scaleValueK, dim3(grid_size), dim3(block_size), 0, 0, x_array, alpha, length); in CeedDeviceScale_Hip()
130 __global__ static void axpyValueK(CeedScalar *__restrict__ y, CeedScalar alpha, CeedScalar *__restr… in axpyValueK() argument
133 if (index < size) y[index] += alpha * x[index]; in axpyValueK()
139 extern "C" int CeedDeviceAXPY_Hip(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedS… in CeedDeviceAXPY_Hip() argument
145 …hipLaunchKernelGGL(axpyValueK, dim3(grid_size), dim3(block_size), 0, 0, y_array, alpha, x_array, l… in CeedDeviceAXPY_Hip()
152 __global__ static void axpbyValueK(CeedScalar *__restrict__ y, CeedScalar alpha, CeedScalar beta, C… in axpbyValueK() argument
157 y[index] += alpha * x[index]; in axpbyValueK()
[all …]
/libCEED/backends/cuda-ref/
H A Dceed-cuda-ref-vector.c669 static int CeedHostScale_Cuda(CeedScalar *x_array, CeedScalar alpha, CeedSize length) { in CeedHostScale_Cuda() argument
670 for (CeedSize i = 0; i < length; i++) x_array[i] *= alpha; in CeedHostScale_Cuda()
677 int CeedDeviceScale_Cuda(CeedScalar *x_array, CeedScalar alpha, CeedSize length);
682 static int CeedVectorScale_Cuda(CeedVector x, CeedScalar alpha) { in CeedVectorScale_Cuda() argument
695 …CeedCallCublas(CeedVectorReturnCeed(x), cublasSscal_64(handle, (int64_t)length, &alpha, impl->d_ar… in CeedVectorScale_Cuda()
697 …CeedCallCublas(CeedVectorReturnCeed(x), cublasDscal_64(handle, (int64_t)length, &alpha, impl->d_ar… in CeedVectorScale_Cuda()
700 CeedCallBackend(CeedDeviceScale_Cuda(impl->d_array, alpha, length)); in CeedVectorScale_Cuda()
704 CeedCallBackend(CeedHostScale_Cuda(impl->h_array, alpha, length)); in CeedVectorScale_Cuda()
713 static int CeedHostAXPY_Cuda(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedSize l… in CeedHostAXPY_Cuda() argument
714 for (CeedSize i = 0; i < length; i++) y_array[i] += alpha * x_array[i]; in CeedHostAXPY_Cuda()
[all …]
/libCEED/backends/hip-ref/
H A Dceed-hip-ref-vector.c775 static int CeedHostScale_Hip(CeedScalar *x_array, CeedScalar alpha, CeedSize length) { in CeedHostScale_Hip() argument
776 for (CeedSize i = 0; i < length; i++) x_array[i] *= alpha; in CeedHostScale_Hip()
783 int CeedDeviceScale_Hip(CeedScalar *x_array, CeedScalar alpha, CeedSize length);
788 static int CeedVectorScale_Hip(CeedVector x, CeedScalar alpha) { in CeedVectorScale_Hip() argument
803 …CeedCallHipblas(CeedVectorReturnCeed(x), hipblasSscal_64(handle, (int64_t)length, &alpha, impl->d_… in CeedVectorScale_Hip()
805 …CeedCallHipblas(CeedVectorReturnCeed(x), hipblasDscal_64(handle, (int64_t)length, &alpha, impl->d_… in CeedVectorScale_Hip()
809 CeedCallBackend(CeedDeviceScale_Hip(impl->d_array, alpha, length)); in CeedVectorScale_Hip()
814 CeedCallBackend(CeedHostScale_Hip(impl->h_array, alpha, length)); in CeedVectorScale_Hip()
823 static int CeedHostAXPY_Hip(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedSize le… in CeedHostAXPY_Hip() argument
824 for (CeedSize i = 0; i < length; i++) y_array[i] += alpha * x_array[i]; in CeedHostAXPY_Hip()
[all …]
/libCEED/backends/sycl-ref/
H A Dceed-sycl-vector.sycl.cpp540 static int CeedHostScale_Sycl(CeedScalar *x_array, CeedScalar alpha, CeedSize length) { in CeedHostScale_Sycl() argument
541 for (CeedSize i = 0; i < length; i++) x_array[i] *= alpha; in CeedHostScale_Sycl()
548 static int CeedDeviceScale_Sycl(sycl::queue &sycl_queue, CeedScalar *x_array, CeedScalar alpha, Cee… in CeedDeviceScale_Sycl() argument
552 sycl_queue.parallel_for(length, e, [=](sycl::id<1> i) { x_array[i] *= alpha; }); in CeedDeviceScale_Sycl()
559 static int CeedVectorScale_Sycl(CeedVector x, CeedScalar alpha) { in CeedVectorScale_Sycl() argument
572 …->d_array) CeedCallBackend(CeedDeviceScale_Sycl(data->sycl_queue, x_impl->d_array, alpha, length)); in CeedVectorScale_Sycl()
573 if (x_impl->h_array) CeedCallBackend(CeedHostScale_Sycl(x_impl->h_array, alpha, length)); in CeedVectorScale_Sycl()
580 static int CeedHostAXPY_Sycl(CeedScalar *y_array, CeedScalar alpha, CeedScalar *x_array, CeedSize l… in CeedHostAXPY_Sycl() argument
581 for (CeedSize i = 0; i < length; i++) y_array[i] += alpha * x_array[i]; in CeedHostAXPY_Sycl()
588 static int CeedDeviceAXPY_Sycl(sycl::queue &sycl_queue, CeedScalar *y_array, CeedScalar alpha, Ceed… in CeedDeviceAXPY_Sycl() argument
[all …]
/libCEED/python/
H A Dceed_vector.py419 def scale(self, alpha): argument
423 err_code = lib.CeedVectorScale(self._pointer[0], alpha)
429 def axpy(self, alpha, x): argument
433 err_code = lib.CeedVectorAXPY(self._pointer[0], alpha, x._pointer[0])
439 def axpby(self, alpha, beta, x): argument
444 self._pointer[0], alpha, beta, x._pointer[0])
/libCEED/tests/
H A Dt580-operator.h14 CEED_QFUNCTION_HELPER int AlphaMatTransposeMatMult2x2(const CeedScalar alpha, const CeedScalar A[2]… in AlphaMatTransposeMatMult2x2() argument
20 C[j][k] += alpha * A[m][j] * B[m][k]; in AlphaMatTransposeMatMult2x2()
/libCEED/rust/libceed/src/
H A Dvector.rs641 pub fn scale(mut self, alpha: crate::Scalar) -> crate::Result<Self> { in scale()
642 self.check_error(unsafe { bind_ceed::CeedVectorScale(self.ptr, alpha) })?; in scale()
668 pub fn axpy(mut self, alpha: crate::Scalar, x: &Vector) -> crate::Result<Self> { in axpy()
669 self.check_error(unsafe { bind_ceed::CeedVectorAXPY(self.ptr, alpha, x.ptr) })?; in axpy()
698 alpha: crate::Scalar, in axpby()
702 self.check_error(unsafe { bind_ceed::CeedVectorAXPBY(self.ptr, alpha, beta, x.ptr) })?; in axpby()
/libCEED/backends/memcheck/
H A Dceed-memcheck-vector.c322 static int CeedVectorScale_Memcheck(CeedVector x, CeedScalar alpha) { in CeedVectorScale_Memcheck() argument
329 for (CeedSize i = 0; i < length; i++) impl->array_allocated[i] *= alpha; in CeedVectorScale_Memcheck()
336 static int CeedVectorAXPY_Memcheck(CeedVector y, CeedScalar alpha, CeedVector x) { in CeedVectorAXPY_Memcheck() argument
344 …for (CeedSize i = 0; i < length; i++) impl_y->array_allocated[i] += alpha * impl_x->array_allocate… in CeedVectorAXPY_Memcheck()
351 static int CeedVectorAXPBY_Memcheck(CeedVector y, CeedScalar alpha, CeedScalar beta, CeedVector x) { in CeedVectorAXPBY_Memcheck() argument
359 …for (CeedSize i = 0; i < length; i++) impl_y->array_allocated[i] = alpha * impl_x->array_allocated… in CeedVectorAXPBY_Memcheck()
/libCEED/examples/fluids/problems/
H A Dstg_shur14.c215 … for (PetscInt i = 0; i < (*stg_ctx)->nmodes; i++) { kappa[i] = kmin * pow((*stg_ctx)->alpha, i); } in GetStgContextData()
226 …CeedScalar u0 = 0.0, alpha = 1.01, stg_dx = -1, stg_h_scale_factor = 1 / user->app_c… in SetupStg() local
234 …PetscCall(PetscOptionsReal("-stg_alpha", "Growth rate of the wavemodes", NULL, alpha, &alpha, NULL… in SetupStg()
248 global_stg_ctx->alpha = alpha; in SetupStg()
/libCEED/examples/fluids/qfunctions/
H A Dstg_shur14_type.h23 CeedScalar alpha; // !< Geometric growth rate of kappa member
H A Ddifferential_filter.h203 …CeedScalar alpha[2] = {Square(dx) * aniso_scale_factor, Square(dy) * aniso_scale_f… in DifferentialFilter_MMS_IC() local
210 alpha[1] *= Square(damping_coeff); in DifferentialFilter_MMS_IC()
215 …CeedScalar phi = phi_bar + alpha[0] * Square(omega6) * sin(6 * omega * x_i[0]) + alpha[1] * Square… in DifferentialFilter_MMS_IC()
H A Dutils.h31 CEED_QFUNCTION_HELPER void ScaleN(CeedScalar *u, const CeedScalar alpha, const CeedInt N) { in ScaleN() argument
32 CeedPragmaSIMD for (CeedInt i = 0; i < N; i++) u[i] *= alpha; in ScaleN()
36 CEED_QFUNCTION_HELPER void SetValueN(CeedScalar *u, const CeedScalar alpha, const CeedInt N) { in SetValueN() argument
37 CeedPragmaSIMD for (CeedInt i = 0; i < N; i++) u[i] = alpha; in SetValueN()
/libCEED/interface/
H A Dceed-vector.c735 int CeedVectorScale(CeedVector x, CeedScalar alpha) { in CeedVectorScale() argument
749 if (x->Scale) return x->Scale(x, alpha); in CeedVectorScale()
754 for (CeedSize i = 0; i < length; i++) x_array[i] *= alpha; in CeedVectorScale()
770 int CeedVectorAXPY(CeedVector y, CeedScalar alpha, CeedVector x) { in CeedVectorAXPY() argument
811 CeedCall(y->AXPY(y, alpha, x)); in CeedVectorAXPY()
822 for (CeedSize i = 0; i < length_y; i++) y_array[i] += alpha * x_array[i]; in CeedVectorAXPY()
841 int CeedVectorAXPBY(CeedVector y, CeedScalar alpha, CeedScalar beta, CeedVector x) { in CeedVectorAXPBY() argument
882 CeedCall(y->AXPBY(y, alpha, beta, x)); in CeedVectorAXPBY()
893 for (CeedSize i = 0; i < length_y; i++) y_array[i] = alpha * x_array[i] + beta * y_array[i]; in CeedVectorAXPBY()
/libCEED/julia/LibCEED.jl/test/
H A Druntests.jl122 alpha = rand(CeedScalar)
124 scale!(cv1, alpha)
125 w1 .*= alpha
132 axpy!(alpha, cv2, cv1)
133 axpy!(alpha, w2, w1)
/libCEED/benchmarks/
H A Dpostprocess_plot.py120 facecolor=colors[i], alpha=0.2)
/libCEED/include/ceed/
H A Dceed.h214 CEED_EXTERN int CeedVectorScale(CeedVector x, CeedScalar alpha);
215 CEED_EXTERN int CeedVectorAXPY(CeedVector y, CeedScalar alpha, CeedVector x);
216 CEED_EXTERN int CeedVectorAXPBY(CeedVector y, CeedScalar alpha, CeedScalar beta, CeedVector x);
/libCEED/julia/LibCEED.jl/src/generated/
H A Dlibceed_bindings.jl239 function CeedVectorScale(x, alpha) argument
240 ccall((:CeedVectorScale, libceed), Cint, (CeedVector, CeedScalar), x, alpha)
243 function CeedVectorAXPY(y, alpha, x) argument
244 ccall((:CeedVectorAXPY, libceed), Cint, (CeedVector, CeedScalar, CeedVector), y, alpha, x)
247 function CeedVectorAXPBY(y, alpha, beta, x) argument
248 …edVectorAXPBY, libceed), Cint, (CeedVector, CeedScalar, CeedScalar, CeedVector), y, alpha, beta, x)
/libCEED/examples/fluids/
H A Dindex.md149 \bm{\dot{q}}_N(\bm q_N) = \alpha \bm q_N + \bm z_N
152 …ior state and $\alpha > 0$, both of which depend on the specific time integration scheme (backward…
157 …\frac{\partial \bm f}{\partial \bm q_N} = \frac{\partial \bm g}{\partial \bm q_N} + \alpha \frac{\…
160 …The scalar "shift" $\alpha$ scales inversely with the time step $\Delta t$, so small time steps re…
748 \kappa^n = \kappa_{\min} (1 + \alpha)^{n-1} \ , \quad \forall n=1, 2, ... , N
/libCEED/examples/nek/bps/
H A Dbps.usr1410 C if (nio.eq.0) write(6,2) iter,enorm,alph(1),pap(1),'alpha'