summaryrefslogtreecommitdiff
path: root/gpu.py
diff options
context:
space:
mode:
Diffstat (limited to 'gpu.py')
-rw-r--r--gpu.py4
1 files changed, 4 insertions, 0 deletions
diff --git a/gpu.py b/gpu.py
index c9e6269..5ad195d 100644
--- a/gpu.py
+++ b/gpu.py
@@ -12,6 +12,7 @@ from tools import timeit
import src
from geometry import standard_wavelengths
from color import map_to_color
+import sys
cuda.init()
@@ -308,6 +309,9 @@ class GPU(object):
output_queue_gpu = temp
output_queue_gpu[:1].set(np.uint32(1))
nphotons = input_queue_gpu[:1].get()[0]
+
+ if gpuarray.max(self.histories_gpu).get() & (1 << 31):
+ print >>sys.stderr, "WARNING: ABORTED PHOTONS IN THIS EVENT"
if 'profile' in __builtins__:
self.context.synchronize()