diff options
Diffstat (limited to 'chroma/gpu')
| -rw-r--r-- | chroma/gpu/daq.py | 2 | ||||
| -rw-r--r-- | chroma/gpu/tools.py | 1 |
2 files changed, 2 insertions, 1 deletions
diff --git a/chroma/gpu/daq.py b/chroma/gpu/daq.py index 82958dc..6ec1a79 100644 --- a/chroma/gpu/daq.py +++ b/chroma/gpu/daq.py @@ -20,7 +20,7 @@ class GPUChannels(object): return event.Channels(t<1e8, t, q, self.flags.get()) class GPUDaq(object): - def __init__(self, gpu_geometry, max_pmt_id, pmt_rms=1.2e-9): + def __init__(self, gpu_geometry, max_pmt_id, pmt_rms=1.2): self.earliest_time_gpu = ga.empty(max_pmt_id+1, dtype=np.float32) self.earliest_time_int_gpu = ga.empty(max_pmt_id+1, dtype=np.uint32) self.channel_history_gpu = ga.zeros_like(self.earliest_time_int_gpu) diff --git a/chroma/gpu/tools.py b/chroma/gpu/tools.py index a11a561..38d67a1 100644 --- a/chroma/gpu/tools.py +++ b/chroma/gpu/tools.py @@ -2,6 +2,7 @@ import numpy as np import pycuda.tools from pycuda import characterize import pycuda.driver as cuda +import pycuda.compiler from pycuda import gpuarray as ga from chroma.cuda import srcdir |