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>
26 #include "program/program.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 gen_device_info
*devinfo
;
40 struct gl_context
*ctx
;
41 struct gl_shader_program
*shader_prog
;
42 struct brw_vue_prog_data
*prog_data
;
47 class register_coalesce_vec4_visitor
: public vec4_visitor
50 register_coalesce_vec4_visitor(struct brw_compiler
*compiler
,
52 struct brw_vue_prog_data
*prog_data
)
53 : vec4_visitor(compiler
, NULL
, NULL
, prog_data
, shader
, NULL
,
54 false /* no_spills */, -1)
56 prog_data
->dispatch_mode
= DISPATCH_MODE_4X2_DUAL_OBJECT
;
60 virtual dst_reg
*make_reg_for_system_value(int /* location */)
62 unreachable("Not reached");
65 virtual void setup_payload()
67 unreachable("Not reached");
70 virtual void emit_prolog()
72 unreachable("Not reached");
75 virtual void emit_thread_end()
77 unreachable("Not reached");
80 virtual void emit_urb_write_header(int /* mrf */)
82 unreachable("Not reached");
85 virtual vec4_instruction
*emit_urb_write_opcode(bool /* complete */)
87 unreachable("Not reached");
92 void register_coalesce_test::SetUp()
94 ctx
= (struct gl_context
*)calloc(1, sizeof(*ctx
));
95 compiler
= (struct brw_compiler
*)calloc(1, sizeof(*compiler
));
96 devinfo
= (struct gen_device_info
*)calloc(1, sizeof(*devinfo
));
97 prog_data
= (struct brw_vue_prog_data
*)calloc(1, sizeof(*prog_data
));
98 compiler
->devinfo
= devinfo
;
101 nir_shader_create(NULL
, MESA_SHADER_VERTEX
, NULL
, NULL
);
103 v
= new register_coalesce_vec4_visitor(compiler
, shader
, prog_data
);
109 _register_coalesce(vec4_visitor
*v
, const char *func
)
114 printf("%s: instructions before:\n", func
);
115 v
->dump_instructions();
119 v
->opt_register_coalesce();
122 printf("%s: instructions after:\n", func
);
123 v
->dump_instructions();
127 TEST_F(register_coalesce_test
, test_compute_to_mrf
)
129 src_reg something
= src_reg(v
, glsl_type::float_type
);
130 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
133 dst_reg m0
= dst_reg(MRF
, 0);
134 m0
.writemask
= WRITEMASK_X
;
135 m0
.type
= BRW_REGISTER_TYPE_F
;
137 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, something
, brw_imm_f(1.0f
)));
138 v
->emit(v
->MOV(m0
, src_reg(temp
)));
140 register_coalesce(v
);
142 EXPECT_EQ(mul
->dst
.file
, MRF
);
146 TEST_F(register_coalesce_test
, test_multiple_use
)
148 src_reg something
= src_reg(v
, glsl_type::float_type
);
149 dst_reg temp
= dst_reg(v
, glsl_type::vec4_type
);
152 dst_reg m0
= dst_reg(MRF
, 0);
153 m0
.writemask
= WRITEMASK_X
;
154 m0
.type
= BRW_REGISTER_TYPE_F
;
156 dst_reg m1
= dst_reg(MRF
, 1);
157 m1
.writemask
= WRITEMASK_XYZW
;
158 m1
.type
= BRW_REGISTER_TYPE_F
;
160 src_reg src
= src_reg(temp
);
161 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, something
, brw_imm_f(1.0f
)));
162 src
.swizzle
= BRW_SWIZZLE_XXXX
;
163 v
->emit(v
->MOV(m0
, src
));
164 src
.swizzle
= BRW_SWIZZLE_XYZW
;
165 v
->emit(v
->MOV(m1
, src
));
167 register_coalesce(v
);
169 EXPECT_NE(mul
->dst
.file
, MRF
);
172 TEST_F(register_coalesce_test
, test_dp4_mrf
)
174 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
175 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
178 dst_reg m0
= dst_reg(MRF
, 0);
179 m0
.writemask
= WRITEMASK_Y
;
180 m0
.type
= BRW_REGISTER_TYPE_F
;
182 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
184 vec4_instruction
*dp4
= v
->emit(v
->DP4(temp
, some_src_1
, some_src_2
));
185 v
->emit(v
->MOV(m0
, src_reg(temp
)));
187 register_coalesce(v
);
189 EXPECT_EQ(dp4
->dst
.file
, MRF
);
190 EXPECT_EQ(dp4
->dst
.writemask
, WRITEMASK_Y
);
193 TEST_F(register_coalesce_test
, test_dp4_grf
)
195 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
196 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
199 dst_reg to
= dst_reg(v
, glsl_type::vec4_type
);
200 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
202 vec4_instruction
*dp4
= v
->emit(v
->DP4(temp
, some_src_1
, some_src_2
));
203 to
.writemask
= WRITEMASK_Y
;
204 v
->emit(v
->MOV(to
, src_reg(temp
)));
206 /* if we don't do something with the result, the automatic dead code
207 * elimination will remove all our instructions.
209 src_reg src
= src_reg(to
);
211 v
->emit(v
->MOV(dst_reg(MRF
, 0), src
));
213 register_coalesce(v
);
215 EXPECT_EQ(dp4
->dst
.nr
, to
.nr
);
216 EXPECT_EQ(dp4
->dst
.writemask
, WRITEMASK_Y
);
219 TEST_F(register_coalesce_test
, test_channel_mul_grf
)
221 src_reg some_src_1
= src_reg(v
, glsl_type::vec4_type
);
222 src_reg some_src_2
= src_reg(v
, glsl_type::vec4_type
);
225 dst_reg to
= dst_reg(v
, glsl_type::vec4_type
);
226 dst_reg temp
= dst_reg(v
, glsl_type::float_type
);
228 vec4_instruction
*mul
= v
->emit(v
->MUL(temp
, some_src_1
, some_src_2
));
229 to
.writemask
= WRITEMASK_Y
;
230 v
->emit(v
->MOV(to
, src_reg(temp
)));
232 /* if we don't do something with the result, the automatic dead code
233 * elimination will remove all our instructions.
235 src_reg src
= src_reg(to
);
237 v
->emit(v
->MOV(dst_reg(MRF
, 0), src
));
239 register_coalesce(v
);
241 EXPECT_EQ(mul
->dst
.nr
, to
.nr
);