2 * Copyright © 2012 Intel Corporation
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:
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
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
20 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
24 #include <gtest/gtest.h>
32 #define register_coalesce(v) _register_coalesce(v, __func__)
34 class register_coalesce_test
: public ::testing::Test
{
38 struct brw_compiler
*compiler
;
39 struct brw_device_info
*devinfo
;
40 struct gl_context
*ctx
;
41 struct gl_shader_program
*shader_prog
;
42 struct brw_vertex_program
*vp
;
47 class register_coalesce_vec4_visitor
: public vec4_visitor
50 register_coalesce_vec4_visitor(struct brw_compiler
*compiler
,
52 : vec4_visitor(compiler
, NULL
, NULL
, NULL
, shader
, NULL
,
53 false /* no_spills */, -1)
58 virtual dst_reg
*make_reg_for_system_value(int location
,
59 const glsl_type
*type
)
61 unreachable("Not reached");
64 virtual void setup_payload()
66 unreachable("Not reached");
69 virtual void emit_prolog()
71 unreachable("Not reached");
74 virtual void emit_thread_end()
76 unreachable("Not reached");
79 virtual void emit_urb_write_header(int mrf
)
81 unreachable("Not reached");
84 virtual vec4_instruction
*emit_urb_write_opcode(bool complete
)
86 unreachable("Not reached");
91 void register_coalesce_test::SetUp()
93 ctx
= (struct gl_context
*)calloc(1, sizeof(*ctx
));
94 compiler
= (struct brw_compiler
*)calloc(1, sizeof(*compiler
));
95 devinfo
= (struct brw_device_info
*)calloc(1, sizeof(*devinfo
));
96 compiler
->devinfo
= devinfo
;
98 vp
= ralloc(NULL
, struct brw_vertex_program
);
100 nir_shader
*shader
= nir_shader_create(NULL
, MESA_SHADER_VERTEX
, NULL
);
102 v
= new register_coalesce_vec4_visitor(compiler
, shader
);
104 _mesa_init_gl_program(&vp
->program
.Base
, GL_VERTEX_SHADER
, 0);
110 _register_coalesce(vec4_visitor
*v
, const char *func
)
115 printf("%s: instructions before:\n", func
);
116 v
->dump_instructions();
120 v
->opt_register_coalesce();
123 printf("%s: instructions after:\n", func
);
124 v
->dump_instructions();
128 TEST_F(register_coalesce_test
, test_compute_to_mrf
)
130 src_reg something
= src_reg(v
, glsl_type::float_type
);
131 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
134 dst_reg m0
= dst_reg(MRF
, 0);
135 m0
.writemask
= WRITEMASK_X
;
136 m0
.type
= BRW_REGISTER_TYPE_F
;
138 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, something
, src_reg(1.0f
)));
139 v
->emit(v
->MOV(m0
, src_reg(temp
)));
141 register_coalesce(v
);
143 EXPECT_EQ(mul
->dst
.file
, MRF
);
147 TEST_F(register_coalesce_test
, test_multiple_use
)
149 src_reg something
= src_reg(v
, glsl_type::float_type
);
150 dst_reg temp
= dst_reg(v
, glsl_type::vec4_type
);
153 dst_reg m0
= dst_reg(MRF
, 0);
154 m0
.writemask
= WRITEMASK_X
;
155 m0
.type
= BRW_REGISTER_TYPE_F
;
157 dst_reg m1
= dst_reg(MRF
, 1);
158 m1
.writemask
= WRITEMASK_XYZW
;
159 m1
.type
= BRW_REGISTER_TYPE_F
;
161 src_reg src
= src_reg(temp
);
162 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, something
, src_reg(1.0f
)));
163 src
.swizzle
= BRW_SWIZZLE_XXXX
;
164 v
->emit(v
->MOV(m0
, src
));
165 src
.swizzle
= BRW_SWIZZLE_XYZW
;
166 v
->emit(v
->MOV(m1
, src
));
168 register_coalesce(v
);
170 EXPECT_NE(mul
->dst
.file
, MRF
);
173 TEST_F(register_coalesce_test
, test_dp4_mrf
)
175 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
176 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
179 dst_reg m0
= dst_reg(MRF
, 0);
180 m0
.writemask
= WRITEMASK_Y
;
181 m0
.type
= BRW_REGISTER_TYPE_F
;
183 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
185 vec4_instruction
*dp4
= v
->emit(v
->DP4(temp
, some_src_1
, some_src_2
));
186 v
->emit(v
->MOV(m0
, src_reg(temp
)));
188 register_coalesce(v
);
190 EXPECT_EQ(dp4
->dst
.file
, MRF
);
191 EXPECT_EQ(dp4
->dst
.writemask
, WRITEMASK_Y
);
194 TEST_F(register_coalesce_test
, test_dp4_grf
)
196 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
197 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
200 dst_reg to
= dst_reg(v
, glsl_type::vec4_type
);
201 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
203 vec4_instruction
*dp4
= v
->emit(v
->DP4(temp
, some_src_1
, some_src_2
));
204 to
.writemask
= WRITEMASK_Y
;
205 v
->emit(v
->MOV(to
, src_reg(temp
)));
207 /* if we don't do something with the result, the automatic dead code
208 * elimination will remove all our instructions.
210 src_reg src
= src_reg(to
);
212 v
->emit(v
->MOV(dst_reg(MRF
, 0), src
));
214 register_coalesce(v
);
216 EXPECT_EQ(dp4
->dst
.nr
, to
.nr
);
217 EXPECT_EQ(dp4
->dst
.writemask
, WRITEMASK_Y
);
220 TEST_F(register_coalesce_test
, test_channel_mul_grf
)
222 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
223 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
226 dst_reg to
= dst_reg(v
, glsl_type::vec4_type
);
227 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
229 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, some_src_1
, some_src_2
));
230 to
.writemask
= WRITEMASK_Y
;
231 v
->emit(v
->MOV(to
, src_reg(temp
)));
233 /* if we don't do something with the result, the automatic dead code
234 * elimination will remove all our instructions.
236 src_reg src
= src_reg(to
);
238 v
->emit(v
->MOV(dst_reg(MRF
, 0), src
));
240 register_coalesce(v
);
242 EXPECT_EQ(mul
->dst
.nr
, to
.nr
);