83 lines
3.1 KiB
C++
83 lines
3.1 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include "unit/tests.hpp"
|
|
#include <rotgen/rotgen.hpp>
|
|
#include <iostream>
|
|
|
|
TTS_CASE_TPL("outer_stride<0> interactions", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
using mat_t = rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>;
|
|
|
|
T contiguous[] = {1,2,3, 4,5,6, 7,8,9, 10,11,12};
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<0>> m(&contiguous[0], 4, 3);
|
|
|
|
TTS_EQUAL(m.innerStride(), 1);
|
|
TTS_EQUAL(m.outerStride(), O::value == rotgen::ColMajor ? 4 : 3);
|
|
|
|
if constexpr(O::value == rotgen::ColMajor)
|
|
{
|
|
T padded[] = {1,2,3,4, 99, 5,6,7,8, 99,9,10,11,12};
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<5>> sp(&padded[0], 4, 3);
|
|
TTS_EQUAL(sp.innerStride(), 1);
|
|
TTS_EQUAL(sp.outerStride(), 5);
|
|
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,rotgen::outer_stride(5));
|
|
TTS_EQUAL(dp.innerStride(), 1);
|
|
TTS_EQUAL(dp.outerStride(), 5);
|
|
|
|
TTS_EQUAL(m , sp);
|
|
TTS_EQUAL(m , dp);
|
|
TTS_EQUAL(dp, sp);
|
|
}
|
|
else
|
|
{
|
|
T padded[] = {1,2,3, 99, 4,5,6, 99, 7,8,9, 99, 10,11,12};
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<4>> sp(&padded[0], 4, 3);
|
|
TTS_EQUAL(sp.innerStride(), 1);
|
|
TTS_EQUAL(sp.outerStride(), 4);
|
|
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,rotgen::outer_stride(4));
|
|
TTS_EQUAL(dp.innerStride(), 1);
|
|
TTS_EQUAL(dp.outerStride(), 4);
|
|
|
|
TTS_EQUAL(m , sp);
|
|
TTS_EQUAL(m , dp);
|
|
TTS_EQUAL(dp, sp);
|
|
}
|
|
};
|
|
|
|
void process_ref(rotgen::ref<const rotgen::matrix<float>> ) {}
|
|
void process_ref(rotgen::ref<const rotgen::matrix<double>>) {}
|
|
void process_ref(rotgen::ref<const rotgen::matrix<float , rotgen::Dynamic, rotgen::Dynamic, rotgen::RowMajor>>) {}
|
|
void process_ref(rotgen::ref<const rotgen::matrix<double, rotgen::Dynamic, rotgen::Dynamic, rotgen::RowMajor>>) {}
|
|
|
|
TTS_CASE_TPL("Extraction of outer_stride<?> blocks", rotgen::tests::types)
|
|
<typename T, typename O>( tts::type< tts::types<T,O>> )
|
|
{
|
|
using mat_t = rotgen::matrix<T, rotgen::Dynamic, rotgen::Dynamic, O::value>;
|
|
|
|
if constexpr(O::value == rotgen::ColMajor)
|
|
{
|
|
T padded[] = {1,2,3,4, 99, 5,6,7,8, 99,9,10,11,12};
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<5>> sp(&padded[0], 4, 3);
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,rotgen::outer_stride(5));
|
|
|
|
TTS_EXPECT_COMPILES(sp, { process_ref(extract(sp,0, 0, 3, 2)); } );
|
|
TTS_EXPECT_COMPILES(dp, { process_ref(extract(dp,0, 0, 3, 2)); } );
|
|
}
|
|
else
|
|
{
|
|
T padded[] = {1,2,3, 99, 4,5,6, 99, 7,8,9, 99, 10,11,12};
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<4>> sp(&padded[0], 4, 3);
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,rotgen::outer_stride(4));
|
|
|
|
TTS_EXPECT_COMPILES(sp, { process_ref(extract(sp,0, 0, 3, 2)); } );
|
|
TTS_EXPECT_COMPILES(dp, { process_ref(extract(dp,0, 0, 3, 2)); } );
|
|
}
|
|
};
|