blob: fcd43f8bc3e537d5c49ed6e2ff7ce7588a3358b5 (
plain)
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
|
/*Lex file for output of cuobjdump*/
%{
#include <stdio.h>
#include <string.h>
#include "cuobjdump_parser.h"
#define YY_NEVER_INTERACTIVE 1
#define YYDEBUG 1
#define yylval cuobjdump_lval
void cuobjdump_error(const char*);
%}
%option stack
%option noyywrap
%option yylineno
%s ptxcode
%s sasscode
%s elfcode
%s header
%s endheader
alpha [a-zA-Z]
numeric [0-9]
whitespace [ \t]
newline [\n]
notnewline [^\n]
newlines {newline}+
%%
"ptxasOptions"{notnewline}*{newline}
[1-9]{numeric}* yylval.string_value = strdup(yytext); return DECIMAL;
"Fatbin ptx code:"{newline} {
yy_push_state(ptxcode);
yy_push_state(header);
yylval.string_value = strdup(yytext);
return PTXHEADER;
}
"Fatbin elf code:"{newline} {
yy_push_state(elfcode);
yy_push_state(header);
yylval.string_value = strdup(yytext);
return ELFHEADER;
}
/*PTX code tokens*/
<ptxcode>{notnewline}*{newline} yylval.string_value = strdup(yytext); return PTXLINE;
/*ELF code tokens*/
<elfcode>{whitespace}*"code for sm_"{numeric}+{newline} {
BEGIN(sasscode);
yylval.string_value = strdup(yytext);
return SASSLINE;
}
<elfcode>{notnewline}*{newline} yylval.string_value = strdup(yytext); return ELFLINE;
/*SASS code tokens*/
<sasscode>{notnewline}*{newline} yylval.string_value = strdup(yytext); return SASSLINE;
/*Header tokens*/
<header>[[:alnum:]_]+ yylval.string_value = strdup(yytext); return IDENTIFIER;
<header>"================" return H_SEPARATOR;
<header>"arch = " return H_ARCH;
<header>"code version = " return H_CODEVERSION;
<header>\[{numeric},{numeric}\] return CODEVERSION;
<header>"producer = " return H_PRODUCER;
<header>"host = " return H_HOST;
<header>"compile_size = " return H_COMPILESIZE;
<header>"identifier = " BEGIN(endheader); return H_IDENTIFIER;
<header>"has debug info"{newline}
/* Looking for the identifier (filename) then the header is done */
<endheader>[[:alnum:]_\.]+{newline} yylval.string_value = strdup(yytext); yy_pop_state(); return FILENAME;
{newline} return NEWLINE;
/*Reached end of file*/
<<EOF>> return 0;
/*No other rule matched. Throw an error*/
. cuobjdump_error("Invalid token");
%%
void cuobjdump_error(const char* message)
{
printf(" "); printf(message); printf(" near \""); printf(yytext); printf("\"");
printf(" on line ");
char line[5]; sprintf(line, "%i", yylineno); printf(line);
printf("\n");
}
|
