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
92 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadLVecStrided1d() local
106 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[comp in WriteLVecStandard1d() local
137 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard1d_Assembly() local
138 …vec_size * e_vec_size + (in_comp * NUM_COMP + comp) * P_1D * P_1D + out_node * P_1D + in_node] += … in WriteLVecStandard1d_Assembly()
152 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard1d_QFAssembly() local
153 … d_v[ind + (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * num_elem)] = r_v[comp]; in WriteLVecStandard1d_QFAssembly()
168 for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[comp]; in WriteLVecStrided1d() local
200 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + COMP_STRIDE * comp]; in ReadLVecStandard2d() local
213 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadLVecStrided2d() local
227 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[comp in WriteLVecStandard2d() local
263 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard2d_Assembly() local
264 …const CeedInt index = (in_comp * NUM_COMP + comp) * elem_size * elem_size + out_node * elem_size +… in WriteLVecStandard2d_Assembly()
266 d_v[elem * e_vec_size * e_vec_size + index] += r_v[comp]; in WriteLVecStandard2d_Assembly()
280 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard2d_QFAssembly() local
281 …[ind + (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * Q_1D * num_elem)] = r_v[comp in WriteLVecStandard2d_QFAssembly()
296 for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[comp]; in WriteLVecStrided2d() local
330 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[z + comp * P_1D] = d_u[ind + COMP_STRIDE * com… in ReadLVecStandard3d() local
345 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[z + comp * P_1D] = d_u[ind + comp * STRIDES_CO… in ReadLVecStrided3d() local
361 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + COMP_STRIDE * comp]; in ReadEVecSliceStandard3d() local
375 for (CeedInt comp = 0; comp < NUM_COMP; comp++) r_u[comp] = d_u[ind + comp * STRIDES_COMP]; in ReadEVecSliceStrided3d() local
390 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) atomicAdd(&d_v[ind + COMP_STRIDE * comp], r_v[z + in WriteLVecStandard3d() local
430 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in WriteLVecStandard3d_Assembly() local
431 …const CeedInt index = (in_comp * NUM_COMP + comp) * elem_size * elem_size + out_node * elem_size +… in WriteLVecStandard3d_Assembly()
433 d_v[elem * e_vec_size * e_vec_size + index] += r_v[z + comp * P_1D]; in WriteLVecStandard3d_Assembly()
449 for (CeedInt comp = 0; comp < NUM_COMP_FIELD; comp++) { in WriteLVecStandard3d_QFAssembly() local
450 …+ (input_offset * NUM_COMP_OUT + output_offset + comp) * (Q_1D * Q_1D * Q_1D * num_elem)] = r_v[z … in WriteLVecStandard3d_QFAssembly()
467 …for (CeedInt comp = 0; comp < NUM_COMP; comp++) d_v[ind + comp * STRIDES_COMP] += r_v[z + comp * P… in WriteLVecStrided3d() local
479 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradColloSlice3d() local
481 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[q + comp * Q_1D]; in GradColloSlice3d()
484 r_V[comp + 0 * NUM_COMP] = 0.0; in GradColloSlice3d()
486 … 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()
489 r_V[comp + 1 * NUM_COMP] = 0.0; in GradColloSlice3d()
491 … 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()
494 r_V[comp + 2 * NUM_COMP] = 0.0; in GradColloSlice3d()
496 r_V[comp + 2 * NUM_COMP] += c_G[i + q * Q_1D] * r_U[i + comp * Q_1D]; in GradColloSlice3d()
509 for (CeedInt comp = 0; comp < NUM_COMP; comp++) { in GradColloSliceTranspose3d() local
512 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[comp + 0 * NUM_COMP]; in GradColloSliceTranspose3d()
515 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()
519 data.slice[data.t_id_x + data.t_id_y * T_1D] = r_U[comp + 1 * NUM_COMP]; in GradColloSliceTranspose3d()
522 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()
526 r_V[i + comp * Q_1D] += c_G[i + q * Q_1D] * r_U[comp + 2 * NUM_COMP]; in GradColloSliceTranspose3d()