Template Function Pennylane::LightningGPU::MPI::make_shared_mpi_worker
Function Documentation
-
template<typename CFP_t>
SharedMPIWorker Pennylane::LightningGPU::MPI::make_shared_mpi_worker(custatevecHandle_t handle, MPIManager &mpi_manager, const std::size_t mpi_buf_size, CFP_t *sv, const std::size_t numLocalQubits, cudaStream_t localStream)
Creates a SharedMPIWorker (a shared pointer to a custatevecSVSwapWorker)
- Parameters
handle – custatevecHandle.
mpi_manager – MPI manager object.
mpi_buf_size – Size to set MPI buffer in MiB (mebibytes).
sv – Pointer to the data requires MPI operation.
numLocalQubits – Number of local qubits.
localStream – Local cuda stream.
api/function_MPIWorker_8hpp_1ab2fdb49fd53b250360d57fdf03cbddf2