36 #include <hip/hip_runtime.h>
55 const dim3 nthrds(1024, 1, 1);
56 const dim3 nblcks(((*n)+1024 - 1)/ 1024, 1, 1);
57 const int nb = ((*n) + 1024 - 1)/ 1024;
71 hipLaunchKernelGGL(HIP_KERNEL_NAME(gmres_part2_kernel<real>),
76 hipLaunchKernelGGL(HIP_KERNEL_NAME(reduce_kernel<real>),
80 #ifdef HAVE_DEVICE_MPI
86 hipMemcpyDeviceToHost,
__global__ void T *__restrict__ T *__restrict__ const T *__restrict__ const T *__restrict__ const T *__restrict__ w
__global__ void T *__restrict__ T *__restrict__ const T *__restrict__ const T *__restrict__ v
void device_mpi_allreduce(void *buf_d, void *buf, int count, int nbytes, int op)
real hip_gmres_part2(void *w, void *v, void *h, void *mult, int *j, int *n)