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.