35 #include <hip/hip_runtime.h>
48 const dim3 nthrds(1024, 1, 1);
49 const dim3 nblcks(((*m)+1024 - 1)/ 1024, 1, 1);
51 hipLaunchKernelGGL(HIP_KERNEL_NAME(dirichlet_apply_scalar_kernel<real>),
53 (
int *) msk, (
real *)
x, *g, *m);
61 void *z,
real *g,
int *m) {
63 const dim3 nthrds(1024, 1, 1);
64 const dim3 nblcks(((*m)+1024 - 1)/ 1024, 1, 1);
66 hipLaunchKernelGGL(HIP_KERNEL_NAME(dirichlet_apply_vector_kernel<real>),
68 (
int *) msk, (
real *)
x,
__global__ void const T *__restrict__ x
void hip_dirichlet_apply_vector(void *msk, void *x, void *y, void *z, real *g, int *m)
void hip_dirichlet_apply_scalar(void *msk, void *x, real *g, int *m)