48 if (
nbytes ==
sizeof(
float)) {
58 else if (
nbytes ==
sizeof(
double)) {
76 if (
nbytes ==
sizeof(
float)) {
88 else if (
nbytes ==
sizeof(
double)) {
__global__ void dirichlet_apply_scalar_kernel(const int *__restrict__ msk, T *__restrict__ x, const T g, const int m)
void device_mpi_allreduce(void *buf_d, void *buf, int count, int nbytes, int op)
void device_mpi_allreduce_inplace(void *buf_d, int count, int nbytes, int op)