GeneralSparse/executor.cc at main · Wangyaoyuu/GeneralSparse · GitHub

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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#include "executor.hpp"
#include <iterator>

using namespace std;

bool execute_binary(string execute_path, string filename, float &total_execute_time, float &gflops)
{
    // execute_path是执行目录
    // 删除执行目录可能存在结果文件
    string command = "cd " + execute_path + " && rm perf_result";
    system(command.c_str());

    // 执行编译
    bool is_compiler_success = compile_spmv_code(execute_path);

    if (is_compiler_success == false)
    {
        return false;
    }

    // 这里代表编译成功，成功之后就开始执行
    command = "cd " + execute_path + " && ./a.out " + get_config()["DATA_SET"].as_string() + filename + "";

    // 执行
    system(command.c_str());

    // 执行完之后查看结果文件是不是写出来了
    ifstream read_perf_result(execute_path + "/perf_result");

    // 有没有对应的文件
    if (!read_perf_result)
    {
        cout << "fail execution" << endl;
        return false;
    }

    float file_time;
    float file_gflops;

    // 获得时间
    read_perf_result >> file_time;
    read_perf_result >> file_gflops;

    // cout << file_time << " " << file_gflops << endl;

    read_perf_result.close();

    total_execute_time = file_time;
    gflops = file_gflops;

    if (file_gflops > get_config()["GFLOPS_UP_BOUND"].as_integer())
    {
        cout << "gflops is too high, maybe some mistake happened in kernal" << endl;
        return false;
    }

    return true;
}

bool compile_spmv_code(string execute_path)
{
    // 删除编译器的输出
    string command = "cd " + execute_path + " && rm compile_result";
    system(command.c_str());

    // 执行编译
    command = "cd " + execute_path + " && sh make_kernel.sh > compile_result 2>&1";
    system(command.c_str());

    // 读取编译结果中的内容，放在compile_result中
    ifstream in(execute_path + "/compile_result");
    istreambuf_iterator<char> begin(in);
    istreambuf_iterator<char> end;
    string compile_result(begin, end);

    string::size_type position = compile_result.find("error");


    // 查看编译的结果
    if (position != compile_result.npos)
    {
        // 编译有错误
        cout << "compile error!" << endl
             << compile_result << endl;
        assert(false);
        return false;
    }
    else
    {
        // 编译没有错误
        cout << "compile success" << endl;
    }

    ifstream fin(execute_path + "/a.out");

    if (!fin)
    {
        cout << "can't find compile result" << endl;
        return false;
    }

    fin.close();

    return true;
}