make it GPU-compatible

ilhamv · Aug 18, 2024 · 5bac9d0 · 5bac9d0
1 parent 0c91c9d
commit 5bac9d0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/mcdc/kernel.py b/mcdc/kernel.py
@@ -1618,7 +1618,9 @@ def get_particle_material(P, mcdc):
 
             # Apply translation
             if cell["fill_translated"]:
-                P["translation"] += cell["translation"]
+                P["translation"][0] += cell["translation"][0]
+                P["translation"][1] += cell["translation"][1]
+                P["translation"][2] += cell["translation"][2]
                 P["translated"] = True
 
             if cell["fill_type"] == FILL_LATTICE:
@@ -2645,7 +2647,9 @@ def distance_to_boundary(P, mcdc):
 
             # Apply translation
             if cell["fill_translated"]:
-                P["translation"] += cell["translation"]
+                P["translation"][0] += cell["translation"][0]
+                P["translation"][1] += cell["translation"][1]
+                P["translation"][2] += cell["translation"][2]
                 P["translated"] = True
 
             if cell["fill_type"] == FILL_LATTICE: