Lines Matching refs:comp

32   for (CeedInt comp = 0; comp < NUM_COMP; comp++) {  in ReadPoint()  local
33 r_u[comp] = d_u[ind + comp * COMP_STRIDE]; in ReadPoint()
46 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WritePoint() local
47 d_u[ind + comp * COMP_STRIDE] += r_u[comp]; in WritePoint()
79 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + COMP_STRIDE * comp]; in ReadLVecStandard1d() local
93 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadLVecStrided1d() local
107 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[comp in WriteLVecStandard1d() local
138 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard1d_Assembly() local
139 …vec_size * e_vec_size + (in_comp * NUM_COMP + comp) * P_1D * P_1D + out_node * P_1D + in_node] += … in WriteLVecStandard1d_Assembly()
153 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard1d_QFAssembly() local
154 … d_v[ind + (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * num_elem)] = r_v[comp]; in WriteLVecStandard1d_QFAssembly()
169 for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[comp]; in WriteLVecStrided1d() local
201 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + COMP_STRIDE * comp]; in ReadLVecStandard2d() local
215 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadLVecStrided2d() local
229 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[comp in WriteLVecStandard2d() local
265 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard2d_Assembly() local
266 …const CeedInt index = (in_comp * NUM_COMP + comp) * elem_size * elem_size + out_node * elem_size +… in WriteLVecStandard2d_Assembly()
268 d_v[elem * e_vec_size * e_vec_size + index] += r_v[comp]; in WriteLVecStandard2d_Assembly()
282 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard2d_QFAssembly() local
283 …[ind + (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * Q_1D * num_elem)] = r_v[comp in WriteLVecStandard2d_QFAssembly()
298 for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[comp]; in WriteLVecStrided2d() local
332 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[z + comp * P_1D] = d_u[ind + COMP_STRIDE * com… in ReadLVecStandard3d() local
348 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[z + comp * P_1D] = d_u[ind + comp * STRIDES_CO… in ReadLVecStrided3d() local
364 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + COMP_STRIDE * comp]; in ReadEVecSliceStandard3d() local
378 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadEVecSliceStrided3d() local
393 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[z + in WriteLVecStandard3d() local
433 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard3d_Assembly() local
434 …const CeedInt index = (in_comp * NUM_COMP + comp) * elem_size * elem_size + out_node * elem_size +… in WriteLVecStandard3d_Assembly()
436 d_v[elem * e_vec_size * e_vec_size + index] += r_v[z + comp * P_1D]; in WriteLVecStandard3d_Assembly()
452 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard3d_QFAssembly() local
453 …+ (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * Q_1D * Q_1D * num_elem)] = r_v[z … in WriteLVecStandard3d_QFAssembly()
470 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[z + comp * P… in WriteLVecStrided3d() local
482 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradColloSlice3d() local
484 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[q + comp * Q_1D]; in GradColloSlice3d()
487 r_V[comp + 0 * NUM_COMP] = 0.0; in GradColloSlice3d()
489 … r_V[comp + 0 * NUM_COMP] += c_G[i + data.t_id_x * Q_1D] * data.slice[i + data.t_id_y * T_1D]; in GradColloSlice3d()
492 r_V[comp + 1 * NUM_COMP] = 0.0; in GradColloSlice3d()
494 … r_V[comp + 1 * NUM_COMP] += c_G[i + data.t_id_y * Q_1D] * data.slice[data.t_id_x + i * T_1D]; in GradColloSlice3d()
497 r_V[comp + 2 * NUM_COMP] = 0.0; in GradColloSlice3d()
499 r_V[comp + 2 * NUM_COMP] += c_G[i + q * Q_1D] * r_U[i + comp * Q_1D]; in GradColloSlice3d()
512 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradColloSliceTranspose3d() local
514 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[comp + 0 * NUM_COMP]; in GradColloSliceTranspose3d()
518 r_V[q + comp * Q_1D] += c_G[data.t_id_x + i * Q_1D] * data.slice[i + data.t_id_y * T_1D]; in GradColloSliceTranspose3d()
522 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[comp + 1 * NUM_COMP]; in GradColloSliceTranspose3d()
525 r_V[q + comp * Q_1D] += c_G[data.t_id_y + i * Q_1D] * data.slice[data.t_id_x + i * T_1D]; in GradColloSliceTranspose3d()
529 r_V[i + comp * Q_1D] += c_G[i + q * Q_1D] * r_U[comp + 2 * NUM_COMP]; in GradColloSliceTranspose3d()