Lightning

Usage

Development

API

  1. Docs
  2. Overview
  3. Template Function Pennylane::LightningGPU::MPI::scatterCSRMatrix

Template Function Pennylane::LightningGPU::MPI::scatterCSRMatrix

Function Documentation

template<class Precision, class IndexT>
auto Pennylane::LightningGPU::MPI::scatterCSRMatrix(MPIManagerGPU &mpi_manager, std::vector<CSRMatrix<Precision, IndexT>> &matrix, std::size_t local_num_rows, std::size_t root) -> CSRMatrix<Precision, IndexT>

Scatter a CSR (Compressed Sparse Row) format matrix.

Template Parameters

  • Precision – Floating-point precision type.

  • IndexT – Integer type used as indices of the sparse matrix.

Parameters

  • mpi_manager – MPIManagerGPU object.

  • matrix – CSR (Compressed Sparse Row) format matrix vector.

  • local_num_rows – Number of rows of local CSR matrix.

  • root – Root rank of the scatter operation.

Previous
Next