Lines Matching refs:PetscDefaultCudaStream

127 …rootbufdisp, nRemoteRootRanks * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
128 …rootsigdisp, nRemoteRootRanks * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
129 …>ndranks, nRemoteRootRanks * sizeof(PetscMPIInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
130 …ranks, (nRemoteRootRanks + 1) * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
150 …leafbufdisp, nRemoteLeafRanks * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
151 …leafsigdisp, nRemoteLeafRanks * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
152 …ndiranks, nRemoteLeafRanks * sizeof(PetscMPIInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
153 …ranks, (nRemoteLeafRanks + 1) * sizeof(PetscInt), cudaMemcpyHostToDevice, PetscDefaultCudaStream));  in PetscSFSetUp_Basic_NVSHMEM()
799   nvshmemx_float_sum_reduce_on_stream(NVSHMEM_TEAM_WORLD, dst, src, num, PetscDefaultCudaStream);  in PetscNvshmemSum()
809   nvshmemx_float_max_reduce_on_stream(NVSHMEM_TEAM_WORLD, dst, src, num, PetscDefaultCudaStream);  in PetscNvshmemMax()
819   nvshmemx_double_sum_reduce_on_stream(NVSHMEM_TEAM_WORLD, dst, src, num, PetscDefaultCudaStream);  in PetscNvshmemSum()
829   nvshmemx_double_max_reduce_on_stream(NVSHMEM_TEAM_WORLD, dst, src, num, PetscDefaultCudaStream);  in PetscNvshmemMax()