From dce2303eb05d307787fd44a81c4814f714d67481 Mon Sep 17 00:00:00 2001
From: Sam Reeve <6740307+streeve@users.noreply.github.com>
Date: Wed, 23 Oct 2024 23:12:14 -0400
Subject: [PATCH] fixup: only use local particles for output profiles

---
 src/CabanaPD_DisplacementProfile.hpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/CabanaPD_DisplacementProfile.hpp b/src/CabanaPD_DisplacementProfile.hpp
index d3b48d94..be5ca283 100644
--- a/src/CabanaPD_DisplacementProfile.hpp
+++ b/src/CabanaPD_DisplacementProfile.hpp
@@ -57,7 +57,7 @@ void createOutputProfile( MPI_Comm comm, const int num_cell,
         }
     };
     Kokkos::RangePolicy<typename memory_space::execution_space> policy(
-        0, x.size() );
+        0, particles.n_local );
     Kokkos::parallel_for( "displacement_profile", policy, measure_profile );
     auto count_host =
         Kokkos::create_mirror_view_and_copy( Kokkos::HostSpace{}, count );