Template Function Pennylane::LightningGPU::MPI::make_shared_mpi_worker

• Defined in File MPIWorker.hpp

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

 

Download Python script

 

Download Notebook

 

View on GitHub