From cf1d66ec13823d255765a71fecc5b14d1324915e Mon Sep 17 00:00:00 2001 From: Joseph Lee Date: Wed, 13 Nov 2024 19:46:02 +0000 Subject: [PATCH] add TODO for using shmem for indices in multirz --- .../src/simulators/lightning_kokkos/gates/BasicGateFunctors.hpp | 1 + 1 file changed, 1 insertion(+) diff --git a/pennylane_lightning/core/src/simulators/lightning_kokkos/gates/BasicGateFunctors.hpp b/pennylane_lightning/core/src/simulators/lightning_kokkos/gates/BasicGateFunctors.hpp index 2c1b49d83..9594a2e81 100644 --- a/pennylane_lightning/core/src/simulators/lightning_kokkos/gates/BasicGateFunctors.hpp +++ b/pennylane_lightning/core/src/simulators/lightning_kokkos/gates/BasicGateFunctors.hpp @@ -73,6 +73,7 @@ template class applyNCNFunctor { Kokkos::parallel_for(Kokkos::TeamPolicy(two2N, Kokkos::AUTO, dim), *this); } + // TODO: Runtime selection for copying indices to scratch level 0/shmem KOKKOS_FUNCTION void operator()(const MemberType &teamMember) const { const std::size_t k = teamMember.league_rank(); const std::size_t offset = Util::parity_2_offset(parity, k);