| /libCEED/include/ceed/jit-source/cuda/ |
| H A D | cuda-ref-basis-tensor-at-points.h | 72 CeedInt post = 1; in InterpAtPoints() local
81 const CeedInt writeLen = pre * post * Q; in InterpAtPoints()
85 const CeedInt c = k % post; in InterpAtPoints()
86 const CeedInt j = (k / post) % Q; in InterpAtPoints()
87 const CeedInt a = k / (post * Q); in InterpAtPoints()
90 …t b = 0; b < P; b++) v_k += s_chebyshev_interp_1d[j * BASIS_P_1D + b] * in[(a * P + b) * post + c]; in InterpAtPoints()
93 post *= Q; in InterpAtPoints()
100 post = 1; in InterpAtPoints()
112 for (CeedInt c = 0; c < post; c++) { in InterpAtPoints()
115 for (CeedInt b = 0; b < Q; b++) v_k += chebyshev_x[b] * in[(a * Q + b) * post + c]; in InterpAtPoints()
[all …]
|
| H A D | cuda-ref-basis-tensor.h | 47 CeedInt post = 1; in Interp() local
55 const CeedInt writeLen = pre * post * Q; in Interp()
59 const CeedInt c = k % post; in Interp()
60 const CeedInt j = (k / post) % Q; in Interp()
61 const CeedInt a = k / (post * Q); in Interp()
64 …nt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c]; in Interp()
68 post *= Q; in Interp()
108 CeedInt post = 1; in Grad() local
119 const CeedInt writeLen = pre * post * Q; in Grad()
123 const CeedInt c = k % post; in Grad()
[all …]
|
| /libCEED/include/ceed/jit-source/hip/ |
| H A D | hip-ref-basis-tensor-at-points.h | 72 CeedInt post = 1; in InterpAtPoints() local
81 const CeedInt writeLen = pre * post * Q; in InterpAtPoints()
85 const CeedInt c = k % post; in InterpAtPoints()
86 const CeedInt j = (k / post) % Q; in InterpAtPoints()
87 const CeedInt a = k / (post * Q); in InterpAtPoints()
90 …t b = 0; b < P; b++) v_k += s_chebyshev_interp_1d[j * BASIS_P_1D + b] * in[(a * P + b) * post + c]; in InterpAtPoints()
93 post *= Q; in InterpAtPoints()
100 post = 1; in InterpAtPoints()
112 for (CeedInt c = 0; c < post; c++) { in InterpAtPoints()
115 for (CeedInt b = 0; b < Q; b++) v_k += chebyshev_x[b] * in[(a * Q + b) * post + c]; in InterpAtPoints()
[all …]
|
| H A D | hip-ref-basis-tensor.h | 47 CeedInt post = 1; in Interp() local
55 const CeedInt writeLen = pre * post * Q; in Interp()
59 const CeedInt c = k % post; in Interp()
60 const CeedInt j = (k / post) % Q; in Interp()
61 const CeedInt a = k / (post * Q); in Interp()
64 …nt b = 0; b < P; b++) v_k += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c]; in Interp()
68 post *= Q; in Interp()
108 CeedInt post = 1; in Grad() local
119 const CeedInt writeLen = pre * post * Q; in Grad()
123 const CeedInt c = k % post; in Grad()
[all …]
|
| /libCEED/backends/ref/ |
| H A D | ceed-ref-basis.c | 67 CeedInt pre = num_comp * CeedIntPow(P, dim - 1), post = num_elem; in CeedBasisApplyCore_Ref() local
73 …CeedCallBackend(CeedTensorContractApply(contract, pre, P, post, Q, interp_1d, t_mode, add && (d ==… in CeedBasisApplyCore_Ref()
76 post *= Q; in CeedBasisApplyCore_Ref()
92 CeedInt pre = num_comp * CeedIntPow(P, dim - 1), post = num_elem; in CeedBasisApplyCore_Ref() local
103 …CeedCallBackend(CeedTensorContractApply(contract, pre, P, post, Q, (t_mode == CEED_NOTRANSPOSE ? i… in CeedBasisApplyCore_Ref()
108 post *= Q; in CeedBasisApplyCore_Ref()
117 pre = num_comp * CeedIntPow(P, dim - 1), post = num_elem; in CeedBasisApplyCore_Ref()
119 …CeedCallBackend(CeedTensorContractApply(contract, pre, P, post, Q, (t_mode == CEED_NOTRANSPOSE ? i… in CeedBasisApplyCore_Ref()
125 post *= Q; in CeedBasisApplyCore_Ref()
133 CeedInt pre = num_comp * CeedIntPow(P, dim - 1), post = num_elem; in CeedBasisApplyCore_Ref() local
[all …]
|
| /libCEED/backends/sycl-ref/ |
| H A D | ceed-sycl-ref-basis.sycl.cpp | 105 CeedInt post = 1; in CeedBasisApplyInterp_Sycl() local
118 const CeedInt writeLen = pre * post * Q; in CeedBasisApplyInterp_Sycl()
120 const CeedInt c = k % post; in CeedBasisApplyInterp_Sycl()
121 const CeedInt j = (k / post) % Q; in CeedBasisApplyInterp_Sycl()
122 const CeedInt a = k / (post * Q); in CeedBasisApplyInterp_Sycl()
126 vk += s_interp_1d[j * stride_0 + b * stride_1] * in[(a * P + b) * post + c]; in CeedBasisApplyInterp_Sycl()
130 post *= Q; in CeedBasisApplyInterp_Sycl()
204 CeedInt post = 1; in CeedBasisApplyGrad_Sycl() local
220 const CeedInt writeLen = pre * post * Q; in CeedBasisApplyGrad_Sycl()
222 const CeedInt c = k % post; in CeedBasisApplyGrad_Sycl()
[all …]
|
| /libCEED/interface/ |
| H A D | ceed-basis.c | 561 CeedInt pre = num_comp * CeedIntPow(Q_1d, dim - 1), post = 1; in CeedBasisApplyAtPoints_Core() local
566 … CeedCall(CeedTensorContractApply(basis->contract, pre, Q_1d, post, 1, chebyshev_x, t_mode, false, in CeedBasisApplyAtPoints_Core()
569 post *= 1; in CeedBasisApplyAtPoints_Core()
582 CeedInt pre = num_comp * CeedIntPow(Q_1d, dim - 1), post = 1; in CeedBasisApplyAtPoints_Core() local
588 … CeedCall(CeedTensorContractApply(basis->contract, pre, Q_1d, post, 1, chebyshev_x, t_mode, false, in CeedBasisApplyAtPoints_Core()
591 post *= 1; in CeedBasisApplyAtPoints_Core()
624 CeedInt pre = num_comp * 1, post = 1; in CeedBasisApplyAtPoints_Core() local
630 …CeedCall(CeedTensorContractApply(basis->contract, pre, 1, post, Q_1d, chebyshev_x, t_mode, p > 0 &… in CeedBasisApplyAtPoints_Core()
633 post *= Q_1d; in CeedBasisApplyAtPoints_Core()
645 CeedInt pre = num_comp * 1, post = 1; in CeedBasisApplyAtPoints_Core() local
[all …]
|