ops.hpp 818 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
#ifndef __INFINIOPTEST_OPS_HPP__
#define __INFINIOPTEST_OPS_HPP__
#include "test.hpp"

/*
 * Declare all the tests here
 */
DECLARE_INFINIOP_TEST(matmul)

#define REGISTER_INFINIOP_TEST(name)                    \
    {                                                   \
        #name,                                          \
        { infiniop_test::name::Test::build,             \
          infiniop_test::name::Test::attribute_names(), \
          infiniop_test::name::Test::tensor_names() }   \
    }

/*
 * Register all the tests here
 */
#define TEST_BUILDER_MAPPINGS           \
    {                                   \
        REGISTER_INFINIOP_TEST(matmul), \
    }

namespace infiniop_test {
extern std::unordered_map<std::string, const TestBuilder> TEST_BUILDERS;

} // namespace infiniop_test

#endif