From 6a48ab0cda125e400fd4386ad66cb1f3a91dd6c2 Mon Sep 17 00:00:00 2001
From: Stephanie Ribet <stephanie.ribet@gmail.com>
Date: Fri, 31 May 2024 11:14:21 -0400
Subject: [PATCH] small bug in dp_mask for gpu

---
 py4DSTEM/process/phase/phase_base_class.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/py4DSTEM/process/phase/phase_base_class.py b/py4DSTEM/process/phase/phase_base_class.py
index 742013f1e..27476cb43 100644
--- a/py4DSTEM/process/phase/phase_base_class.py
+++ b/py4DSTEM/process/phase/phase_base_class.py
@@ -686,6 +686,7 @@ def _calculate_intensities_center_of_mass(
 
                 # calculate CoM
                 if dp_mask is not None:
+                    dp_mask = copy_to_device(dp_mask, device)
                     intensities_mask = intensities * dp_mask
                 else:
                     intensities_mask = intensities