pack_args.cpp

#include <test.hpp>
#include <migraphx/gpu/pack_args.hpp>

template <class T>
int packed_sizes()
{
    return sizeof(T);
}

template <class T, class U, class... Ts>
int packed_sizes()
{
    return sizeof(T) + packed_sizes<U, Ts...>();
}

template <class... Ts>
int sizes()
{
    return migraphx::gpu::pack_args({Ts{}...}).size();
}

template <class... Ts>
int padding()
{
    EXPECT(sizes<Ts...>() >= packed_sizes<Ts...>());
    return sizes<Ts...>() - packed_sizes<Ts...>();
}

struct float_struct
{
    float x, y;
};

TEST_CASE(alignment_padding)
{
    EXPECT(padding<short, short>() == 0);
    EXPECT(padding<float, float_struct>() == 0);
    EXPECT(padding<short, float_struct>() == 2);
    EXPECT(padding<short, int>() == 2);
    EXPECT(padding<char, short, int, char>() == 1);
}

int main(int argc, const char* argv[]) { test::run(argc, argv); }