Skip to content
Snippets Groups Projects
Commit 729b9d01 authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Merge branch 'test-gpu-in-gpu-test' into 'master'

Actually test GPU in test_boundary_handling.test_simple

See merge request pycodegen/lbmpy!25
parents fb08518d 020315cf
1 merge request!25Actually test GPU in test_boundary_handling.test_simple
Pipeline #23618 passed with stages
in 13 minutes and 36 seconds
import numpy as np
import pytest
from lbmpy.boundaries import UBB, NeumannByCopy, NoSlip, StreamInConstant
......@@ -10,15 +9,24 @@ from lbmpy.lbstep import LatticeBoltzmannStep
from pystencils import create_data_handling, make_slice
@pytest.mark.parametrize("gpu", [True, False])
def test_simple(gpu):
import pytest
pytest.importorskip('pycuda')
dh = create_data_handling((10, 5), parallel=False)
dh.add_array('pdfs', values_per_cell=9, cpu=True, gpu=gpu)
lb_func = create_lb_function(stencil='D2Q9', compressible=False, relaxation_rate=1.8)
bh = LatticeBoltzmannBoundaryHandling(lb_func.method, dh, 'pdfs')
@pytest.mark.parametrize("target", ['cpu', 'gpu', 'opencl'])
def test_simple(target):
if target == 'gpu':
import pytest
pytest.importorskip('pycuda')
elif target == 'opencl':
import pytest
pytest.importorskip('pyopencl')
import pystencils.opencl.autoinit
dh = create_data_handling((10, 5), parallel=False, default_target=target)
dh.add_array('pdfs', values_per_cell=9, cpu=True, gpu=target!='cpu')
lb_func = create_lb_function(stencil='D2Q9',
compressible=False,
relaxation_rate=1.8,
optimization={'target': target})
bh = LatticeBoltzmannBoundaryHandling(lb_func.method, dh, 'pdfs', target=target)
wall = NoSlip()
moving_wall = UBB((0.001, 0))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment