Lines Matching refs:CUDA
6 static constexpr impl::MatDense_MPI_CUPM<DeviceType::CUDA> mat_cupm{};
74 PetscCall(MatCreateDenseCUPM<DeviceType::CUDA>(comm, m, n, M, N, data, A)); in MatCreateDenseCUDA()
103 PetscCall(MatDenseCUPMPlaceArray<DeviceType::CUDA>(mat, array)); in MatDenseCUDAPlaceArray()
126 PetscCall(MatDenseCUPMResetArray<DeviceType::CUDA>(mat)); in MatDenseCUDAResetArray()
154 PetscCall(MatDenseCUPMReplaceArray<DeviceType::CUDA>(mat, array)); in MatDenseCUDAReplaceArray()
185 PetscCall(MatDenseCUPMGetArrayWrite<DeviceType::CUDA>(A, a)); in MatDenseCUDAGetArrayWrite()
207 PetscCall(MatDenseCUPMRestoreArrayWrite<DeviceType::CUDA>(A, a)); in MatDenseCUDARestoreArrayWrite()
237 PetscCall(MatDenseCUPMGetArrayRead<DeviceType::CUDA>(A, a)); in MatDenseCUDAGetArrayRead()
263 PetscCall(MatDenseCUPMRestoreArrayRead<DeviceType::CUDA>(A, a)); in MatDenseCUDARestoreArrayRead()
293 PetscCall(MatDenseCUPMGetArray<DeviceType::CUDA>(A, a)); in MatDenseCUDAGetArray()
315 PetscCall(MatDenseCUPMRestoreArray<DeviceType::CUDA>(A, a)); in MatDenseCUDARestoreArray()
337 PetscCall(MatDenseCUPMSetPreallocation<DeviceType::CUDA>(A, device_array)); in MatDenseCUDASetPreallocation()