diff options
Diffstat (limited to 'test/test_pdf.py')
| -rw-r--r-- | test/test_pdf.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/test/test_pdf.py b/test/test_pdf.py index 1d9e20a..0ad87d8 100644 --- a/test/test_pdf.py +++ b/test/test_pdf.py @@ -38,8 +38,10 @@ class TestPDF(unittest.TestCase): gpu_photons = gpu.GPUPhotons(ev.photons_beg) gpu_photons.propagate(gpu_geometry, rng_states, nthreads_per_block, max_blocks) - gpu_channels = gpu_daq.acquire(gpu_photons, rng_states, - nthreads_per_block, max_blocks) + gpu_daq.begin_acquire() + gpu_daq.acquire(gpu_photons, rng_states, + nthreads_per_block, max_blocks) + gpu_channels = gpu_daq.end_acquire() gpu_pdf.add_hits_to_pdf(gpu_channels) hitcount, pdf = gpu_pdf.get_pdfs() |