2
0
mirror of https://github.com/boostorg/compute.git synced 2026-01-22 05:02:15 +00:00
Files
compute/test/test_program.cpp

98 lines
3.1 KiB
C++

//---------------------------------------------------------------------------//
// Copyright (c) 2013 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 TestProgram
#include <boost/test/unit_test.hpp>
#include <boost/compute/kernel.hpp>
#include <boost/compute/system.hpp>
#include <boost/compute/program.hpp>
#include "context_setup.hpp"
const char source[] =
"__kernel void foo(__global float *x, const uint n) { }\n"
"__kernel void bar(__global int *x, __global int *y) { }\n";
BOOST_AUTO_TEST_CASE(get_program_info)
{
// create program
boost::compute::program program =
boost::compute::program::create_with_source(source, context);
// build program
program.build();
// check program info
#ifndef BOOST_COMPUTE_USE_OFFLINE_CACHE
BOOST_CHECK(program.source().empty() == false);
#endif
BOOST_CHECK(program.get_context() == context);
}
BOOST_AUTO_TEST_CASE(create_kernel)
{
boost::compute::program program =
boost::compute::program::create_with_source(source, context);
program.build();
boost::compute::kernel foo = program.create_kernel("foo");
boost::compute::kernel bar = program.create_kernel("bar");
// try to create a kernel that doesn't exist
BOOST_CHECK_THROW(program.create_kernel("baz"), boost::compute::opencl_error);
}
BOOST_AUTO_TEST_CASE(create_with_binary)
{
// create program from source
boost::compute::program source_program =
boost::compute::program::create_with_source(source, context);
source_program.build();
// create kernels in source program
boost::compute::kernel source_foo_kernel = source_program.create_kernel("foo");
boost::compute::kernel source_bar_kernel = source_program.create_kernel("bar");
// check source kernels
BOOST_CHECK_EQUAL(source_foo_kernel.name(), std::string("foo"));
BOOST_CHECK_EQUAL(source_bar_kernel.name(), std::string("bar"));
// get binary
std::vector<unsigned char> binary = source_program.binary();
// create program from binary
boost::compute::program binary_program =
boost::compute::program::create_with_binary(binary, context);
binary_program.build();
// create kernels in binary program
boost::compute::kernel binary_foo_kernel = binary_program.create_kernel("foo");
boost::compute::kernel binary_bar_kernel = binary_program.create_kernel("bar");
// check binary kernels
BOOST_CHECK_EQUAL(binary_foo_kernel.name(), std::string("foo"));
BOOST_CHECK_EQUAL(binary_bar_kernel.name(), std::string("bar"));
}
BOOST_AUTO_TEST_CASE(create_with_source_doctest)
{
//! [create_with_source]
std::string source = "__kernel void foo(__global int *data) { }";
boost::compute::program foo_program =
boost::compute::program::create_with_source(source, context);
//! [create_with_source]
foo_program.build();
}
BOOST_AUTO_TEST_SUITE_END()