compute/test/test_invoke.cpp
Kyle Lutz ef109b75d5 Add invoke() utility function
This adds an invoke() function which calls an OpenCL function on a
device with the given arguments.
2015-08-19 08:18:15 -07:00

60 lines
1.7 KiB
C++

//---------------------------------------------------------------------------//
// Copyright (c) 2013-2015 Kyle Lutz <kyle.r.lutz@gmail.com>
//
// Distributed under the Boost Software License, Version 1.0
// See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt
//
// See http://kylelutz.github.com/compute for more information.
//---------------------------------------------------------------------------//
#define BOOST_TEST_MODULE TestInvoke
#include <boost/test/unit_test.hpp>
#include <boost/compute/function.hpp>
#include <boost/compute/lambda.hpp>
#include <boost/compute/functional/integer.hpp>
#include <boost/compute/functional/math.hpp>
#include <boost/compute/utility/invoke.hpp>
#include "context_setup.hpp"
namespace compute = boost::compute;
BOOST_AUTO_TEST_CASE(invoke_builtin)
{
BOOST_CHECK_EQUAL(compute::invoke(compute::abs<int>(), queue, -3), 3);
BOOST_CHECK_CLOSE(compute::invoke(compute::pow<float>(), queue, 2.f, 8.f), 256.f, 1e-4);
}
BOOST_AUTO_TEST_CASE(invoke_function)
{
BOOST_COMPUTE_FUNCTION(int, plus_two, (int x),
{
return x + 2;
});
BOOST_CHECK_EQUAL(compute::invoke(plus_two, queue, 2), 4);
BOOST_COMPUTE_FUNCTION(float, get_max, (float x, float y),
{
if(x > y)
return x;
else
return y;
});
BOOST_CHECK_EQUAL(compute::invoke(get_max, queue, 10.f, 20.f), 20.f);
}
BOOST_AUTO_TEST_CASE(invoke_lambda)
{
using boost::compute::lambda::_1;
using boost::compute::lambda::_2;
BOOST_CHECK_EQUAL(compute::invoke(_1 / 2, queue, 4), 2);
BOOST_CHECK_EQUAL(compute::invoke(_1 * _2 + 1, queue, 3, 3), 10);
}
BOOST_AUTO_TEST_SUITE_END()