From 25d917047774eb20578fa701754a3dd05988a0d7 Mon Sep 17 00:00:00 2001 From: markus <markus.holzer@fau.de> Date: Wed, 17 Jun 2020 17:53:34 +0200 Subject: [PATCH] Fixed pycuda warning --- pystencils_tests/test_buffer_gpu.py | 3 ++- pytest.ini | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/pystencils_tests/test_buffer_gpu.py b/pystencils_tests/test_buffer_gpu.py index f9ee96e9b..f63254cd4 100644 --- a/pystencils_tests/test_buffer_gpu.py +++ b/pystencils_tests/test_buffer_gpu.py @@ -39,7 +39,8 @@ def _generate_fields(dt=np.uint8, stencil_directions=1, layout='numpy'): gpu_src_arr = gpuarray.to_gpu(src_arr) gpu_dst_arr = gpuarray.zeros_like(gpu_src_arr) - gpu_buffer_arr = gpuarray.zeros(np.prod(src_arr.shape), dtype=dt) + size = int(np.prod(src_arr.shape)) + gpu_buffer_arr = gpuarray.zeros(size, dtype=dt) fields.append((src_arr, gpu_src_arr, gpu_dst_arr, gpu_buffer_arr)) return fields diff --git a/pytest.ini b/pytest.ini index ad6eb67d5..2795fb9d8 100644 --- a/pytest.ini +++ b/pytest.ini @@ -39,7 +39,7 @@ exclude_lines = if __name__ == .__main__.: skip_covered = True -fail_under = 74 +fail_under = 75 [html] directory = coverage_report -- GitLab