#include "ijk.h"
#include <benchmark/benchmark.h>
#include <iostream>

template<class T, auto f> void test(benchmark::State &state)
{
    T a[N][N], b[N][N], c[N][N];

    for (size_t i = 0; i < N; ++i)
        for (size_t j = 0; j < N; ++j)
        {
            a[i][j] = 0;
            b[i][j] = B;
            c[i][j] = C;
        }

    for (auto _ : state) { f(a, b, c); }
}

#define test_type(t)                                                          \
    BENCHMARK(test<t, ijk<t>>);                                               \
    BENCHMARK(test<t, ikj<t>>);                                               \
    BENCHMARK(test<t, jik<t>>);                                               \
    BENCHMARK(test<t, jki<t>>);                                               \
    BENCHMARK(test<t, kij<t>>);                                               \
    BENCHMARK(test<t, kji<t>>);

test_type(char);
test_type(short);
test_type(int);
test_type(long);
test_type(__int128);
test_type(float);
test_type(double);
test_type(long double);
test_type(__float128);

BENCHMARK_MAIN();