Пример #1
0
def pyopencl_unique(cpu_data):
    """
    SUT
    This is a function that sets up a GPU context and exposes the pyopencl algorithm uniqe to the TSTL random tester
    The Unique function resembles the UNIX command uniq,
    and requires that values must be sorted to get a list of "unique" values.
    Therefore, this method sorts the cpu_data array before transferring it to the GPU.

    It executes the method python_unique as its oracle.
    :param cpu_data: A numpy array to execute the pyopencl unique algorithm on.
    :return: A boolean that describes if the CPU oracle and the SUT results match
    """
    # platforms = cl.get_platforms()
    # ctx = cl.Context(dev_type=cl.device_type.ALL, properties=[(cl.context_properties.PLATFORM, platforms[0])])
    # queue = cl.CommandQueue(ctx)
    queue = get_queue()

    cpu_data.sort()
    gpu_data = pyopencl.array.to_device(queue, cpu_data)

    unique_gpu, count_unique_gpu, evt = unique(gpu_data)
    unique_cpu = python_unique(cpu_data)

    count_unique_gpu = count_unique_gpu.get()
    unique_gpu = unique_gpu.get()

    return (unique_gpu[:count_unique_gpu] == unique_cpu).all()
Пример #2
0
def test_unique(ctx_factory):
    from pytest import importorskip

    importorskip("mako")

    context = ctx_factory()
    queue = cl.CommandQueue(context)

    from pyopencl.clrandom import rand as clrand

    for n in scan_test_counts:
        a_dev = clrand(queue, (n,), dtype=np.int32, a=0, b=1000)
        a = a_dev.get()
        a = np.sort(a)
        a_dev = cl_array.to_device(queue, a)

        a_unique_host = np.unique(a)

        from pyopencl.algorithm import unique

        a_unique_dev, count_unique_dev, evt = unique(a_dev)

        count_unique_dev = count_unique_dev.get()

        assert (a_unique_dev.get()[:count_unique_dev] == a_unique_host).all()
        from gc import collect

        collect()
Пример #3
0
def test_unique(ctx_factory):
    context = ctx_factory()
    queue = cl.CommandQueue(context)

    from pyopencl.clrandom import rand as clrand
    for n in scan_test_counts:
        a_dev = clrand(queue, (n,), dtype=np.int32, a=0, b=1000)
        a = a_dev.get()
        a = np.sort(a)
        a_dev = cl_array.to_device(queue, a)

        a_unique_host = np.unique(a)

        from pyopencl.algorithm import unique
        a_unique_dev, count_unique_dev = unique(a_dev)

        count_unique_dev = count_unique_dev.get()

        assert (a_unique_dev.get()[:count_unique_dev] == a_unique_host).all()
        from gc import collect
        collect()