summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorStan Seibert <stan@mtrr.org>2011-11-30 09:59:54 -0500
committertlatorre <tlatorre@uchicago.edu>2021-05-09 08:42:38 -0700
commit2893305a07789ab880821093f8d719e3eb6c8e31 (patch)
tree79d62840a72b2a27be83314dd44b91d004e73144
parent6ae6a660d0276bf671362e8456edbf8ead6f3e5c (diff)
downloadchroma-2893305a07789ab880821093f8d719e3eb6c8e31.tar.gz
chroma-2893305a07789ab880821093f8d719e3eb6c8e31.tar.bz2
chroma-2893305a07789ab880821093f8d719e3eb6c8e31.zip
Skip photon propagation queue if we are propagating weighted photons
-rw-r--r--chroma/gpu/photon.py2
1 files changed, 1 insertions, 1 deletions
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