Home
last modified time | relevance | path

Searched refs:PetscDefaultCudaStream (Results 1 – 8 of 8) sorted by relevance

/petsc/src/ksp/ksp/impls/hpddm/cuda/
H A Dhpddm.cu28 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), db, N, dptr[0]); in KSPSolve_HPDDM_CUDA_Private()
29 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dx, N, dptr[1]); in KSPSolve_HPDDM_CUDA_Private()
33 thrust::copy_n(thrust::cuda::par.on(PetscDefaultCudaStream), dptr[1], N, dx); in KSPSolve_HPDDM_CUDA_Private()
/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu127 …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()
[all …]
/petsc/src/sys/objects/kokkos/
H A Dkinit.kokkos.cxx85 …cCallCXX(PetscKokkosExecutionSpacePtr = new Kokkos::DefaultExecutionSpace(PetscDefaultCudaStream)); in PetscKokkosInitializeCheck()
/petsc/src/sys/objects/device/interface/
H A Dglobal_dcontext.cxx34 PetscDefaultCudaStream = *static_cast<cudaStream_t *>(handle); in PetscSetDefaultCUPMStreamFromDeviceContext()
/petsc/include/
H A Dpetscdevice_cuda.h163 PETSC_EXTERN cudaStream_t PetscDefaultCudaStream; // The default stream used by PETSc variable
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu734 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_Cholesky()
747 …PetscCallThrust(thrust::transform(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::device_poi… in MatSolve_SeqAIJCUSPARSE_Cholesky()
749 …PetscCallThrust(thrust::transform(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::device_poi… in MatSolve_SeqAIJCUSPARSE_Cholesky()
762 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_Cholesky()
1431 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_LU()
1452 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolve_SeqAIJCUSPARSE_LU()
1502 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolveTranspose_SeqAIJCUSPARSE_LU()
1522 …PetscCallThrust(thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutatio… in MatSolveTranspose_SeqAIJCUSPARSE_LU()
1562 …thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutation_iterator(bGPU,… in MatSolveTranspose_SeqAIJCUSPARSE()
1573 …thrust::copy(thrust::cuda::par.on(PetscDefaultCudaStream), thrust::make_permutation_iterator(xGPU,… in MatSolveTranspose_SeqAIJCUSPARSE()
[all …]
/petsc/src/vec/is/sf/interface/
H A Dsf.c384 …PetscCallMPI(MPIX_Info_set_hex(info, "value", &PetscDefaultCudaStream, sizeof(PetscDefaultCudaStre… in PetscSFSetFromOptions()
/petsc/src/sys/objects/
H A Dpinit.c754 cudaStream_t PetscDefaultCudaStream = NULL; variable