gemm_common.hpp 236 Bytes
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
5
6
7
8
9
#ifndef GEMM_COMMON_HPP
#define GEMM_COMMON_HPP

enum GemmMatrixLayout
{
    MK_KN_MN, // 0
    MK_NK_MN, // 1
    KM_KN_MN, // 2
    KM_NK_MN, // 3
Chao Liu's avatar
Chao Liu committed
10
11
12
13
    MK_KN_NM, // 4
    MK_NK_NM, // 5
    KM_KN_NM, // 6
    KM_NK_NM, // 7
Chao Liu's avatar
Chao Liu committed
14
15
16
};

#endif