e151e136d6
Closes #18 Co-authored-by: Jules Pénuchot <jules@penuchot.com> See merge request oss/rotgen!50
57 lines
1.9 KiB
C++
57 lines
1.9 KiB
C++
//==================================================================================================
|
|
/*
|
|
ROTGEN - Runtime Overlay for Eigen
|
|
Copyright : CODE RECKONS
|
|
SPDX-License-Identifier: BSL-1.0
|
|
*/
|
|
//==================================================================================================
|
|
#include <rotgen/rotgen.hpp>
|
|
|
|
#include "unit/tests.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(innerStride(m), 1);
|
|
TTS_EQUAL(outerStride(m), 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(innerStride(sp), 1);
|
|
TTS_EQUAL(outerStride(sp), 5);
|
|
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,
|
|
rotgen::outer_stride(5));
|
|
TTS_EQUAL(innerStride(dp), 1);
|
|
TTS_EQUAL(outerStride(dp), 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(innerStride(sp), 1);
|
|
TTS_EQUAL(outerStride(sp), 4);
|
|
|
|
rotgen::map<mat_t, 0, rotgen::outer_stride<>> dp(&padded[0], 4, 3,
|
|
rotgen::outer_stride(4));
|
|
TTS_EQUAL(innerStride(dp), 1);
|
|
TTS_EQUAL(outerStride(dp), 4);
|
|
|
|
TTS_EQUAL(m, sp);
|
|
TTS_EQUAL(m, dp);
|
|
TTS_EQUAL(dp, sp);
|
|
}
|
|
};
|