Home
last modified time | relevance | path

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

/libCEED/include/ceed/jit-source/cuda/
H A Dcuda-ref-basis-tensor-at-points.h88 CeedScalar v_k = 0; in InterpAtPoints() local
90 …for (CeedInt b = 0; b < P; b++) v_k += s_chebyshev_interp_1d[j * BASIS_P_1D + b] * in[(a * P + b) … in InterpAtPoints()
91 out[k] = v_k; in InterpAtPoints()
113 CeedScalar v_k = 0; in InterpAtPoints() local
115 for (CeedInt b = 0; b < Q; b++) v_k += chebyshev_x[b] * in[(a * Q + b) * post + c]; in InterpAtPoints()
116 out[a * post + c] = v_k; in InterpAtPoints()
207 CeedScalar v_k = 0; in InterpTransposeAtPoints() local
209 …for (CeedInt b = 0; b < Q; b++) v_k += s_chebyshev_interp_1d[j + b * BASIS_P_1D] * in[(a * Q + b) … in InterpTransposeAtPoints()
210 if (d == BASIS_DIM - 1) out[k] += v_k; in InterpTransposeAtPoints()
211 else out[k] = v_k; in InterpTransposeAtPoints()
[all …]
H A Dcuda-ref-basis-tensor.h62 CeedScalar v_k = 0; in Interp() local
64 …for (CeedInt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) *… in Interp()
65 if (is_transpose && d == BASIS_DIM - 1) out[k] += v_k; in Interp()
66 else out[k] = v_k; in Interp()
126 CeedScalar v_k = 0; in Grad() local
128 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in Grad()
129 if (is_transpose && dim_2 == BASIS_DIM - 1) out[k] += v_k; in Grad()
130 else out[k] = v_k; in Grad()
/libCEED/include/ceed/jit-source/hip/
H A Dhip-ref-basis-tensor-at-points.h88 CeedScalar v_k = 0; in InterpAtPoints() local
90 …for (CeedInt b = 0; b < P; b++) v_k += s_chebyshev_interp_1d[j * BASIS_P_1D + b] * in[(a * P + b) … in InterpAtPoints()
91 out[k] = v_k; in InterpAtPoints()
113 CeedScalar v_k = 0; in InterpAtPoints() local
115 for (CeedInt b = 0; b < Q; b++) v_k += chebyshev_x[b] * in[(a * Q + b) * post + c]; in InterpAtPoints()
116 out[a * post + c] = v_k; in InterpAtPoints()
207 CeedScalar v_k = 0; in InterpTransposeAtPoints() local
209 …for (CeedInt b = 0; b < Q; b++) v_k += s_chebyshev_interp_1d[j + b * BASIS_P_1D] * in[(a * Q + b) … in InterpTransposeAtPoints()
210 if (d == BASIS_DIM - 1) out[k] += v_k; in InterpTransposeAtPoints()
211 else out[k] = v_k; in InterpTransposeAtPoints()
[all …]
H A Dhip-ref-basis-tensor.h62 CeedScalar v_k = 0; in Interp() local
64 …for (CeedInt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) *… in Interp()
65 if (is_transpose && d == BASIS_DIM - 1) out[k] += v_k; in Interp()
66 else out[k] = v_k; in Interp()
126 CeedScalar v_k = 0; in Grad() local
128 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in Grad()
129 if (is_transpose && dim_2 == BASIS_DIM - 1) out[k] += v_k; in Grad()
130 else out[k] = v_k; in Grad()
/libCEED/backends/sycl-ref/
H A Dceed-sycl-ref-basis.sycl.cpp226 CeedScalar v_k = 0; in CeedBasisApplyGrad_Sycl() local
227 …for (CeedInt b = 0; b < P; b++) v_k += op[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c… in CeedBasisApplyGrad_Sycl()
229 if (is_transpose && dim_2 == dim - 1) out[k] += v_k; in CeedBasisApplyGrad_Sycl()
230 else out[k] = v_k; in CeedBasisApplyGrad_Sycl()