From 2893305a07789ab880821093f8d719e3eb6c8e31 Mon Sep 17 00:00:00 2001
From: Stan Seibert <stan@mtrr.org>
Date: Wed, 30 Nov 2011 09:59:54 -0500
Subject: Skip photon propagation queue if we are propagating weighted photons

---
 chroma/gpu/photon.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/chroma/gpu/photon.py b/chroma/gpu/photon.py
index 80fe77c..83dfcd4 100644
--- a/chroma/gpu/photon.py
+++ b/chroma/gpu/photon.py
@@ -120,7 +120,7 @@ class GPUPhotons(object):
 
         while step < max_steps:
             # Just finish the rest of the steps if the # of photons is low
-            if nphotons < nthreads_per_block * 16 * 8:
+            if nphotons < nthreads_per_block * 16 * 8 or use_weights:
                 nsteps = max_steps - step
             else:
                 nsteps = 1
-- 
cgit