mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-16 10:59:55 +00:00
0e75841071
* add parameters
* tweak gemm
* tweak
* update conv
* update script
* adding bwd 1x1
* update script
* adding 1x1 bwd
* debugging bwd 1x1 failure
* update script
* update script
* test
* test v100
* clean up
[ROCm/composable_kernel commit: b3e8d57d51]
17 lines
236 B
C++
17 lines
236 B
C++
#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
|
|
MK_KN_NM, // 4
|
|
MK_NK_NM, // 5
|
|
KM_KN_NM, // 6
|
|
KM_NK_NM, // 7
|
|
};
|
|
|
|
#endif
|