Home
last modified time | relevance | path

Searched refs:r_B (Results 1 – 3 of 3) sorted by relevance

/libCEED/include/ceed/jit-source/sycl/ !
H A Dsycl-shared-basis-tensor-templates.h285 CeedScalar r_B[T_1D]; in ContractX3d() local
287 r_B[i] = B[i + item_id_x * P_1D]; in ContractX3d()
297 V[k] += r_B[i] * scratch[i + item_id_y * T_1D]; // Contract x direction in ContractX3d()
312 CeedScalar r_B[T_1D]; in ContractY3d() local
314 r_B[i] = B[i + item_id_y * P_1D]; in ContractY3d()
324 V[k] += r_B[i] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractY3d()
375 CeedScalar r_B[T_1D]; in ContractTransposeY3d() local
377 r_B[i] = B[item_id_y + i * P_1D]; in ContractTransposeY3d()
387 V[k] += r_B[i] * scratch[item_id_x + i * T_1D]; // Contract y direction in ContractTransposeY3d()
402 CeedScalar r_B[T_1D]; in ContractTransposeAddY3d() local
[all …]
/libCEED/include/ceed/jit-source/cuda/ !
H A Dcuda-shared-basis-tensor-templates.h323 CeedScalar r_B[P_1D]; in ContractX3d() local
325 r_B[i] = B[i + data.t_id_x * P_1D]; in ContractX3d()
335 V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractX3d()
346 CeedScalar r_B[P_1D]; in ContractY3d() local
348 r_B[i] = B[i + data.t_id_y * P_1D]; in ContractY3d()
358 V[k] += r_B[i] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractY3d()
399 CeedScalar r_B[Q_1D]; in ContractTransposeY3d() local
401 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeY3d()
411 V[k] += r_B[i] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractTransposeY3d()
422 CeedScalar r_B[Q_1D]; in ContractTransposeAddY3d() local
[all …]
/libCEED/include/ceed/jit-source/hip/ !
H A Dhip-shared-basis-tensor-templates.h322 CeedScalar r_B[P_1D]; in ContractX3d() local
324 r_B[i] = B[i + data.t_id_x * P_1D]; in ContractX3d()
334 V[k] += r_B[i] * data.slice[i + data.t_id_y * T_1D]; // Contract x direction in ContractX3d()
345 CeedScalar r_B[P_1D]; in ContractY3d() local
347 r_B[i] = B[i + data.t_id_y * P_1D]; in ContractY3d()
357 V[k] += r_B[i] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractY3d()
398 CeedScalar r_B[Q_1D]; in ContractTransposeY3d() local
400 r_B[i] = B[data.t_id_y + i * P_1D]; in ContractTransposeY3d()
410 V[k] += r_B[i] * data.slice[data.t_id_x + i * T_1D]; // Contract y direction in ContractTransposeY3d()
421 CeedScalar r_B[Q_1D]; in ContractTransposeAddY3d() local
[all …]