gpuTrace/cuda_helper_preprocessor.py at master · kpouget/gpuTrace · 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
#! /usr/bin/env python3

import sys
import re
import collections

import gdb_helper
import parse_gpu_program

def print_head(_lookup_table):
    print("#include <stddef.h>")
    print("#include <stdio.h>")
    print('#include "cuda_helper.h"')
    #print("typedef float realw;")
    print("\n#define NB_CUDA_KERNEL {}".format(len(_lookup_table)))
    print("int cuda_get_nb_kernels(void) {return NB_CUDA_KERNEL;}")
    print("void cuda_init_helper(void) {}")

def get_parameters(params, fname):
    yield from params

def get_safe_function_name(fname):
    return fname.replace("<", "__").replace(">", "__")

def get_arg_struct_name(fname):
    return "struct {}_args_s".format(get_safe_function_name(fname))

def print_parameter_structs(_lookup_table):
    def is_pointer_type(ptype):
        return ptype[-1] == '*'

    for address, (fname, params) in _lookup_table.items():
        print("%s {" % get_arg_struct_name(fname))
        for ptype, pname in reversed([p for p in get_parameters(params, fname)]):
            print("  {}{}{};".format(ptype, "" if is_pointer_type(ptype) else " ", pname))
        print("};")
        print()

PRINT_TYPE_LOOKUP = {
    "int": "%d",
    "float": "%.2f",
    "char *": "%p",
    "const char *": "%s"
    }

def print_spacer():
    print("/****************************/")

def print_lookup_table_struct(_lookup_table):
    lookup_table_items = ",\n".join([i for i in print_and_get_lookup_table_items(_lookup_table)])
    print("static struct kernel_lookup_s function_lookup_table[] = { \n%s\n};" % lookup_table_items)
    print("""
struct kernel_lookup_s *cuda_get_lookup_table(void) {
    return function_lookup_table;
}""")


def print_preprocessor(binary):
    lookup_table = parse_gpu_program.cuda_get_lookup_table(binary)

    print_head(lookup_table)

    print_parameter_structs(lookup_table)

    print_lookup_table_struct(lookup_table)

def print_and_get_lookup_table_items(_lookup_table):
    for address, (fname, params) in _lookup_table.items():
        nb_params = len(params)
        yield '  {(void *) %s, "%s", %d, %s}' % \
            (address, fname, nb_params,
             print_and_get_param_info_items(params, fname))

def print_and_get_param_info_items(params, fname):

    def get_param_items():
        for ptype, pname in get_parameters(params, fname):
            yield '{"%s", "%s"}' % (pname, ptype)

    varname = "{}_params".format(get_safe_function_name(fname))
    print("struct param_info_s %s[] = {\n  %s\n};\n" % (varname, ",\n  ".join([i for i in get_param_items()])))
    return varname


if __name__ == "__main__":
    binary = sys.argv[1]

    print_preprocessor(binary)