109 lines
No EOL
3.7 KiB
C++
109 lines
No EOL
3.7 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include "unit/tests.hpp"
|
|
#include "unit/common/arithmetic.hpp"
|
|
#include <rotgen/rotgen.hpp>
|
|
|
|
TTS_CASE_TPL("Test dynamic map transposition-like operations", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
auto const cases = rotgen::tests::generate_matrix_references();
|
|
for (const auto& [rows, cols, fn] : cases)
|
|
{
|
|
using mat_t = rotgen::matrix<T,rotgen::Dynamic,rotgen::Dynamic,O::value>;
|
|
mat_t base(rows, cols);
|
|
rotgen::tests::prepare(rows,cols,fn,base);
|
|
|
|
rotgen::map<mat_t> input(base.data(),rows,cols);
|
|
rotgen::tests::check_shape_functions(input);
|
|
}
|
|
};
|
|
|
|
TTS_CASE_TPL("Test static map transposition-like operations", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
auto const cases = rotgen::tests::generate_static_matrix_references();
|
|
|
|
auto process = []<typename D>(D const& desc)
|
|
{
|
|
rotgen::matrix<T,rotgen::Dynamic,rotgen::Dynamic,O::value> base(D::rows,D::cols);
|
|
rotgen::tests::prepare(base.rows(),base.cols(),desc.init_fn,base);
|
|
|
|
rotgen::map<rotgen::matrix<T,D::rows,D::cols,O::value>> input(base.data());
|
|
rotgen::tests::check_shape_functions(input);
|
|
};
|
|
|
|
std::apply([&](auto const&... d) { (process(d),...);}, cases);
|
|
};
|
|
|
|
TTS_CASE_TPL("Test dynamic map reduction-like operations", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
auto const cases = rotgen::tests::generate_matrix_references();
|
|
for (const auto& [rows, cols, fn] : cases)
|
|
{
|
|
rotgen::matrix<T,rotgen::Dynamic,rotgen::Dynamic,O::value> base(rows, cols);
|
|
rotgen::tests::prepare(rows,cols,fn,base);
|
|
|
|
rotgen::map<rotgen::matrix<T,rotgen::Dynamic,rotgen::Dynamic,O::value>> input(base.data(),rows,cols);
|
|
rotgen::tests::check_reduction_functions(input);
|
|
}
|
|
};
|
|
|
|
TTS_CASE_TPL("Test static map reduction-like operations", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
auto const cases = rotgen::tests::generate_static_matrix_references();
|
|
|
|
auto process = []<typename D>(D const& desc)
|
|
{
|
|
rotgen::matrix<T,rotgen::Dynamic,rotgen::Dynamic,O::value> base(D::rows,D::cols);
|
|
rotgen::tests::prepare(base.rows(),base.cols(),desc.init_fn,base);
|
|
|
|
rotgen::map<rotgen::matrix<T,D::rows,D::cols,O::value>> input(base.data());
|
|
rotgen::tests::check_reduction_functions(input);
|
|
};
|
|
|
|
std::apply([&](auto const&... d) { (process(d),...);}, cases);
|
|
};
|
|
|
|
TTS_CASE_TPL("Test dot product", float, double)
|
|
<typename T>( tts::type<T> )
|
|
{
|
|
{
|
|
auto v = rotgen::setConstant<rotgen::matrix<T,1,rotgen::Dynamic>>(1,16,2);
|
|
auto a = rotgen::map<decltype(v)>(v.data(),1,8);
|
|
auto b = rotgen::map<decltype(v)>(v.data()+8,1,8);
|
|
|
|
TTS_EQUAL(rotgen::dot(a,b), 32);
|
|
}
|
|
|
|
{
|
|
auto v = rotgen::setConstant<rotgen::matrix<T,rotgen::Dynamic,1>>(16,1,2);
|
|
auto a = rotgen::map<decltype(v)>(v.data(),8,1);
|
|
auto b = rotgen::map<decltype(v)>(v.data()+8,8,1);
|
|
|
|
TTS_EQUAL(rotgen::dot(a,b), 32);
|
|
}
|
|
|
|
{
|
|
auto v = rotgen::setConstant<rotgen::matrix<T,1,rotgen::Dynamic>>(1,16,2);
|
|
auto a = rotgen::map<rotgen::matrix<T,1,8>>(v.data());
|
|
auto b = rotgen::map<rotgen::matrix<T,1,8>>(v.data()+8);
|
|
|
|
TTS_EQUAL(rotgen::dot(a,b), 32);
|
|
}
|
|
|
|
{
|
|
auto v = rotgen::setConstant<rotgen::matrix<T,1,rotgen::Dynamic>>(1,16,2);
|
|
auto a = rotgen::map<rotgen::matrix<T,8,1>>(v.data());
|
|
auto b = rotgen::map<rotgen::matrix<T,8,1>>(v.data()+8);
|
|
|
|
TTS_EQUAL(rotgen::dot(a,b), 32);
|
|
}
|
|
}; |