asm_gemm_kernel_config.h 459 Bytes
Newer Older
Xiaowei.zhang's avatar
Xiaowei.zhang committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#pragma once
#include <string>
#include <vector>

struct KernelCfg {
    std::string kernel_name;
    std::string co_file;
    int mt0, mt1, numThreads, wgm;
};

struct MatchProblem {
    int M;
    int N; 
    int K;
};


KernelCfg get_kernel_cfg_by_index(int index, std::string& filename);
KernelCfg get_kernel_cfg_by_csv(MatchProblem& prob, std::string& filename);
std::vector<std::vector<std::string>> read_csv(const std::string& filename, int* solidx);