1 /*
2 * Copyright (C) 2019 Ryan Houdek <Sonicadvance1@gmail.com>
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
18 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 * SOFTWARE.
22 */
23
24 #include "disassemble.h"
25
26 #include "main/mtypes.h"
27 #include "compiler/glsl/standalone.h"
28 #include "compiler/glsl/glsl_to_nir.h"
29 #include "compiler/glsl/gl_nir.h"
30 #include "compiler/nir_types.h"
31 #include "util/u_dynarray.h"
32 #include "bifrost_compile.h"
33 #include "test/bit.h"
34
35 static panfrost_program *
compile_shader(char ** argv,bool vertex_only)36 compile_shader(char **argv, bool vertex_only)
37 {
38 struct gl_shader_program *prog;
39 nir_shader *nir[2];
40 unsigned shader_types[2] = {
41 MESA_SHADER_VERTEX,
42 MESA_SHADER_FRAGMENT,
43 };
44
45 struct standalone_options options = {
46 .glsl_version = 300, /* ES - needed for precision */
47 .do_link = true,
48 .lower_precision = true
49 };
50
51 static struct gl_context local_ctx;
52
53 prog = standalone_compile_shader(&options, 2, argv, &local_ctx);
54 prog->_LinkedShaders[MESA_SHADER_FRAGMENT]->Program->info.stage = MESA_SHADER_FRAGMENT;
55
56 panfrost_program *compiled;
57 for (unsigned i = 0; i < 2; ++i) {
58 nir[i] = glsl_to_nir(&local_ctx, prog, shader_types[i], &bifrost_nir_options);
59 NIR_PASS_V(nir[i], nir_lower_global_vars_to_local);
60 NIR_PASS_V(nir[i], nir_lower_io_to_temporaries, nir_shader_get_entrypoint(nir[i]), true, i == 0);
61 NIR_PASS_V(nir[i], nir_split_var_copies);
62 NIR_PASS_V(nir[i], nir_lower_var_copies);
63
64 /* before buffers and vars_to_ssa */
65 NIR_PASS_V(nir[i], gl_nir_lower_images, true);
66
67 NIR_PASS_V(nir[i], gl_nir_lower_buffers, prog);
68 NIR_PASS_V(nir[i], nir_opt_constant_folding);
69
70 struct panfrost_compile_inputs inputs = {
71 .gpu_id = 0x7212, /* Mali G52 */
72 };
73
74 compiled = bifrost_compile_shader_nir(NULL, nir[i], &inputs);
75
76 if (vertex_only)
77 return compiled;
78 }
79
80 return compiled;
81 }
82
83 static void
disassemble(const char * filename,bool verbose)84 disassemble(const char *filename, bool verbose)
85 {
86 FILE *fp = fopen(filename, "rb");
87 assert(fp);
88
89 fseek(fp, 0, SEEK_END);
90 unsigned filesize = ftell(fp);
91 rewind(fp);
92
93 unsigned char *code = malloc(filesize);
94 unsigned res = fread(code, 1, filesize, fp);
95 if (res != filesize) {
96 printf("Couldn't read full file\n");
97 }
98 fclose(fp);
99
100 disassemble_bifrost(stdout, code, filesize, verbose);
101 free(code);
102 }
103
104 static void
test_vertex(char ** argv)105 test_vertex(char **argv)
106 {
107 void *memctx = NULL; /* TODO */
108 struct panfrost_device *dev = bit_initialize(memctx);
109
110 float iubo[] = {
111 0.1, 0.2, 0.3, 0.4
112 };
113
114 float iattr[] = {
115 0.5, 0.6, 0.7, 0.8
116 };
117
118 float expected[] = {
119 0.6, 0.8, 1.0, 1.2
120 };
121
122 bit_vertex(dev, compile_shader(argv, true),
123 (uint32_t *) iubo, sizeof(iubo),
124 (uint32_t *) iattr, sizeof(iattr),
125 (uint32_t *) expected, sizeof(expected),
126 BIT_DEBUG_ALL);
127 }
128
129 static void
tests(void)130 tests(void)
131 {
132 void *memctx = NULL; /* TODO */
133 struct panfrost_device *dev = bit_initialize(memctx);
134 bit_packing(dev, BIT_DEBUG_FAIL);
135 }
136
137 static void
run(const char * filename)138 run(const char *filename)
139 {
140 FILE *fp = fopen(filename, "rb");
141 assert(fp);
142
143 fseek(fp, 0, SEEK_END);
144 unsigned filesize = ftell(fp);
145 rewind(fp);
146
147 unsigned char *code = malloc(filesize);
148 unsigned res = fread(code, 1, filesize, fp);
149 if (res != filesize) {
150 printf("Couldn't read full file\n");
151 }
152 fclose(fp);
153
154 void *memctx = NULL; /* TODO */
155 struct panfrost_device *dev = bit_initialize(memctx);
156
157 panfrost_program prog = {
158 .compiled = {
159 .data = code,
160 .size = filesize
161 },
162 };
163
164 bit_vertex(dev, &prog, NULL, 0, NULL, 0, NULL, 0, BIT_DEBUG_ALL);
165
166 free(code);
167 }
168
169 int
main(int argc,char ** argv)170 main(int argc, char **argv)
171 {
172 if (argc < 2) {
173 printf("Pass a command\n");
174 exit(1);
175 }
176
177 if (strcmp(argv[1], "compile") == 0)
178 compile_shader(&argv[2], false);
179 else if (strcmp(argv[1], "disasm") == 0)
180 disassemble(argv[2], false);
181 else if (strcmp(argv[1], "disasm-verbose") == 0)
182 disassemble(argv[2], true);
183 else if (strcmp(argv[1], "tests") == 0)
184 tests();
185 else if (strcmp(argv[1], "test-vertex") == 0)
186 test_vertex(&argv[2]);
187 else if (strcmp(argv[1], "run") == 0)
188 run(argv[2]);
189 else
190 unreachable("Unknown command. Valid: compile/disasm");
191
192 return 0;
193 }
194