Home
last modified time | relevance | path

Searched refs:dim_2 (Results 1 – 5 of 5) sorted by relevance

/libCEED/include/ceed/jit-source/cuda/
H A Dcuda-ref-basis-tensor-at-points.h284 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in GradAtPoints() local
287 … const CeedScalar *in = dim_2 == 0 ? s_chebyshev_coeffs : (dim_2 % 2 ? buffer_2 : buffer_1); in GradAtPoints()
288 … CeedScalar *out = dim_2 == BASIS_DIM - 1 ? (cur_v + p) : (dim_2 % 2 ? buffer_1 : buffer_2); in GradAtPoints()
291 …if (dim_1 == dim_2) ChebyshevDerivativeAtPoint<BASIS_Q_1D>(coords[elem * v_stride + dim_2 * v_comp… in GradAtPoints()
292 …else ChebyshevPolynomialsAtPoint<BASIS_Q_1D>(coords[elem * v_stride + dim_2 * v_comp_stride + p], … in GradAtPoints()
357 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in GradTransposeAtPoints() local
360 const CeedScalar *in = dim_2 == 0 ? (cur_u + p) : (dim_2 % 2 ? buffer_2 : buffer_1); in GradTransposeAtPoints()
361 …CeedScalar *out = dim_2 == BASIS_DIM - 1 ? s_chebyshev_coeffs : (dim_2 % 2 ? buffer_1 : buff… in GradTransposeAtPoints()
364 …if (dim_1 == dim_2) ChebyshevDerivativeAtPoint<BASIS_Q_1D>(coords[elem * u_stride + dim_2 * u_comp… in GradTransposeAtPoints()
365 …else ChebyshevPolynomialsAtPoint<BASIS_Q_1D>(coords[elem * u_stride + dim_2 * u_comp_stride + p], … in GradTransposeAtPoints()
[all …]
H A Dcuda-ref-basis-tensor.h112 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in Grad() local
116 const CeedScalar *op = dim_1 == dim_2 ? s_grad_1d : s_interp_1d; in Grad()
117 const CeedScalar *in = dim_2 == 0 ? cur_u : (dim_2 % 2 ? s_buffer_2 : s_buffer_1); in Grad()
118 …CeedScalar *out = dim_2 == BASIS_DIM - 1 ? cur_v : (dim_2 % 2 ? s_buffer_1 : s_buffer_2… in Grad()
129 if (is_transpose && dim_2 == BASIS_DIM - 1) out[k] += v_k; in Grad()
/libCEED/include/ceed/jit-source/hip/
H A Dhip-ref-basis-tensor-at-points.h284 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in GradAtPoints() local
287 … const CeedScalar *in = dim_2 == 0 ? s_chebyshev_coeffs : (dim_2 % 2 ? buffer_2 : buffer_1); in GradAtPoints()
288 … CeedScalar *out = dim_2 == BASIS_DIM - 1 ? (&cur_v[p]) : (dim_2 % 2 ? buffer_1 : buffer_2); in GradAtPoints()
291 …if (dim_1 == dim_2) ChebyshevDerivativeAtPoint<BASIS_Q_1D>(coords[elem * v_stride + dim_2 * v_comp… in GradAtPoints()
292 …else ChebyshevPolynomialsAtPoint<BASIS_Q_1D>(coords[elem * v_stride + dim_2 * v_comp_stride + p], … in GradAtPoints()
357 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in GradTransposeAtPoints() local
360 const CeedScalar *in = dim_2 == 0 ? (&cur_u[p]) : (dim_2 % 2 ? buffer_2 : buffer_1); in GradTransposeAtPoints()
361 …CeedScalar *out = dim_2 == BASIS_DIM - 1 ? s_chebyshev_coeffs : (dim_2 % 2 ? buffer_1 : buff… in GradTransposeAtPoints()
364 …if (dim_1 == dim_2) ChebyshevDerivativeAtPoint<BASIS_Q_1D>(coords[elem * u_stride + dim_2 * u_comp… in GradTransposeAtPoints()
365 …else ChebyshevPolynomialsAtPoint<BASIS_Q_1D>(coords[elem * u_stride + dim_2 * u_comp_stride + p], … in GradTransposeAtPoints()
[all …]
H A Dhip-ref-basis-tensor.h112 for (CeedInt dim_2 = 0; dim_2 < BASIS_DIM; dim_2++) { in Grad() local
116 const CeedScalar *op = dim_1 == dim_2 ? s_grad_1d : s_interp_1d; in Grad()
117 const CeedScalar *in = dim_2 == 0 ? cur_u : (dim_2 % 2 ? s_buffer_2 : s_buffer_1); in Grad()
118 …CeedScalar *out = dim_2 == BASIS_DIM - 1 ? cur_v : (dim_2 % 2 ? s_buffer_1 : s_buffer_2… in Grad()
129 if (is_transpose && dim_2 == BASIS_DIM - 1) out[k] += v_k; in Grad()
/libCEED/backends/sycl-ref/
H A Dceed-sycl-ref-basis.sycl.cpp208 for (CeedInt dim_2 = 0; dim_2 < dim; dim_2++) { in CeedBasisApplyGrad_Sycl() local
215 const CeedScalar *op = dim_1 == dim_2 ? s_grad_1d : s_interp_1d; in CeedBasisApplyGrad_Sycl()
216 const CeedScalar *in = (dim_2 == 0 ? cur_u : (dim_2 % 2 ? s_buffer_2 : s_buffer_1)); in CeedBasisApplyGrad_Sycl()
217 … CeedScalar *out = dim_2 == dim - 1 ? cur_v : (dim_2 % 2 ? s_buffer_1 : s_buffer_2); in CeedBasisApplyGrad_Sycl()
229 if (is_transpose && dim_2 == dim - 1) out[k] += v_k; in CeedBasisApplyGrad_Sycl()