43 use,
intrinsic :: iso_c_binding, only : c_ptr, c_int
55 dx_d, dy_d, dz_d, dxt_d, dyt_d, dzt_d,&
56 h1_d, drdx_d, drdy_d, drdz_d, &
57 dsdx_d, dsdy_d, dsdz_d, &
58 dtdx_d, dtdy_d, dtdz_d, jacinv_d, weight3_d, nelv, lx) &
59 bind(c, name=
'hip_ax_helm_stress_vector')
60 use,
intrinsic :: iso_c_binding
61 type(c_ptr),
value :: au_d, av_d, aw_d
62 type(c_ptr),
value :: u_d, v_d, w_d
63 type(c_ptr),
value :: dx_d, dy_d, dz_d
64 type(c_ptr),
value :: dxt_d, dyt_d, dzt_d
65 type(c_ptr),
value :: h1_d
66 type(c_ptr),
value :: drdx_d, drdy_d, drdz_d
67 type(c_ptr),
value :: dsdx_d, dsdy_d, dsdz_d
68 type(c_ptr),
value :: dtdx_d, dtdy_d, dtdz_d
69 type(c_ptr),
value :: jacinv_d, weight3_d
70 integer(c_int) :: nel, lx
76 h2_d, B_d, n) bind(c, name='hip_ax_helm_stress_vector_part2')
77 use,
intrinsic :: iso_c_binding
78 type(c_ptr),
value :: au_d, av_d, aw_d
79 type(c_ptr),
value :: u_d, v_d, w_d
80 type(c_ptr),
value :: h2_d, B_d
88 dx_d, dy_d, dz_d, dxt_d, dyt_d, dzt_d,&
89 h1_d, drdx_d, drdy_d, drdz_d, &
90 dsdx_d, dsdy_d, dsdz_d, &
91 dtdx_d, dtdy_d, dtdz_d, jacinv_d, weight3_d, nelv, lx) &
92 bind(c, name=
'cuda_ax_helm_stress_vector')
93 use,
intrinsic :: iso_c_binding
94 type(c_ptr),
value :: au_d, av_d, aw_d
95 type(c_ptr),
value :: u_d, v_d, w_d
96 type(c_ptr),
value :: dx_d, dy_d, dz_d
97 type(c_ptr),
value :: dxt_d, dyt_d, dzt_d
98 type(c_ptr),
value :: h1_d
99 type(c_ptr),
value :: drdx_d, drdy_d, drdz_d
100 type(c_ptr),
value :: dsdx_d, dsdy_d, dsdz_d
101 type(c_ptr),
value :: dtdx_d, dtdy_d, dtdz_d
102 type(c_ptr),
value :: jacinv_d, weight3_d
103 integer(c_int) :: nel, lx
109 h2_d, B_d, n) bind(c, name='cuda_ax_helm_stress_vector_part2')
110 use,
intrinsic :: iso_c_binding
111 type(c_ptr),
value :: au_d, av_d, aw_d
112 type(c_ptr),
value :: u_d, v_d, w_d
113 type(c_ptr),
value :: h2_d, B_d
122 u, v, w, coef, msh, Xh)
124 type(
space_t),
intent(inout) :: Xh
125 type(
mesh_t),
intent(inout) :: msh
126 type(
coef_t),
intent(inout) :: coef
127 real(kind=
rp),
intent(inout) :: au(xh%lx, xh%ly, xh%lz, msh%nelv)
128 real(kind=
rp),
intent(inout) :: av(xh%lx, xh%ly, xh%lz, msh%nelv)
129 real(kind=
rp),
intent(inout) :: aw(xh%lx, xh%ly, xh%lz, msh%nelv)
130 real(kind=
rp),
intent(inout) :: u(xh%lx, xh%ly, xh%lz, msh%nelv)
131 real(kind=
rp),
intent(inout) :: v(xh%lx, xh%ly, xh%lz, msh%nelv)
132 real(kind=
rp),
intent(inout) :: w(xh%lx, xh%ly, xh%lz, msh%nelv)
133 type(c_ptr) :: u_d, v_d, w_d
134 type(c_ptr) :: au_d, av_d, aw_d
146 xh%dx_d, xh%dy_d, xh%dz_d, xh%dxt_d, xh%dyt_d, xh%dzt_d, coef%h1_d, &
147 coef%drdx_d, coef%drdy_d, coef%drdz_d, &
148 coef%dsdx_d, coef%dsdy_d, coef%dsdz_d, &
149 coef%dtdx_d, coef%dtdy_d, coef%dtdz_d, &
150 coef%jacinv_d, xh%w3_d, msh%nelv, xh%lx)
153 xh%dx_d, xh%dy_d, xh%dz_d, xh%dxt_d, xh%dyt_d, xh%dzt_d, coef%h1_d, &
154 coef%drdx_d, coef%drdy_d, coef%drdz_d, &
155 coef%dsdx_d, coef%dsdy_d, coef%dsdz_d, &
156 coef%dtdx_d, coef%dtdy_d, coef%dtdz_d, &
157 coef%jacinv_d, xh%w3_d, msh%nelv, xh%lx)
159 call neko_error(
'OPENCL is not implemented for full stress formulation')
165 coef%h2_d, coef%B_d, coef%dof%size())
168 coef%h2_d, coef%B_d, coef%dof%size())
170 call device_addcol4(au_d ,coef%h2_d, coef%B_d, u_d, coef%dof%size())
171 call device_addcol4(av_d ,coef%h2_d, coef%B_d, v_d, coef%dof%size())
172 call device_addcol4(aw_d ,coef%h2_d, coef%B_d, w_d, coef%dof%size())
void cuda_ax_helm_stress_vector(void *au, void *av, void *aw, void *u, void *v, void *w, void *dx, void *dy, void *dz, void *dxt, void *dyt, void *dzt, void *h1, void *drdx, void *drdy, void *drdz, void *dsdx, void *dsdy, void *dsdz, void *dtdx, void *dtdy, void *dtdz, void *jacinv, void *w3, int *nelv, int *lx)
void cuda_ax_helm_stress_vector_part2(void *au, void *av, void *aw, void *u, void *v, void *w, void *h2, void *B, int *n)
Return the device pointer for an associated Fortran array.
subroutine ax_helm_full_device_compute_vector(this, au, av, aw, u, v, w, coef, msh, Xh)
subroutine, public device_addcol4(a_d, b_d, c_d, d_d, n)
Returns .
Device abstraction, common interface for various accelerators.
integer, parameter, public rp
Global precision used in computations.
Defines a function space.
Matrix-vector product for a Helmholtz problem.
Coefficients defined on a given (mesh, ) tuple. Arrays use indices (i,j,k,e): element e,...
The function space for the SEM solution fields.