Lines Matching refs:Q_1D
214 …ffset3d(const CeedInt num_comp, const CeedInt strides_comp, const CeedInt Q_1D, const CeedInt num_… in readSliceQuadsOffset3d() argument
220 if (item_id_x < Q_1D && item_id_y < Q_1D && elem < num_elem) { in readSliceQuadsOffset3d()
221 const CeedInt node = item_id_x + Q_1D * (item_id_y + Q_1D * q); in readSliceQuadsOffset3d()
222 const CeedInt ind = indices[node + elem * Q_1D * Q_1D * Q_1D]; in readSliceQuadsOffset3d()
230 inline void readSliceQuadsStrided3d(const CeedInt num_comp, const CeedInt Q_1D, CeedInt strides_nod… in readSliceQuadsStrided3d() argument
237 if (item_id_x < Q_1D && item_id_y < Q_1D && elem < num_elem) { in readSliceQuadsStrided3d()
238 const CeedInt node = item_id_x + Q_1D * (item_id_y + Q_1D * q); in readSliceQuadsStrided3d()
285 inline void gradCollo3d(const CeedInt num_comp, const CeedInt Q_1D, const CeedInt q, const private … in gradCollo3d() argument
291 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradCollo3d()
292 scratch[item_id_x + item_id_y * T_1D] = r_U[q + comp * Q_1D]; in gradCollo3d()
296 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradCollo3d()
299 for (CeedInt i = 0; i < Q_1D; ++i) in gradCollo3d()
300 …r_V[comp + 0 * num_comp] += s_G[i + item_id_x * Q_1D] * scratch[i + item_id_y * T_1D]; // Contrac… in gradCollo3d()
304 for (CeedInt i = 0; i < Q_1D; ++i) in gradCollo3d()
305 …r_V[comp + 1 * num_comp] += s_G[i + item_id_y * Q_1D] * scratch[item_id_x + i * T_1D]; // Contrac… in gradCollo3d()
309 …for (CeedInt i = 0; i < Q_1D; ++i) r_V[comp + 2 * num_comp] += s_G[i + q * Q_1D] * r_U[i + comp * … in gradCollo3d()
319 inline void gradColloTranspose3d(const CeedInt num_comp, const CeedInt Q_1D, const CeedInt q, const… in gradColloTranspose3d() argument
326 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradColloTranspose3d()
331 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradColloTranspose3d()
332 for (CeedInt i = 0; i < Q_1D; ++i) in gradColloTranspose3d()
333 …r_V[q + comp * Q_1D] += s_G[item_id_x + i * Q_1D] * scratch[i + item_id_y * T_1D]; // Contract x … in gradColloTranspose3d()
338 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradColloTranspose3d()
343 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradColloTranspose3d()
344 for (CeedInt i = 0; i < Q_1D; ++i) in gradColloTranspose3d()
345 …r_V[q + comp * Q_1D] += s_G[item_id_y + i * Q_1D] * scratch[item_id_x + i * T_1D]; // Contract y … in gradColloTranspose3d()
350 if (item_id_x < Q_1D && item_id_y < Q_1D) { in gradColloTranspose3d()
351 for (CeedInt i = 0; i < Q_1D; ++i) in gradColloTranspose3d()
352 …r_V[i + comp * Q_1D] += s_G[i + q * Q_1D] * r_U[comp + 2 * num_comp]; // PARTIAL contract z direc… in gradColloTranspose3d()