diff options
Diffstat (limited to 'gpu.py')
-rw-r--r-- | gpu.py | 4 |
1 files changed, 4 insertions, 0 deletions
@@ -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() |