[bug] fix incorrect normalization for GPU mmc, fix #100

fangq · Oct 16, 2024 · c646344 · c646344
1 parent 8867302
commit c646344
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/mmc_core.cl b/src/mmc_core.cl
@@ -1704,8 +1704,12 @@ __device__ void onephoton(unsigned int id, __local float* ppath, __constant MCXP
         savedebugdata(&r, id, reporter, gdebugdata, gcfg);
     }
 
-    for (oldeid = 0; oldeid < GPU_PARAM(gcfg, srcnum); oldeid++) {
-        energyesc[oldeid] += r.weight * ppath[GPU_PARAM(gcfg, reclen) + oldeid];
+    if (GPU_PARAM(gcfg, srcnum) == 1) {
+        *energyesc += r.weight;
+    } else {
+        for (oldeid = 0; oldeid < GPU_PARAM(gcfg, srcnum); oldeid++) {
+            energyesc[oldeid] += r.weight * ppath[GPU_PARAM(gcfg, reclen) + oldeid];
+        }
     }
 }