Missing copyright for mem-stats header files.
[official-gcc.git] / gcc / hsa-gen.c
blob768c2cf8f7853116fcd41c678f608f7c548af6cd
1 /* A pass for lowering gimple to HSAIL
2 Copyright (C) 2013-2016 Free Software Foundation, Inc.
3 Contributed by Martin Jambor <mjambor@suse.cz> and
4 Martin Liska <mliska@suse.cz>.
6 This file is part of GCC.
8 GCC is free software; you can redistribute it and/or modify
9 it under the terms of the GNU General Public License as published by
10 the Free Software Foundation; either version 3, or (at your option)
11 any later version.
13 GCC is distributed in the hope that it will be useful,
14 but WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
16 GNU General Public License for more details.
18 You should have received a copy of the GNU General Public License
19 along with GCC; see the file COPYING3. If not see
20 <http://www.gnu.org/licenses/>. */
22 #include "config.h"
23 #include "system.h"
24 #include "coretypes.h"
25 #include "tm.h"
26 #include "is-a.h"
27 #include "hash-table.h"
28 #include "vec.h"
29 #include "tree.h"
30 #include "tree-pass.h"
31 #include "cfg.h"
32 #include "function.h"
33 #include "basic-block.h"
34 #include "fold-const.h"
35 #include "gimple.h"
36 #include "gimple-iterator.h"
37 #include "bitmap.h"
38 #include "dumpfile.h"
39 #include "gimple-pretty-print.h"
40 #include "diagnostic-core.h"
41 #include "alloc-pool.h"
42 #include "gimple-ssa.h"
43 #include "tree-phinodes.h"
44 #include "stringpool.h"
45 #include "tree-ssanames.h"
46 #include "tree-dfa.h"
47 #include "ssa-iterators.h"
48 #include "cgraph.h"
49 #include "print-tree.h"
50 #include "symbol-summary.h"
51 #include "hsa.h"
52 #include "cfghooks.h"
53 #include "tree-cfg.h"
54 #include "cfgloop.h"
55 #include "cfganal.h"
56 #include "builtins.h"
57 #include "params.h"
58 #include "gomp-constants.h"
59 #include "internal-fn.h"
60 #include "builtins.h"
61 #include "stor-layout.h"
63 /* Print a warning message and set that we have seen an error. */
65 #define HSA_SORRY_ATV(location, message, ...) \
66 do \
67 { \
68 hsa_fail_cfun (); \
69 if (warning_at (EXPR_LOCATION (hsa_cfun->m_decl), OPT_Whsa, \
70 HSA_SORRY_MSG)) \
71 inform (location, message, __VA_ARGS__); \
72 } \
73 while (false);
75 /* Same as previous, but highlight a location. */
77 #define HSA_SORRY_AT(location, message) \
78 do \
79 { \
80 hsa_fail_cfun (); \
81 if (warning_at (EXPR_LOCATION (hsa_cfun->m_decl), OPT_Whsa, \
82 HSA_SORRY_MSG)) \
83 inform (location, message); \
84 } \
85 while (false);
87 /* Default number of threads used by kernel dispatch. */
89 #define HSA_DEFAULT_NUM_THREADS 64
91 /* Following structures are defined in the final version
92 of HSA specification. */
94 /* HSA queue packet is shadow structure, originally provided by AMD. */
96 struct hsa_queue_packet
98 uint16_t header;
99 uint16_t setup;
100 uint16_t workgroup_size_x;
101 uint16_t workgroup_size_y;
102 uint16_t workgroup_size_z;
103 uint16_t reserved0;
104 uint32_t grid_size_x;
105 uint32_t grid_size_y;
106 uint32_t grid_size_z;
107 uint32_t private_segment_size;
108 uint32_t group_segment_size;
109 uint64_t kernel_object;
110 void *kernarg_address;
111 uint64_t reserved2;
112 uint64_t completion_signal;
115 /* HSA queue is shadow structure, originally provided by AMD. */
117 struct hsa_queue
119 int type;
120 uint32_t features;
121 void *base_address;
122 uint64_t doorbell_signal;
123 uint32_t size;
124 uint32_t reserved1;
125 uint64_t id;
128 /* Alloc pools for allocating basic hsa structures such as operands,
129 instructions and other basic entities. */
130 static object_allocator<hsa_op_address> *hsa_allocp_operand_address;
131 static object_allocator<hsa_op_immed> *hsa_allocp_operand_immed;
132 static object_allocator<hsa_op_reg> *hsa_allocp_operand_reg;
133 static object_allocator<hsa_op_code_list> *hsa_allocp_operand_code_list;
134 static object_allocator<hsa_op_operand_list> *hsa_allocp_operand_operand_list;
135 static object_allocator<hsa_insn_basic> *hsa_allocp_inst_basic;
136 static object_allocator<hsa_insn_phi> *hsa_allocp_inst_phi;
137 static object_allocator<hsa_insn_mem> *hsa_allocp_inst_mem;
138 static object_allocator<hsa_insn_atomic> *hsa_allocp_inst_atomic;
139 static object_allocator<hsa_insn_signal> *hsa_allocp_inst_signal;
140 static object_allocator<hsa_insn_seg> *hsa_allocp_inst_seg;
141 static object_allocator<hsa_insn_cmp> *hsa_allocp_inst_cmp;
142 static object_allocator<hsa_insn_br> *hsa_allocp_inst_br;
143 static object_allocator<hsa_insn_sbr> *hsa_allocp_inst_sbr;
144 static object_allocator<hsa_insn_call> *hsa_allocp_inst_call;
145 static object_allocator<hsa_insn_arg_block> *hsa_allocp_inst_arg_block;
146 static object_allocator<hsa_insn_comment> *hsa_allocp_inst_comment;
147 static object_allocator<hsa_insn_queue> *hsa_allocp_inst_queue;
148 static object_allocator<hsa_insn_srctype> *hsa_allocp_inst_srctype;
149 static object_allocator<hsa_insn_packed> *hsa_allocp_inst_packed;
150 static object_allocator<hsa_insn_cvt> *hsa_allocp_inst_cvt;
151 static object_allocator<hsa_insn_alloca> *hsa_allocp_inst_alloca;
152 static object_allocator<hsa_bb> *hsa_allocp_bb;
154 /* List of pointers to all instructions that come from an object allocator. */
155 static vec <hsa_insn_basic *> hsa_instructions;
157 /* List of pointers to all operands that come from an object allocator. */
158 static vec <hsa_op_base *> hsa_operands;
160 hsa_symbol::hsa_symbol ()
161 : m_decl (NULL_TREE), m_name (NULL), m_name_number (0),
162 m_directive_offset (0), m_type (BRIG_TYPE_NONE),
163 m_segment (BRIG_SEGMENT_NONE), m_linkage (BRIG_LINKAGE_NONE), m_dim (0),
164 m_cst_value (NULL), m_global_scope_p (false), m_seen_error (false),
165 m_allocation (BRIG_ALLOCATION_AUTOMATIC)
170 hsa_symbol::hsa_symbol (BrigType16_t type, BrigSegment8_t segment,
171 BrigLinkage8_t linkage, bool global_scope_p,
172 BrigAllocation allocation)
173 : m_decl (NULL_TREE), m_name (NULL), m_name_number (0),
174 m_directive_offset (0), m_type (type), m_segment (segment),
175 m_linkage (linkage), m_dim (0), m_cst_value (NULL),
176 m_global_scope_p (global_scope_p), m_seen_error (false),
177 m_allocation (allocation)
181 unsigned HOST_WIDE_INT
182 hsa_symbol::total_byte_size ()
184 unsigned HOST_WIDE_INT s
185 = hsa_type_bit_size (~BRIG_TYPE_ARRAY_MASK & m_type);
186 gcc_assert (s % BITS_PER_UNIT == 0);
187 s /= BITS_PER_UNIT;
189 if (m_dim)
190 s *= m_dim;
192 return s;
195 /* Forward declaration. */
197 static BrigType16_t
198 hsa_type_for_tree_type (const_tree type, unsigned HOST_WIDE_INT *dim_p,
199 bool min32int);
201 void
202 hsa_symbol::fillup_for_decl (tree decl)
204 m_decl = decl;
205 m_type = hsa_type_for_tree_type (TREE_TYPE (decl), &m_dim, false);
207 if (hsa_seen_error ())
208 m_seen_error = true;
211 /* Constructor of class representing global HSA function/kernel information and
212 state. FNDECL is function declaration, KERNEL_P is true if the function
213 is going to become a HSA kernel. If the function has body, SSA_NAMES_COUNT
214 should be set to number of SSA names used in the function. */
216 hsa_function_representation::hsa_function_representation
217 (tree fdecl, bool kernel_p, unsigned ssa_names_count)
218 : m_name (NULL),
219 m_reg_count (0), m_input_args (vNULL),
220 m_output_arg (NULL), m_spill_symbols (vNULL), m_global_symbols (vNULL),
221 m_private_variables (vNULL), m_called_functions (vNULL),
222 m_called_internal_fns (vNULL), m_hbb_count (0),
223 m_in_ssa (true), m_kern_p (kernel_p), m_declaration_p (false),
224 m_decl (fdecl), m_internal_fn (NULL), m_shadow_reg (NULL),
225 m_kernel_dispatch_count (0), m_maximum_omp_data_size (0),
226 m_seen_error (false), m_temp_symbol_count (0), m_ssa_map ()
228 int sym_init_len = (vec_safe_length (cfun->local_decls) / 2) + 1;;
229 m_local_symbols = new hash_table <hsa_noop_symbol_hasher> (sym_init_len);
230 m_ssa_map.safe_grow_cleared (ssa_names_count);
233 /* Constructor of class representing HSA function information that
234 is derived for an internal function. */
235 hsa_function_representation::hsa_function_representation (hsa_internal_fn *fn)
236 : m_reg_count (0), m_input_args (vNULL),
237 m_output_arg (NULL), m_local_symbols (NULL),
238 m_spill_symbols (vNULL), m_global_symbols (vNULL),
239 m_private_variables (vNULL), m_called_functions (vNULL),
240 m_called_internal_fns (vNULL), m_hbb_count (0),
241 m_in_ssa (true), m_kern_p (false), m_declaration_p (true), m_decl (NULL),
242 m_internal_fn (fn), m_shadow_reg (NULL), m_kernel_dispatch_count (0),
243 m_maximum_omp_data_size (0), m_seen_error (false), m_temp_symbol_count (0),
244 m_ssa_map () {}
246 /* Destructor of class holding function/kernel-wide information and state. */
248 hsa_function_representation::~hsa_function_representation ()
250 /* Kernel names are deallocated at the end of BRIG output when deallocating
251 hsa_decl_kernel_mapping. */
252 if (!m_kern_p || m_seen_error)
253 free (m_name);
255 for (unsigned i = 0; i < m_input_args.length (); i++)
256 delete m_input_args[i];
257 m_input_args.release ();
259 delete m_output_arg;
260 delete m_local_symbols;
262 for (unsigned i = 0; i < m_spill_symbols.length (); i++)
263 delete m_spill_symbols[i];
264 m_spill_symbols.release ();
266 hsa_symbol *sym;
267 for (unsigned i = 0; i < m_global_symbols.iterate (i, &sym); i++)
268 if (sym->m_linkage != BRIG_ALLOCATION_PROGRAM)
269 delete sym;
270 m_global_symbols.release ();
272 for (unsigned i = 0; i < m_private_variables.length (); i++)
273 delete m_private_variables[i];
274 m_private_variables.release ();
275 m_called_functions.release ();
276 m_ssa_map.release ();
278 for (unsigned i = 0; i < m_called_internal_fns.length (); i++)
279 delete m_called_internal_fns[i];
282 hsa_op_reg *
283 hsa_function_representation::get_shadow_reg ()
285 /* If we compile a function with kernel dispatch and does not set
286 an optimization level, the function won't be inlined and
287 we return NULL. */
288 if (!m_kern_p)
289 return NULL;
291 if (m_shadow_reg)
292 return m_shadow_reg;
294 /* Append the shadow argument. */
295 hsa_symbol *shadow = new hsa_symbol (BRIG_TYPE_U64, BRIG_SEGMENT_KERNARG,
296 BRIG_LINKAGE_FUNCTION);
297 m_input_args.safe_push (shadow);
298 shadow->m_name = "hsa_runtime_shadow";
300 hsa_op_reg *r = new hsa_op_reg (BRIG_TYPE_U64);
301 hsa_op_address *addr = new hsa_op_address (shadow);
303 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, BRIG_TYPE_U64, r, addr);
304 hsa_bb_for_bb (ENTRY_BLOCK_PTR_FOR_FN (cfun))->append_insn (mem);
305 m_shadow_reg = r;
307 return r;
310 bool hsa_function_representation::has_shadow_reg_p ()
312 return m_shadow_reg != NULL;
315 void
316 hsa_function_representation::init_extra_bbs ()
318 hsa_init_new_bb (ENTRY_BLOCK_PTR_FOR_FN (cfun));
319 hsa_init_new_bb (EXIT_BLOCK_PTR_FOR_FN (cfun));
322 hsa_symbol *
323 hsa_function_representation::create_hsa_temporary (BrigType16_t type)
325 hsa_symbol *s = new hsa_symbol (type, BRIG_SEGMENT_PRIVATE,
326 BRIG_LINKAGE_FUNCTION);
327 s->m_name_number = m_temp_symbol_count++;
329 hsa_cfun->m_private_variables.safe_push (s);
330 return s;
333 BrigLinkage8_t
334 hsa_function_representation::get_linkage ()
336 if (m_internal_fn)
337 return BRIG_LINKAGE_PROGRAM;
339 return m_kern_p || TREE_PUBLIC (m_decl) ?
340 BRIG_LINKAGE_PROGRAM : BRIG_LINKAGE_MODULE;
343 /* Hash map of simple OMP builtins. */
344 static hash_map <nofree_string_hash, omp_simple_builtin> *omp_simple_builtins
345 = NULL;
347 /* Warning messages for OMP builtins. */
349 #define HSA_WARN_LOCK_ROUTINE "support for HSA does not implement OpenMP " \
350 "lock routines"
351 #define HSA_WARN_TIMING_ROUTINE "support for HSA does not implement OpenMP " \
352 "timing routines"
353 #define HSA_WARN_MEMORY_ROUTINE "OpenMP device memory library routines have " \
354 "undefined semantics within target regions, support for HSA ignores them"
355 #define HSA_WARN_AFFINITY "Support for HSA does not implement OpenMP " \
356 "affinity feateres"
358 /* Initialize hash map with simple OMP builtins. */
360 static void
361 hsa_init_simple_builtins ()
363 if (omp_simple_builtins != NULL)
364 return;
366 omp_simple_builtins
367 = new hash_map <nofree_string_hash, omp_simple_builtin> ();
369 omp_simple_builtin omp_builtins[] =
371 omp_simple_builtin ("omp_get_initial_device", NULL, false,
372 new hsa_op_immed (GOMP_DEVICE_HOST,
373 (BrigType16_t) BRIG_TYPE_S32)),
374 omp_simple_builtin ("omp_is_initial_device", NULL, false,
375 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
376 omp_simple_builtin ("omp_get_dynamic", NULL, false,
377 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
378 omp_simple_builtin ("omp_set_dynamic", NULL, false, NULL),
379 omp_simple_builtin ("omp_init_lock", HSA_WARN_LOCK_ROUTINE, true),
380 omp_simple_builtin ("omp_init_lock_with_hint", HSA_WARN_LOCK_ROUTINE,
381 true),
382 omp_simple_builtin ("omp_init_nest_lock_with_hint", HSA_WARN_LOCK_ROUTINE,
383 true),
384 omp_simple_builtin ("omp_destroy_lock", HSA_WARN_LOCK_ROUTINE, true),
385 omp_simple_builtin ("omp_set_lock", HSA_WARN_LOCK_ROUTINE, true),
386 omp_simple_builtin ("omp_unset_lock", HSA_WARN_LOCK_ROUTINE, true),
387 omp_simple_builtin ("omp_test_lock", HSA_WARN_LOCK_ROUTINE, true),
388 omp_simple_builtin ("omp_get_wtime", HSA_WARN_TIMING_ROUTINE, true),
389 omp_simple_builtin ("omp_get_wtick", HSA_WARN_TIMING_ROUTINE, true),
390 omp_simple_builtin ("omp_target_alloc", HSA_WARN_MEMORY_ROUTINE, false,
391 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_U64)),
392 omp_simple_builtin ("omp_target_free", HSA_WARN_MEMORY_ROUTINE, false),
393 omp_simple_builtin ("omp_target_is_present", HSA_WARN_MEMORY_ROUTINE,
394 false,
395 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
396 omp_simple_builtin ("omp_target_memcpy", HSA_WARN_MEMORY_ROUTINE, false,
397 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
398 omp_simple_builtin ("omp_target_memcpy_rect", HSA_WARN_MEMORY_ROUTINE,
399 false,
400 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
401 omp_simple_builtin ("omp_target_associate_ptr", HSA_WARN_MEMORY_ROUTINE,
402 false,
403 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
404 omp_simple_builtin ("omp_target_disassociate_ptr",
405 HSA_WARN_MEMORY_ROUTINE,
406 false,
407 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
408 omp_simple_builtin ("omp_set_max_active_levels",
409 "Support for HSA only allows only one active level, "
410 "call to omp_set_max_active_levels will be ignored "
411 "in the generated HSAIL",
412 false, NULL),
413 omp_simple_builtin ("omp_get_max_active_levels", NULL, false,
414 new hsa_op_immed (1, (BrigType16_t) BRIG_TYPE_S32)),
415 omp_simple_builtin ("omp_in_final", NULL, false,
416 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
417 omp_simple_builtin ("omp_get_proc_bind", HSA_WARN_AFFINITY, false,
418 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
419 omp_simple_builtin ("omp_get_num_places", HSA_WARN_AFFINITY, false,
420 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
421 omp_simple_builtin ("omp_get_place_num_procs", HSA_WARN_AFFINITY, false,
422 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
423 omp_simple_builtin ("omp_get_place_proc_ids", HSA_WARN_AFFINITY, false,
424 NULL),
425 omp_simple_builtin ("omp_get_place_num", HSA_WARN_AFFINITY, false,
426 new hsa_op_immed (-1, (BrigType16_t) BRIG_TYPE_S32)),
427 omp_simple_builtin ("omp_get_partition_num_places", HSA_WARN_AFFINITY,
428 false,
429 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
430 omp_simple_builtin ("omp_get_partition_place_nums", HSA_WARN_AFFINITY,
431 false, NULL),
432 omp_simple_builtin ("omp_set_default_device",
433 "omp_set_default_device has undefined semantics "
434 "within target regions, support for HSA ignores it",
435 false, NULL),
436 omp_simple_builtin ("omp_get_default_device",
437 "omp_get_default_device has undefined semantics "
438 "within target regions, support for HSA ignores it",
439 false,
440 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
441 omp_simple_builtin ("omp_get_num_devices",
442 "omp_get_num_devices has undefined semantics "
443 "within target regions, support for HSA ignores it",
444 false,
445 new hsa_op_immed (0, (BrigType16_t) BRIG_TYPE_S32)),
446 omp_simple_builtin ("omp_get_num_procs", NULL, true, NULL),
447 omp_simple_builtin ("omp_get_cancellation", NULL, true, NULL),
448 omp_simple_builtin ("omp_set_nested", NULL, true, NULL),
449 omp_simple_builtin ("omp_get_nested", NULL, true, NULL),
450 omp_simple_builtin ("omp_set_schedule", NULL, true, NULL),
451 omp_simple_builtin ("omp_get_schedule", NULL, true, NULL),
452 omp_simple_builtin ("omp_get_thread_limit", NULL, true, NULL),
453 omp_simple_builtin ("omp_get_team_size", NULL, true, NULL),
454 omp_simple_builtin ("omp_get_ancestor_thread_num", NULL, true, NULL),
455 omp_simple_builtin ("omp_get_max_task_priority", NULL, true, NULL)
458 unsigned count = sizeof (omp_builtins) / sizeof (omp_simple_builtin);
460 for (unsigned i = 0; i < count; i++)
461 omp_simple_builtins->put (omp_builtins[i].m_name, omp_builtins[i]);
464 /* Allocate HSA structures that we need only while generating with this. */
466 static void
467 hsa_init_data_for_cfun ()
469 hsa_init_compilation_unit_data ();
470 hsa_allocp_operand_address
471 = new object_allocator<hsa_op_address> ("HSA address operands");
472 hsa_allocp_operand_immed
473 = new object_allocator<hsa_op_immed> ("HSA immediate operands");
474 hsa_allocp_operand_reg
475 = new object_allocator<hsa_op_reg> ("HSA register operands");
476 hsa_allocp_operand_code_list
477 = new object_allocator<hsa_op_code_list> ("HSA code list operands");
478 hsa_allocp_operand_operand_list
479 = new object_allocator<hsa_op_operand_list> ("HSA operand list operands");
480 hsa_allocp_inst_basic
481 = new object_allocator<hsa_insn_basic> ("HSA basic instructions");
482 hsa_allocp_inst_phi
483 = new object_allocator<hsa_insn_phi> ("HSA phi operands");
484 hsa_allocp_inst_mem
485 = new object_allocator<hsa_insn_mem> ("HSA memory instructions");
486 hsa_allocp_inst_atomic
487 = new object_allocator<hsa_insn_atomic> ("HSA atomic instructions");
488 hsa_allocp_inst_signal
489 = new object_allocator<hsa_insn_signal> ("HSA signal instructions");
490 hsa_allocp_inst_seg
491 = new object_allocator<hsa_insn_seg> ("HSA segment conversion "
492 "instructions");
493 hsa_allocp_inst_cmp
494 = new object_allocator<hsa_insn_cmp> ("HSA comparison instructions");
495 hsa_allocp_inst_br
496 = new object_allocator<hsa_insn_br> ("HSA branching instructions");
497 hsa_allocp_inst_sbr
498 = new object_allocator<hsa_insn_sbr> ("HSA switch branching instructions");
499 hsa_allocp_inst_call
500 = new object_allocator<hsa_insn_call> ("HSA call instructions");
501 hsa_allocp_inst_arg_block
502 = new object_allocator<hsa_insn_arg_block> ("HSA arg block instructions");
503 hsa_allocp_inst_comment
504 = new object_allocator<hsa_insn_comment> ("HSA comment instructions");
505 hsa_allocp_inst_queue
506 = new object_allocator<hsa_insn_queue> ("HSA queue instructions");
507 hsa_allocp_inst_srctype
508 = new object_allocator<hsa_insn_srctype> ("HSA source type instructions");
509 hsa_allocp_inst_packed
510 = new object_allocator<hsa_insn_packed> ("HSA packed instructions");
511 hsa_allocp_inst_cvt
512 = new object_allocator<hsa_insn_cvt> ("HSA convert instructions");
513 hsa_allocp_inst_alloca
514 = new object_allocator<hsa_insn_alloca> ("HSA alloca instructions");
515 hsa_allocp_bb = new object_allocator<hsa_bb> ("HSA basic blocks");
518 /* Deinitialize HSA subsystem and free all allocated memory. */
520 static void
521 hsa_deinit_data_for_cfun (void)
523 basic_block bb;
525 FOR_ALL_BB_FN (bb, cfun)
526 if (bb->aux)
528 hsa_bb *hbb = hsa_bb_for_bb (bb);
529 hbb->~hsa_bb ();
530 bb->aux = NULL;
533 for (unsigned int i = 0; i < hsa_operands.length (); i++)
534 hsa_destroy_operand (hsa_operands[i]);
536 hsa_operands.release ();
538 for (unsigned i = 0; i < hsa_instructions.length (); i++)
539 hsa_destroy_insn (hsa_instructions[i]);
541 hsa_instructions.release ();
543 if (omp_simple_builtins != NULL)
545 delete omp_simple_builtins;
546 omp_simple_builtins = NULL;
549 delete hsa_allocp_operand_address;
550 delete hsa_allocp_operand_immed;
551 delete hsa_allocp_operand_reg;
552 delete hsa_allocp_operand_code_list;
553 delete hsa_allocp_operand_operand_list;
554 delete hsa_allocp_inst_basic;
555 delete hsa_allocp_inst_phi;
556 delete hsa_allocp_inst_atomic;
557 delete hsa_allocp_inst_mem;
558 delete hsa_allocp_inst_signal;
559 delete hsa_allocp_inst_seg;
560 delete hsa_allocp_inst_cmp;
561 delete hsa_allocp_inst_br;
562 delete hsa_allocp_inst_sbr;
563 delete hsa_allocp_inst_call;
564 delete hsa_allocp_inst_arg_block;
565 delete hsa_allocp_inst_comment;
566 delete hsa_allocp_inst_queue;
567 delete hsa_allocp_inst_srctype;
568 delete hsa_allocp_inst_packed;
569 delete hsa_allocp_inst_cvt;
570 delete hsa_allocp_inst_alloca;
571 delete hsa_allocp_bb;
572 delete hsa_cfun;
575 /* Return the type which holds addresses in the given SEGMENT. */
577 static BrigType16_t
578 hsa_get_segment_addr_type (BrigSegment8_t segment)
580 switch (segment)
582 case BRIG_SEGMENT_NONE:
583 gcc_unreachable ();
585 case BRIG_SEGMENT_FLAT:
586 case BRIG_SEGMENT_GLOBAL:
587 case BRIG_SEGMENT_READONLY:
588 case BRIG_SEGMENT_KERNARG:
589 return hsa_machine_large_p () ? BRIG_TYPE_U64 : BRIG_TYPE_U32;
591 case BRIG_SEGMENT_GROUP:
592 case BRIG_SEGMENT_PRIVATE:
593 case BRIG_SEGMENT_SPILL:
594 case BRIG_SEGMENT_ARG:
595 return BRIG_TYPE_U32;
597 gcc_unreachable ();
600 /* Return integer brig type according to provided SIZE in bytes. If SIGN
601 is set to true, return signed integer type. */
603 static BrigType16_t
604 get_integer_type_by_bytes (unsigned size, bool sign)
606 if (sign)
607 switch (size)
609 case 1:
610 return BRIG_TYPE_S8;
611 case 2:
612 return BRIG_TYPE_S16;
613 case 4:
614 return BRIG_TYPE_S32;
615 case 8:
616 return BRIG_TYPE_S64;
617 default:
618 break;
620 else
621 switch (size)
623 case 1:
624 return BRIG_TYPE_U8;
625 case 2:
626 return BRIG_TYPE_U16;
627 case 4:
628 return BRIG_TYPE_U32;
629 case 8:
630 return BRIG_TYPE_U64;
631 default:
632 break;
635 return 0;
638 /* Return HSA type for tree TYPE, which has to fit into BrigType16_t. Pointers
639 are assumed to use flat addressing. If min32int is true, always expand
640 integer types to one that has at least 32 bits. */
642 static BrigType16_t
643 hsa_type_for_scalar_tree_type (const_tree type, bool min32int)
645 HOST_WIDE_INT bsize;
646 const_tree base;
647 BrigType16_t res = BRIG_TYPE_NONE;
649 gcc_checking_assert (TYPE_P (type));
650 gcc_checking_assert (!AGGREGATE_TYPE_P (type));
651 if (POINTER_TYPE_P (type))
652 return hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT);
654 if (TREE_CODE (type) == VECTOR_TYPE || TREE_CODE (type) == COMPLEX_TYPE)
655 base = TREE_TYPE (type);
656 else
657 base = type;
659 if (!tree_fits_uhwi_p (TYPE_SIZE (base)))
661 HSA_SORRY_ATV (EXPR_LOCATION (type),
662 "support for HSA does not implement huge or "
663 "variable-sized type %T", type);
664 return res;
667 bsize = tree_to_uhwi (TYPE_SIZE (base));
668 unsigned byte_size = bsize / BITS_PER_UNIT;
669 if (INTEGRAL_TYPE_P (base))
670 res = get_integer_type_by_bytes (byte_size, !TYPE_UNSIGNED (base));
671 else if (SCALAR_FLOAT_TYPE_P (base))
673 switch (bsize)
675 case 16:
676 res = BRIG_TYPE_F16;
677 break;
678 case 32:
679 res = BRIG_TYPE_F32;
680 break;
681 case 64:
682 res = BRIG_TYPE_F64;
683 break;
684 default:
685 break;
689 if (res == BRIG_TYPE_NONE)
691 HSA_SORRY_ATV (EXPR_LOCATION (type),
692 "support for HSA does not implement type %T", type);
693 return res;
696 if (TREE_CODE (type) == VECTOR_TYPE)
698 HOST_WIDE_INT tsize = tree_to_uhwi (TYPE_SIZE (type));
700 if (bsize == tsize)
702 HSA_SORRY_ATV (EXPR_LOCATION (type),
703 "support for HSA does not implement a vector type "
704 "where a type and unit size are equal: %T", type);
705 return res;
708 switch (tsize)
710 case 32:
711 res |= BRIG_TYPE_PACK_32;
712 break;
713 case 64:
714 res |= BRIG_TYPE_PACK_64;
715 break;
716 case 128:
717 res |= BRIG_TYPE_PACK_128;
718 break;
719 default:
720 HSA_SORRY_ATV (EXPR_LOCATION (type),
721 "support for HSA does not implement type %T", type);
725 if (min32int)
727 /* Registers/immediate operands can only be 32bit or more except for
728 f16. */
729 if (res == BRIG_TYPE_U8 || res == BRIG_TYPE_U16)
730 res = BRIG_TYPE_U32;
731 else if (res == BRIG_TYPE_S8 || res == BRIG_TYPE_S16)
732 res = BRIG_TYPE_S32;
735 if (TREE_CODE (type) == COMPLEX_TYPE)
737 unsigned bsize = 2 * hsa_type_bit_size (res);
738 res = hsa_bittype_for_bitsize (bsize);
741 return res;
744 /* Returns the BRIG type we need to load/store entities of TYPE. */
746 static BrigType16_t
747 mem_type_for_type (BrigType16_t type)
749 /* HSA has non-intuitive constraints on load/store types. If it's
750 a bit-type it _must_ be B128, if it's not a bit-type it must be
751 64bit max. So for loading entities of 128 bits (e.g. vectors)
752 we have to to B128, while for loading the rest we have to use the
753 input type (??? or maybe also flattened to a equally sized non-vector
754 unsigned type?). */
755 if ((type & BRIG_TYPE_PACK_MASK) == BRIG_TYPE_PACK_128)
756 return BRIG_TYPE_B128;
757 else if (hsa_btype_p (type))
759 unsigned bitsize = hsa_type_bit_size (type);
760 if (bitsize < 128)
761 return hsa_uint_for_bitsize (bitsize);
763 return type;
766 /* Return HSA type for tree TYPE. If it cannot fit into BrigType16_t, some
767 kind of array will be generated, setting DIM appropriately. Otherwise, it
768 will be set to zero. */
770 static BrigType16_t
771 hsa_type_for_tree_type (const_tree type, unsigned HOST_WIDE_INT *dim_p = NULL,
772 bool min32int = false)
774 gcc_checking_assert (TYPE_P (type));
775 if (!tree_fits_uhwi_p (TYPE_SIZE_UNIT (type)))
777 HSA_SORRY_ATV (EXPR_LOCATION (type), "support for HSA does not "
778 "implement huge or variable-sized type %T", type);
779 return BRIG_TYPE_NONE;
782 if (RECORD_OR_UNION_TYPE_P (type))
784 if (dim_p)
785 *dim_p = tree_to_uhwi (TYPE_SIZE_UNIT (type));
786 return BRIG_TYPE_U8 | BRIG_TYPE_ARRAY;
789 if (TREE_CODE (type) == ARRAY_TYPE)
791 /* We try to be nice and use the real base-type when this is an array of
792 scalars and only resort to an array of bytes if the type is more
793 complex. */
795 unsigned HOST_WIDE_INT dim = 1;
797 while (TREE_CODE (type) == ARRAY_TYPE)
799 tree domain = TYPE_DOMAIN (type);
800 if (!TYPE_MIN_VALUE (domain)
801 || !TYPE_MAX_VALUE (domain)
802 || !tree_fits_shwi_p (TYPE_MIN_VALUE (domain))
803 || !tree_fits_shwi_p (TYPE_MAX_VALUE (domain)))
805 HSA_SORRY_ATV (EXPR_LOCATION (type),
806 "support for HSA does not implement array %T with "
807 "unknown bounds", type);
808 return BRIG_TYPE_NONE;
810 HOST_WIDE_INT min = tree_to_shwi (TYPE_MIN_VALUE (domain));
811 HOST_WIDE_INT max = tree_to_shwi (TYPE_MAX_VALUE (domain));
812 dim = dim * (unsigned HOST_WIDE_INT) (max - min + 1);
813 type = TREE_TYPE (type);
816 BrigType16_t res;
817 if (RECORD_OR_UNION_TYPE_P (type))
819 dim = dim * tree_to_uhwi (TYPE_SIZE_UNIT (type));
820 res = BRIG_TYPE_U8;
822 else
823 res = hsa_type_for_scalar_tree_type (type, false);
825 if (dim_p)
826 *dim_p = dim;
827 return res | BRIG_TYPE_ARRAY;
830 /* Scalar case: */
831 if (dim_p)
832 *dim_p = 0;
834 return hsa_type_for_scalar_tree_type (type, min32int);
837 /* Returns true if converting from STYPE into DTYPE needs the _CVT
838 opcode. If false a normal _MOV is enough. */
840 static bool
841 hsa_needs_cvt (BrigType16_t dtype, BrigType16_t stype)
843 if (hsa_btype_p (dtype))
844 return false;
846 /* float <-> int conversions are real converts. */
847 if (hsa_type_float_p (dtype) != hsa_type_float_p (stype))
848 return true;
849 /* When both types have different size, then we need CVT as well. */
850 if (hsa_type_bit_size (dtype) != hsa_type_bit_size (stype))
851 return true;
852 return false;
855 /* Lookup or create the associated hsa_symbol structure with a given VAR_DECL
856 or lookup the hsa_structure corresponding to a PARM_DECL. */
858 static hsa_symbol *
859 get_symbol_for_decl (tree decl)
861 hsa_symbol **slot;
862 hsa_symbol dummy (BRIG_TYPE_NONE, BRIG_SEGMENT_NONE, BRIG_LINKAGE_NONE);
864 gcc_assert (TREE_CODE (decl) == PARM_DECL
865 || TREE_CODE (decl) == RESULT_DECL
866 || TREE_CODE (decl) == VAR_DECL);
868 dummy.m_decl = decl;
870 bool is_in_global_vars
871 = TREE_CODE (decl) == VAR_DECL && is_global_var (decl);
873 if (is_in_global_vars)
874 slot = hsa_global_variable_symbols->find_slot (&dummy, INSERT);
875 else
876 slot = hsa_cfun->m_local_symbols->find_slot (&dummy, INSERT);
878 gcc_checking_assert (slot);
879 if (*slot)
881 /* If the symbol is problematic, mark current function also as
882 problematic. */
883 if ((*slot)->m_seen_error)
884 hsa_fail_cfun ();
886 return *slot;
888 else
890 hsa_symbol *sym;
891 gcc_assert (TREE_CODE (decl) == VAR_DECL);
893 if (is_in_global_vars)
895 sym = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_GLOBAL,
896 BRIG_LINKAGE_PROGRAM, true,
897 BRIG_ALLOCATION_PROGRAM);
898 hsa_cfun->m_global_symbols.safe_push (sym);
900 else
902 /* PARM_DECL and RESULT_DECL should be already in m_local_symbols. */
903 gcc_assert (TREE_CODE (decl) == VAR_DECL);
905 sym = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_PRIVATE,
906 BRIG_LINKAGE_FUNCTION);
907 hsa_cfun->m_private_variables.safe_push (sym);
910 sym->fillup_for_decl (decl);
911 sym->m_name = hsa_get_declaration_name (decl);
913 *slot = sym;
914 return sym;
918 /* For a given HSA function declaration, return a host
919 function declaration. */
921 tree
922 hsa_get_host_function (tree decl)
924 hsa_function_summary *s
925 = hsa_summaries->get (cgraph_node::get_create (decl));
926 gcc_assert (s->m_kind != HSA_NONE);
927 gcc_assert (s->m_gpu_implementation_p);
929 return s->m_binded_function->decl;
932 /* Return true if function DECL has a host equivalent function. */
934 static char *
935 get_brig_function_name (tree decl)
937 tree d = decl;
939 hsa_function_summary *s = hsa_summaries->get (cgraph_node::get_create (d));
940 if (s->m_kind != HSA_NONE && s->m_gpu_implementation_p)
941 d = s->m_binded_function->decl;
943 /* IPA split can create a function that has no host equivalent. */
944 if (d == NULL)
945 d = decl;
947 char *name = xstrdup (hsa_get_declaration_name (d));
948 hsa_sanitize_name (name);
950 return name;
953 /* Create a spill symbol of type TYPE. */
955 hsa_symbol *
956 hsa_get_spill_symbol (BrigType16_t type)
958 hsa_symbol *sym = new hsa_symbol (type, BRIG_SEGMENT_SPILL,
959 BRIG_LINKAGE_FUNCTION);
960 hsa_cfun->m_spill_symbols.safe_push (sym);
961 return sym;
964 /* Create a symbol for a read-only string constant. */
965 hsa_symbol *
966 hsa_get_string_cst_symbol (tree string_cst)
968 gcc_checking_assert (TREE_CODE (string_cst) == STRING_CST);
970 hsa_symbol **slot = hsa_cfun->m_string_constants_map.get (string_cst);
971 if (slot)
972 return *slot;
974 hsa_op_immed *cst = new hsa_op_immed (string_cst);
975 hsa_symbol *sym = new hsa_symbol (cst->m_type, BRIG_SEGMENT_GLOBAL,
976 BRIG_LINKAGE_MODULE, true,
977 BRIG_ALLOCATION_AGENT);
978 sym->m_cst_value = cst;
979 sym->m_dim = TREE_STRING_LENGTH (string_cst);
980 sym->m_name_number = hsa_cfun->m_global_symbols.length ();
982 hsa_cfun->m_global_symbols.safe_push (sym);
983 hsa_cfun->m_string_constants_map.put (string_cst, sym);
984 return sym;
987 /* Constructor of the ancestor of all operands. K is BRIG kind that identified
988 what the operator is. */
990 hsa_op_base::hsa_op_base (BrigKind16_t k)
991 : m_next (NULL), m_brig_op_offset (0), m_kind (k)
993 hsa_operands.safe_push (this);
996 /* Constructor of ancestor of all operands which have a type. K is BRIG kind
997 that identified what the operator is. T is the type of the operator. */
999 hsa_op_with_type::hsa_op_with_type (BrigKind16_t k, BrigType16_t t)
1000 : hsa_op_base (k), m_type (t)
1004 hsa_op_with_type *
1005 hsa_op_with_type::get_in_type (BrigType16_t dtype, hsa_bb *hbb)
1007 if (m_type == dtype)
1008 return this;
1010 hsa_op_reg *dest;
1012 if (hsa_needs_cvt (dtype, m_type))
1014 dest = new hsa_op_reg (dtype);
1015 hbb->append_insn (new hsa_insn_cvt (dest, this));
1017 else
1019 dest = new hsa_op_reg (m_type);
1020 hbb->append_insn (new hsa_insn_basic (2, BRIG_OPCODE_MOV,
1021 dest->m_type, dest, this));
1023 /* We cannot simply for instance: 'mov_u32 $_3, 48 (s32)' because
1024 type of the operand must be same as type of the instruction. */
1025 dest->m_type = dtype;
1028 return dest;
1031 /* Constructor of class representing HSA immediate values. TREE_VAL is the
1032 tree representation of the immediate value. If min32int is true,
1033 always expand integer types to one that has at least 32 bits. */
1035 hsa_op_immed::hsa_op_immed (tree tree_val, bool min32int)
1036 : hsa_op_with_type (BRIG_KIND_OPERAND_CONSTANT_BYTES,
1037 hsa_type_for_tree_type (TREE_TYPE (tree_val), NULL,
1038 min32int)),
1039 m_brig_repr (NULL)
1041 if (hsa_seen_error ())
1042 return;
1044 gcc_checking_assert ((is_gimple_min_invariant (tree_val)
1045 && (!POINTER_TYPE_P (TREE_TYPE (tree_val))
1046 || TREE_CODE (tree_val) == INTEGER_CST))
1047 || TREE_CODE (tree_val) == CONSTRUCTOR);
1048 m_tree_value = tree_val;
1049 m_brig_repr_size = hsa_get_imm_brig_type_len (m_type);
1051 if (TREE_CODE (m_tree_value) == STRING_CST)
1052 m_brig_repr_size = TREE_STRING_LENGTH (m_tree_value);
1053 else if (TREE_CODE (m_tree_value) == CONSTRUCTOR)
1055 m_brig_repr_size
1056 = tree_to_uhwi (TYPE_SIZE_UNIT (TREE_TYPE (m_tree_value)));
1058 /* Verify that all elements of a constructor are constants. */
1059 for (unsigned i = 0;
1060 i < vec_safe_length (CONSTRUCTOR_ELTS (m_tree_value)); i++)
1062 tree v = CONSTRUCTOR_ELT (m_tree_value, i)->value;
1063 if (!CONSTANT_CLASS_P (v))
1065 HSA_SORRY_AT (EXPR_LOCATION (tree_val),
1066 "HSA ctor should have only constants");
1067 return;
1072 emit_to_buffer (m_tree_value);
1075 /* Constructor of class representing HSA immediate values. INTEGER_VALUE is the
1076 integer representation of the immediate value. TYPE is BRIG type. */
1078 hsa_op_immed::hsa_op_immed (HOST_WIDE_INT integer_value, BrigType16_t type)
1079 : hsa_op_with_type (BRIG_KIND_OPERAND_CONSTANT_BYTES, type),
1080 m_tree_value (NULL), m_brig_repr (NULL)
1082 gcc_assert (hsa_type_integer_p (type));
1083 m_int_value = integer_value;
1084 m_brig_repr_size = hsa_type_bit_size (type) / BITS_PER_UNIT;
1086 hsa_bytes bytes;
1088 switch (m_brig_repr_size)
1090 case 1:
1091 bytes.b8 = (uint8_t) m_int_value;
1092 break;
1093 case 2:
1094 bytes.b16 = (uint16_t) m_int_value;
1095 break;
1096 case 4:
1097 bytes.b32 = (uint32_t) m_int_value;
1098 break;
1099 case 8:
1100 bytes.b64 = (uint64_t) m_int_value;
1101 break;
1102 default:
1103 gcc_unreachable ();
1106 m_brig_repr = XNEWVEC (char, m_brig_repr_size);
1107 memcpy (m_brig_repr, &bytes, m_brig_repr_size);
1110 hsa_op_immed::hsa_op_immed ()
1111 : hsa_op_with_type (BRIG_KIND_NONE, BRIG_TYPE_NONE), m_brig_repr (NULL)
1115 /* New operator to allocate immediate operands from pool alloc. */
1117 void *
1118 hsa_op_immed::operator new (size_t)
1120 return hsa_allocp_operand_immed->allocate_raw ();
1123 /* Destructor. */
1125 hsa_op_immed::~hsa_op_immed ()
1127 free (m_brig_repr);
1130 /* Change type of the immediate value to T. */
1132 void
1133 hsa_op_immed::set_type (BrigType16_t t)
1135 m_type = t;
1138 /* Constructor of class representing HSA registers and pseudo-registers. T is
1139 the BRIG type of the new register. */
1141 hsa_op_reg::hsa_op_reg (BrigType16_t t)
1142 : hsa_op_with_type (BRIG_KIND_OPERAND_REGISTER, t), m_gimple_ssa (NULL_TREE),
1143 m_def_insn (NULL), m_spill_sym (NULL), m_order (hsa_cfun->m_reg_count++),
1144 m_lr_begin (0), m_lr_end (0), m_reg_class (0), m_hard_num (0)
1148 /* New operator to allocate a register from pool alloc. */
1150 void *
1151 hsa_op_reg::operator new (size_t)
1153 return hsa_allocp_operand_reg->allocate_raw ();
1156 /* Verify register operand. */
1158 void
1159 hsa_op_reg::verify_ssa ()
1161 /* Verify that each HSA register has a definition assigned.
1162 Exceptions are VAR_DECL and PARM_DECL that are a default
1163 definition. */
1164 gcc_checking_assert (m_def_insn
1165 || (m_gimple_ssa != NULL
1166 && (!SSA_NAME_VAR (m_gimple_ssa)
1167 || (TREE_CODE (SSA_NAME_VAR (m_gimple_ssa))
1168 != PARM_DECL))
1169 && SSA_NAME_IS_DEFAULT_DEF (m_gimple_ssa)));
1171 /* Verify that every use of the register is really present
1172 in an instruction. */
1173 for (unsigned i = 0; i < m_uses.length (); i++)
1175 hsa_insn_basic *use = m_uses[i];
1177 bool is_visited = false;
1178 for (unsigned j = 0; j < use->operand_count (); j++)
1180 hsa_op_base *u = use->get_op (j);
1181 hsa_op_address *addr; addr = dyn_cast <hsa_op_address *> (u);
1182 if (addr && addr->m_reg)
1183 u = addr->m_reg;
1185 if (u == this)
1187 bool r = !addr && use->op_output_p (j);
1189 if (r)
1191 error ("HSA SSA name defined by instruction that is supposed "
1192 "to be using it");
1193 debug_hsa_operand (this);
1194 debug_hsa_insn (use);
1195 internal_error ("HSA SSA verification failed");
1198 is_visited = true;
1202 if (!is_visited)
1204 error ("HSA SSA name not among operands of instruction that is "
1205 "supposed to use it");
1206 debug_hsa_operand (this);
1207 debug_hsa_insn (use);
1208 internal_error ("HSA SSA verification failed");
1213 hsa_op_address::hsa_op_address (hsa_symbol *sym, hsa_op_reg *r,
1214 HOST_WIDE_INT offset)
1215 : hsa_op_base (BRIG_KIND_OPERAND_ADDRESS), m_symbol (sym), m_reg (r),
1216 m_imm_offset (offset)
1220 hsa_op_address::hsa_op_address (hsa_symbol *sym, HOST_WIDE_INT offset)
1221 : hsa_op_base (BRIG_KIND_OPERAND_ADDRESS), m_symbol (sym), m_reg (NULL),
1222 m_imm_offset (offset)
1226 hsa_op_address::hsa_op_address (hsa_op_reg *r, HOST_WIDE_INT offset)
1227 : hsa_op_base (BRIG_KIND_OPERAND_ADDRESS), m_symbol (NULL), m_reg (r),
1228 m_imm_offset (offset)
1232 /* New operator to allocate address operands from pool alloc. */
1234 void *
1235 hsa_op_address::operator new (size_t)
1237 return hsa_allocp_operand_address->allocate_raw ();
1240 /* Constructor of an operand referring to HSAIL code. */
1242 hsa_op_code_ref::hsa_op_code_ref () : hsa_op_base (BRIG_KIND_OPERAND_CODE_REF),
1243 m_directive_offset (0)
1247 /* Constructor of an operand representing a code list. Set it up so that it
1248 can contain ELEMENTS number of elements. */
1250 hsa_op_code_list::hsa_op_code_list (unsigned elements)
1251 : hsa_op_base (BRIG_KIND_OPERAND_CODE_LIST)
1253 m_offsets.create (1);
1254 m_offsets.safe_grow_cleared (elements);
1257 /* New operator to allocate code list operands from pool alloc. */
1259 void *
1260 hsa_op_code_list::operator new (size_t)
1262 return hsa_allocp_operand_code_list->allocate_raw ();
1265 /* Constructor of an operand representing an operand list.
1266 Set it up so that it can contain ELEMENTS number of elements. */
1268 hsa_op_operand_list::hsa_op_operand_list (unsigned elements)
1269 : hsa_op_base (BRIG_KIND_OPERAND_OPERAND_LIST)
1271 m_offsets.create (elements);
1272 m_offsets.safe_grow (elements);
1275 /* New operator to allocate operand list operands from pool alloc. */
1277 void *
1278 hsa_op_operand_list::operator new (size_t)
1280 return hsa_allocp_operand_operand_list->allocate_raw ();
1283 hsa_op_operand_list::~hsa_op_operand_list ()
1285 m_offsets.release ();
1289 hsa_op_reg *
1290 hsa_function_representation::reg_for_gimple_ssa (tree ssa)
1292 hsa_op_reg *hreg;
1294 gcc_checking_assert (TREE_CODE (ssa) == SSA_NAME);
1295 if (m_ssa_map[SSA_NAME_VERSION (ssa)])
1296 return m_ssa_map[SSA_NAME_VERSION (ssa)];
1298 hreg = new hsa_op_reg (hsa_type_for_scalar_tree_type (TREE_TYPE (ssa),
1299 true));
1300 hreg->m_gimple_ssa = ssa;
1301 m_ssa_map[SSA_NAME_VERSION (ssa)] = hreg;
1303 return hreg;
1306 void
1307 hsa_op_reg::set_definition (hsa_insn_basic *insn)
1309 if (hsa_cfun->m_in_ssa)
1311 gcc_checking_assert (!m_def_insn);
1312 m_def_insn = insn;
1314 else
1315 m_def_insn = NULL;
1318 /* Constructor of the class which is the bases of all instructions and directly
1319 represents the most basic ones. NOPS is the number of operands that the
1320 operand vector will contain (and which will be cleared). OP is the opcode
1321 of the instruction. This constructor does not set type. */
1323 hsa_insn_basic::hsa_insn_basic (unsigned nops, int opc)
1324 : m_prev (NULL),
1325 m_next (NULL), m_bb (NULL), m_opcode (opc), m_number (0),
1326 m_type (BRIG_TYPE_NONE), m_brig_offset (0)
1328 if (nops > 0)
1329 m_operands.safe_grow_cleared (nops);
1331 hsa_instructions.safe_push (this);
1334 /* Make OP the operand number INDEX of operands of this instruction. If OP is a
1335 register or an address containing a register, then either set the definition
1336 of the register to this instruction if it an output operand or add this
1337 instruction to the uses if it is an input one. */
1339 void
1340 hsa_insn_basic::set_op (int index, hsa_op_base *op)
1342 /* Each address operand is always use. */
1343 hsa_op_address *addr = dyn_cast <hsa_op_address *> (op);
1344 if (addr && addr->m_reg)
1345 addr->m_reg->m_uses.safe_push (this);
1346 else
1348 hsa_op_reg *reg = dyn_cast <hsa_op_reg *> (op);
1349 if (reg)
1351 if (op_output_p (index))
1352 reg->set_definition (this);
1353 else
1354 reg->m_uses.safe_push (this);
1358 m_operands[index] = op;
1361 /* Get INDEX-th operand of the instruction. */
1363 hsa_op_base *
1364 hsa_insn_basic::get_op (int index)
1366 return m_operands[index];
1369 /* Get address of INDEX-th operand of the instruction. */
1371 hsa_op_base **
1372 hsa_insn_basic::get_op_addr (int index)
1374 return &m_operands[index];
1377 /* Get number of operands of the instruction. */
1378 unsigned int
1379 hsa_insn_basic::operand_count ()
1381 return m_operands.length ();
1384 /* Constructor of the class which is the bases of all instructions and directly
1385 represents the most basic ones. NOPS is the number of operands that the
1386 operand vector will contain (and which will be cleared). OPC is the opcode
1387 of the instruction, T is the type of the instruction. */
1389 hsa_insn_basic::hsa_insn_basic (unsigned nops, int opc, BrigType16_t t,
1390 hsa_op_base *arg0, hsa_op_base *arg1,
1391 hsa_op_base *arg2, hsa_op_base *arg3)
1392 : m_prev (NULL), m_next (NULL), m_bb (NULL), m_opcode (opc),m_number (0),
1393 m_type (t), m_brig_offset (0)
1395 if (nops > 0)
1396 m_operands.safe_grow_cleared (nops);
1398 if (arg0 != NULL)
1400 gcc_checking_assert (nops >= 1);
1401 set_op (0, arg0);
1404 if (arg1 != NULL)
1406 gcc_checking_assert (nops >= 2);
1407 set_op (1, arg1);
1410 if (arg2 != NULL)
1412 gcc_checking_assert (nops >= 3);
1413 set_op (2, arg2);
1416 if (arg3 != NULL)
1418 gcc_checking_assert (nops >= 4);
1419 set_op (3, arg3);
1422 hsa_instructions.safe_push (this);
1425 /* New operator to allocate basic instruction from pool alloc. */
1427 void *
1428 hsa_insn_basic::operator new (size_t)
1430 return hsa_allocp_inst_basic->allocate_raw ();
1433 /* Verify the instruction. */
1435 void
1436 hsa_insn_basic::verify ()
1438 hsa_op_address *addr;
1439 hsa_op_reg *reg;
1441 /* Iterate all register operands and verify that the instruction
1442 is set in uses of the register. */
1443 for (unsigned i = 0; i < operand_count (); i++)
1445 hsa_op_base *use = get_op (i);
1447 if ((addr = dyn_cast <hsa_op_address *> (use)) && addr->m_reg)
1449 gcc_assert (addr->m_reg->m_def_insn != this);
1450 use = addr->m_reg;
1453 if ((reg = dyn_cast <hsa_op_reg *> (use)) && !op_output_p (i))
1455 unsigned j;
1456 for (j = 0; j < reg->m_uses.length (); j++)
1458 if (reg->m_uses[j] == this)
1459 break;
1462 if (j == reg->m_uses.length ())
1464 error ("HSA instruction uses a register but is not among "
1465 "recorded register uses");
1466 debug_hsa_operand (reg);
1467 debug_hsa_insn (this);
1468 internal_error ("HSA instruction verification failed");
1474 /* Constructor of an instruction representing a PHI node. NOPS is the number
1475 of operands (equal to the number of predecessors). */
1477 hsa_insn_phi::hsa_insn_phi (unsigned nops, hsa_op_reg *dst)
1478 : hsa_insn_basic (nops, HSA_OPCODE_PHI), m_dest (dst)
1480 dst->set_definition (this);
1483 /* New operator to allocate PHI instruction from pool alloc. */
1485 void *
1486 hsa_insn_phi::operator new (size_t)
1488 return hsa_allocp_inst_phi->allocate_raw ();
1491 /* Constructor of class representing instruction for conditional jump, CTRL is
1492 the control register determining whether the jump will be carried out, the
1493 new instruction is automatically added to its uses list. */
1495 hsa_insn_br::hsa_insn_br (hsa_op_reg *ctrl)
1496 : hsa_insn_basic (1, BRIG_OPCODE_CBR, BRIG_TYPE_B1, ctrl),
1497 m_width (BRIG_WIDTH_1)
1501 /* New operator to allocate branch instruction from pool alloc. */
1503 void *
1504 hsa_insn_br::operator new (size_t)
1506 return hsa_allocp_inst_br->allocate_raw ();
1509 /* Constructor of class representing instruction for switch jump, CTRL is
1510 the index register. */
1512 hsa_insn_sbr::hsa_insn_sbr (hsa_op_reg *index, unsigned jump_count)
1513 : hsa_insn_basic (1, BRIG_OPCODE_SBR, BRIG_TYPE_B1, index),
1514 m_width (BRIG_WIDTH_1), m_jump_table (vNULL), m_default_bb (NULL),
1515 m_label_code_list (new hsa_op_code_list (jump_count))
1519 /* New operator to allocate switch branch instruction from pool alloc. */
1521 void *
1522 hsa_insn_sbr::operator new (size_t)
1524 return hsa_allocp_inst_sbr->allocate_raw ();
1527 /* Replace all occurrences of OLD_BB with NEW_BB in the statements
1528 jump table. */
1530 void
1531 hsa_insn_sbr::replace_all_labels (basic_block old_bb, basic_block new_bb)
1533 for (unsigned i = 0; i < m_jump_table.length (); i++)
1534 if (m_jump_table[i] == old_bb)
1535 m_jump_table[i] = new_bb;
1538 hsa_insn_sbr::~hsa_insn_sbr ()
1540 m_jump_table.release ();
1543 /* Constructor of comparison instruction. CMP is the comparison operation and T
1544 is the result type. */
1546 hsa_insn_cmp::hsa_insn_cmp (BrigCompareOperation8_t cmp, BrigType16_t t,
1547 hsa_op_base *arg0, hsa_op_base *arg1,
1548 hsa_op_base *arg2)
1549 : hsa_insn_basic (3 , BRIG_OPCODE_CMP, t, arg0, arg1, arg2), m_compare (cmp)
1553 /* New operator to allocate compare instruction from pool alloc. */
1555 void *
1556 hsa_insn_cmp::operator new (size_t)
1558 return hsa_allocp_inst_cmp->allocate_raw ();
1561 /* Constructor of classes representing memory accesses. OPC is the opcode (must
1562 be BRIG_OPCODE_ST or BRIG_OPCODE_LD) and T is the type. The instruction
1563 operands are provided as ARG0 and ARG1. */
1565 hsa_insn_mem::hsa_insn_mem (int opc, BrigType16_t t, hsa_op_base *arg0,
1566 hsa_op_base *arg1)
1567 : hsa_insn_basic (2, opc, t, arg0, arg1),
1568 m_align (hsa_natural_alignment (t)), m_equiv_class (0)
1570 gcc_checking_assert (opc == BRIG_OPCODE_LD || opc == BRIG_OPCODE_ST);
1573 /* Constructor for descendants allowing different opcodes and number of
1574 operands, it passes its arguments directly to hsa_insn_basic
1575 constructor. The instruction operands are provided as ARG[0-3]. */
1578 hsa_insn_mem::hsa_insn_mem (unsigned nops, int opc, BrigType16_t t,
1579 hsa_op_base *arg0, hsa_op_base *arg1,
1580 hsa_op_base *arg2, hsa_op_base *arg3)
1581 : hsa_insn_basic (nops, opc, t, arg0, arg1, arg2, arg3),
1582 m_align (hsa_natural_alignment (t)), m_equiv_class (0)
1586 /* New operator to allocate memory instruction from pool alloc. */
1588 void *
1589 hsa_insn_mem::operator new (size_t)
1591 return hsa_allocp_inst_mem->allocate_raw ();
1594 /* Constructor of class representing atomic instructions and signals. OPC is
1595 the principal opcode, aop is the specific atomic operation opcode. T is the
1596 type of the instruction. The instruction operands
1597 are provided as ARG[0-3]. */
1599 hsa_insn_atomic::hsa_insn_atomic (int nops, int opc,
1600 enum BrigAtomicOperation aop,
1601 BrigType16_t t, BrigMemoryOrder memorder,
1602 hsa_op_base *arg0,
1603 hsa_op_base *arg1, hsa_op_base *arg2,
1604 hsa_op_base *arg3)
1605 : hsa_insn_mem (nops, opc, t, arg0, arg1, arg2, arg3), m_atomicop (aop),
1606 m_memoryorder (memorder),
1607 m_memoryscope (BRIG_MEMORY_SCOPE_SYSTEM)
1609 gcc_checking_assert (opc == BRIG_OPCODE_ATOMICNORET ||
1610 opc == BRIG_OPCODE_ATOMIC ||
1611 opc == BRIG_OPCODE_SIGNAL ||
1612 opc == BRIG_OPCODE_SIGNALNORET);
1615 /* New operator to allocate signal instruction from pool alloc. */
1617 void *
1618 hsa_insn_atomic::operator new (size_t)
1620 return hsa_allocp_inst_atomic->allocate_raw ();
1623 /* Constructor of class representing signal instructions. OPC is the prinicpal
1624 opcode, sop is the specific signal operation opcode. T is the type of the
1625 instruction. The instruction operands are provided as ARG[0-3]. */
1627 hsa_insn_signal::hsa_insn_signal (int nops, int opc,
1628 enum BrigAtomicOperation sop,
1629 BrigType16_t t, hsa_op_base *arg0,
1630 hsa_op_base *arg1, hsa_op_base *arg2,
1631 hsa_op_base *arg3)
1632 : hsa_insn_atomic (nops, opc, sop, t, BRIG_MEMORY_ORDER_SC_ACQUIRE_RELEASE,
1633 arg0, arg1, arg2, arg3)
1637 /* New operator to allocate signal instruction from pool alloc. */
1639 void *
1640 hsa_insn_signal::operator new (size_t)
1642 return hsa_allocp_inst_signal->allocate_raw ();
1645 /* Constructor of class representing segment conversion instructions. OPC is
1646 the opcode which must be either BRIG_OPCODE_STOF or BRIG_OPCODE_FTOS. DEST
1647 and SRCT are destination and source types respectively, SEG is the segment
1648 we are converting to or from. The instruction operands are
1649 provided as ARG0 and ARG1. */
1651 hsa_insn_seg::hsa_insn_seg (int opc, BrigType16_t dest, BrigType16_t srct,
1652 BrigSegment8_t seg, hsa_op_base *arg0,
1653 hsa_op_base *arg1)
1654 : hsa_insn_basic (2, opc, dest, arg0, arg1), m_src_type (srct),
1655 m_segment (seg)
1657 gcc_checking_assert (opc == BRIG_OPCODE_STOF || opc == BRIG_OPCODE_FTOS);
1660 /* New operator to allocate address conversion instruction from pool alloc. */
1662 void *
1663 hsa_insn_seg::operator new (size_t)
1665 return hsa_allocp_inst_seg->allocate_raw ();
1668 /* Constructor of class representing a call instruction. CALLEE is the tree
1669 representation of the function being called. */
1671 hsa_insn_call::hsa_insn_call (tree callee)
1672 : hsa_insn_basic (0, BRIG_OPCODE_CALL), m_called_function (callee),
1673 m_output_arg (NULL), m_args_code_list (NULL), m_result_code_list (NULL)
1677 hsa_insn_call::hsa_insn_call (hsa_internal_fn *fn)
1678 : hsa_insn_basic (0, BRIG_OPCODE_CALL), m_called_function (NULL),
1679 m_called_internal_fn (fn), m_output_arg (NULL), m_args_code_list (NULL),
1680 m_result_code_list (NULL)
1684 /* New operator to allocate call instruction from pool alloc. */
1686 void *
1687 hsa_insn_call::operator new (size_t)
1689 return hsa_allocp_inst_call->allocate_raw ();
1692 hsa_insn_call::~hsa_insn_call ()
1694 for (unsigned i = 0; i < m_input_args.length (); i++)
1695 delete m_input_args[i];
1697 delete m_output_arg;
1699 m_input_args.release ();
1700 m_input_arg_insns.release ();
1703 /* Constructor of class representing the argument block required to invoke
1704 a call in HSAIL. */
1705 hsa_insn_arg_block::hsa_insn_arg_block (BrigKind brig_kind,
1706 hsa_insn_call * call)
1707 : hsa_insn_basic (0, HSA_OPCODE_ARG_BLOCK), m_kind (brig_kind),
1708 m_call_insn (call)
1712 /* New operator to allocate argument block instruction from pool alloc. */
1714 void *
1715 hsa_insn_arg_block::operator new (size_t)
1717 return hsa_allocp_inst_arg_block->allocate_raw ();
1720 hsa_insn_comment::hsa_insn_comment (const char *s)
1721 : hsa_insn_basic (0, BRIG_KIND_DIRECTIVE_COMMENT)
1723 unsigned l = strlen (s);
1725 /* Append '// ' to the string. */
1726 char *buf = XNEWVEC (char, l + 4);
1727 sprintf (buf, "// %s", s);
1728 m_comment = buf;
1731 /* New operator to allocate comment instruction from pool alloc. */
1733 void *
1734 hsa_insn_comment::operator new (size_t)
1736 return hsa_allocp_inst_comment->allocate_raw ();
1739 hsa_insn_comment::~hsa_insn_comment ()
1741 gcc_checking_assert (m_comment);
1742 free (m_comment);
1743 m_comment = NULL;
1746 /* Constructor of class representing the queue instruction in HSAIL. */
1747 hsa_insn_queue::hsa_insn_queue (int nops, BrigOpcode opcode)
1748 : hsa_insn_basic (nops, opcode, BRIG_TYPE_U64)
1752 /* New operator to allocate source type instruction from pool alloc. */
1754 void *
1755 hsa_insn_srctype::operator new (size_t)
1757 return hsa_allocp_inst_srctype->allocate_raw ();
1760 /* Constructor of class representing the source type instruction in HSAIL. */
1762 hsa_insn_srctype::hsa_insn_srctype (int nops, BrigOpcode opcode,
1763 BrigType16_t destt, BrigType16_t srct,
1764 hsa_op_base *arg0, hsa_op_base *arg1,
1765 hsa_op_base *arg2 = NULL)
1766 : hsa_insn_basic (nops, opcode, destt, arg0, arg1, arg2),
1767 m_source_type (srct)
1770 /* New operator to allocate packed instruction from pool alloc. */
1772 void *
1773 hsa_insn_packed::operator new (size_t)
1775 return hsa_allocp_inst_packed->allocate_raw ();
1778 /* Constructor of class representing the packed instruction in HSAIL. */
1780 hsa_insn_packed::hsa_insn_packed (int nops, BrigOpcode opcode,
1781 BrigType16_t destt, BrigType16_t srct,
1782 hsa_op_base *arg0, hsa_op_base *arg1,
1783 hsa_op_base *arg2)
1784 : hsa_insn_srctype (nops, opcode, destt, srct, arg0, arg1, arg2)
1786 m_operand_list = new hsa_op_operand_list (nops - 1);
1789 /* New operator to allocate convert instruction from pool alloc. */
1791 void *
1792 hsa_insn_cvt::operator new (size_t)
1794 return hsa_allocp_inst_cvt->allocate_raw ();
1797 /* Constructor of class representing the convert instruction in HSAIL. */
1799 hsa_insn_cvt::hsa_insn_cvt (hsa_op_with_type *dest, hsa_op_with_type *src)
1800 : hsa_insn_basic (2, BRIG_OPCODE_CVT, dest->m_type, dest, src)
1804 /* New operator to allocate alloca from pool alloc. */
1806 void *
1807 hsa_insn_alloca::operator new (size_t)
1809 return hsa_allocp_inst_alloca->allocate_raw ();
1812 /* Constructor of class representing the alloca in HSAIL. */
1814 hsa_insn_alloca::hsa_insn_alloca (hsa_op_with_type *dest,
1815 hsa_op_with_type *size, unsigned alignment)
1816 : hsa_insn_basic (2, BRIG_OPCODE_ALLOCA, dest->m_type, dest, size),
1817 m_align (BRIG_ALIGNMENT_8)
1819 gcc_assert (dest->m_type == BRIG_TYPE_U32);
1820 if (alignment)
1821 m_align = hsa_alignment_encoding (alignment);
1824 /* Append an instruction INSN into the basic block. */
1826 void
1827 hsa_bb::append_insn (hsa_insn_basic *insn)
1829 gcc_assert (insn->m_opcode != 0 || insn->operand_count () == 0);
1830 gcc_assert (!insn->m_bb);
1832 insn->m_bb = m_bb;
1833 insn->m_prev = m_last_insn;
1834 insn->m_next = NULL;
1835 if (m_last_insn)
1836 m_last_insn->m_next = insn;
1837 m_last_insn = insn;
1838 if (!m_first_insn)
1839 m_first_insn = insn;
1842 /* Insert HSA instruction NEW_INSN immediately before an existing instruction
1843 OLD_INSN. */
1845 static void
1846 hsa_insert_insn_before (hsa_insn_basic *new_insn, hsa_insn_basic *old_insn)
1848 hsa_bb *hbb = hsa_bb_for_bb (old_insn->m_bb);
1850 if (hbb->m_first_insn == old_insn)
1851 hbb->m_first_insn = new_insn;
1852 new_insn->m_prev = old_insn->m_prev;
1853 new_insn->m_next = old_insn;
1854 if (old_insn->m_prev)
1855 old_insn->m_prev->m_next = new_insn;
1856 old_insn->m_prev = new_insn;
1859 /* Append HSA instruction NEW_INSN immediately after an existing instruction
1860 OLD_INSN. */
1862 static void
1863 hsa_append_insn_after (hsa_insn_basic *new_insn, hsa_insn_basic *old_insn)
1865 hsa_bb *hbb = hsa_bb_for_bb (old_insn->m_bb);
1867 if (hbb->m_last_insn == old_insn)
1868 hbb->m_last_insn = new_insn;
1869 new_insn->m_prev = old_insn;
1870 new_insn->m_next = old_insn->m_next;
1871 if (old_insn->m_next)
1872 old_insn->m_next->m_prev = new_insn;
1873 old_insn->m_next = new_insn;
1876 /* Return a register containing the calculated value of EXP which must be an
1877 expression consisting of PLUS_EXPRs, MULT_EXPRs, NOP_EXPRs, SSA_NAMEs and
1878 integer constants as returned by get_inner_reference.
1879 Newly generated HSA instructions will be appended to HBB.
1880 Perform all calculations in ADDRTYPE. */
1882 static hsa_op_with_type *
1883 gen_address_calculation (tree exp, hsa_bb *hbb, BrigType16_t addrtype)
1885 int opcode;
1887 if (TREE_CODE (exp) == NOP_EXPR)
1888 exp = TREE_OPERAND (exp, 0);
1890 switch (TREE_CODE (exp))
1892 case SSA_NAME:
1893 return hsa_cfun->reg_for_gimple_ssa (exp)->get_in_type (addrtype, hbb);
1895 case INTEGER_CST:
1897 hsa_op_immed *imm = new hsa_op_immed (exp);
1898 if (addrtype != imm->m_type)
1899 imm->m_type = addrtype;
1900 return imm;
1903 case PLUS_EXPR:
1904 opcode = BRIG_OPCODE_ADD;
1905 break;
1907 case MULT_EXPR:
1908 opcode = BRIG_OPCODE_MUL;
1909 break;
1911 default:
1912 gcc_unreachable ();
1915 hsa_op_reg *res = new hsa_op_reg (addrtype);
1916 hsa_insn_basic *insn = new hsa_insn_basic (3, opcode, addrtype);
1917 insn->set_op (0, res);
1919 hsa_op_with_type *op1 = gen_address_calculation (TREE_OPERAND (exp, 0), hbb,
1920 addrtype);
1921 hsa_op_with_type *op2 = gen_address_calculation (TREE_OPERAND (exp, 1), hbb,
1922 addrtype);
1923 insn->set_op (1, op1);
1924 insn->set_op (2, op2);
1926 hbb->append_insn (insn);
1927 return res;
1930 /* If R1 is NULL, just return R2, otherwise append an instruction adding them
1931 to HBB and return the register holding the result. */
1933 static hsa_op_reg *
1934 add_addr_regs_if_needed (hsa_op_reg *r1, hsa_op_reg *r2, hsa_bb *hbb)
1936 gcc_checking_assert (r2);
1937 if (!r1)
1938 return r2;
1940 hsa_op_reg *res = new hsa_op_reg (r1->m_type);
1941 gcc_assert (!hsa_needs_cvt (r1->m_type, r2->m_type));
1942 hsa_insn_basic *insn = new hsa_insn_basic (3, BRIG_OPCODE_ADD, res->m_type);
1943 insn->set_op (0, res);
1944 insn->set_op (1, r1);
1945 insn->set_op (2, r2);
1946 hbb->append_insn (insn);
1947 return res;
1950 /* Helper of gen_hsa_addr. Update *SYMBOL, *ADDRTYPE, *REG and *OFFSET to
1951 reflect BASE which is the first operand of a MEM_REF or a TARGET_MEM_REF. */
1953 static void
1954 process_mem_base (tree base, hsa_symbol **symbol, BrigType16_t *addrtype,
1955 hsa_op_reg **reg, offset_int *offset, hsa_bb *hbb)
1957 if (TREE_CODE (base) == SSA_NAME)
1959 gcc_assert (!*reg);
1960 hsa_op_with_type *ssa
1961 = hsa_cfun->reg_for_gimple_ssa (base)->get_in_type (*addrtype, hbb);
1962 *reg = dyn_cast <hsa_op_reg *> (ssa);
1964 else if (TREE_CODE (base) == ADDR_EXPR)
1966 tree decl = TREE_OPERAND (base, 0);
1968 if (!DECL_P (decl) || TREE_CODE (decl) == FUNCTION_DECL)
1970 HSA_SORRY_AT (EXPR_LOCATION (base),
1971 "support for HSA does not implement a memory reference "
1972 "to a non-declaration type");
1973 return;
1976 gcc_assert (!*symbol);
1978 *symbol = get_symbol_for_decl (decl);
1979 *addrtype = hsa_get_segment_addr_type ((*symbol)->m_segment);
1981 else if (TREE_CODE (base) == INTEGER_CST)
1982 *offset += wi::to_offset (base);
1983 else
1984 gcc_unreachable ();
1987 /* Forward declaration of a function. */
1989 static void
1990 gen_hsa_addr_insns (tree val, hsa_op_reg *dest, hsa_bb *hbb);
1992 /* Generate HSA address operand for a given tree memory reference REF. If
1993 instructions need to be created to calculate the address, they will be added
1994 to the end of HBB. If a caller provider OUTPUT_BITSIZE and OUTPUT_BITPOS,
1995 the function assumes that the caller will handle possible
1996 bit-field references. Otherwise if we reference a bit-field, sorry message
1997 is displayed. */
1999 static hsa_op_address *
2000 gen_hsa_addr (tree ref, hsa_bb *hbb, HOST_WIDE_INT *output_bitsize = NULL,
2001 HOST_WIDE_INT *output_bitpos = NULL)
2003 hsa_symbol *symbol = NULL;
2004 hsa_op_reg *reg = NULL;
2005 offset_int offset = 0;
2006 tree origref = ref;
2007 tree varoffset = NULL_TREE;
2008 BrigType16_t addrtype = hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT);
2009 HOST_WIDE_INT bitsize = 0, bitpos = 0;
2010 BrigType16_t flat_addrtype = hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT);
2012 if (TREE_CODE (ref) == STRING_CST)
2014 symbol = hsa_get_string_cst_symbol (ref);
2015 goto out;
2017 else if (TREE_CODE (ref) == BIT_FIELD_REF
2018 && ((tree_to_uhwi (TREE_OPERAND (ref, 1)) % BITS_PER_UNIT) != 0
2019 || (tree_to_uhwi (TREE_OPERAND (ref, 2)) % BITS_PER_UNIT) != 0))
2021 HSA_SORRY_ATV (EXPR_LOCATION (origref),
2022 "support for HSA does not implement "
2023 "bit field references such as %E", ref);
2024 goto out;
2027 if (handled_component_p (ref))
2029 enum machine_mode mode;
2030 int unsignedp, volatilep, preversep;
2032 ref = get_inner_reference (ref, &bitsize, &bitpos, &varoffset, &mode,
2033 &unsignedp, &preversep, &volatilep, false);
2035 offset = bitpos;
2036 offset = wi::rshift (offset, LOG2_BITS_PER_UNIT, SIGNED);
2039 switch (TREE_CODE (ref))
2041 case ADDR_EXPR:
2043 addrtype = hsa_get_segment_addr_type (BRIG_SEGMENT_PRIVATE);
2044 symbol = hsa_cfun->create_hsa_temporary (flat_addrtype);
2045 hsa_op_reg *r = new hsa_op_reg (flat_addrtype);
2046 gen_hsa_addr_insns (ref, r, hbb);
2047 hbb->append_insn (new hsa_insn_mem (BRIG_OPCODE_ST, r->m_type,
2048 r, new hsa_op_address (symbol)));
2050 break;
2052 case SSA_NAME:
2054 addrtype = hsa_get_segment_addr_type (BRIG_SEGMENT_PRIVATE);
2055 symbol = hsa_cfun->create_hsa_temporary (flat_addrtype);
2056 hsa_op_reg *r = hsa_cfun->reg_for_gimple_ssa (ref);
2058 hbb->append_insn (new hsa_insn_mem (BRIG_OPCODE_ST, r->m_type,
2059 r, new hsa_op_address (symbol)));
2061 break;
2063 case PARM_DECL:
2064 case VAR_DECL:
2065 case RESULT_DECL:
2066 gcc_assert (!symbol);
2067 symbol = get_symbol_for_decl (ref);
2068 addrtype = hsa_get_segment_addr_type (symbol->m_segment);
2069 break;
2071 case MEM_REF:
2072 process_mem_base (TREE_OPERAND (ref, 0), &symbol, &addrtype, &reg,
2073 &offset, hbb);
2075 if (!integer_zerop (TREE_OPERAND (ref, 1)))
2076 offset += wi::to_offset (TREE_OPERAND (ref, 1));
2077 break;
2079 case TARGET_MEM_REF:
2080 process_mem_base (TMR_BASE (ref), &symbol, &addrtype, &reg, &offset, hbb);
2081 if (TMR_INDEX (ref))
2083 hsa_op_reg *disp1;
2084 hsa_op_base *idx = hsa_cfun->reg_for_gimple_ssa
2085 (TMR_INDEX (ref))->get_in_type (addrtype, hbb);
2086 if (TMR_STEP (ref) && !integer_onep (TMR_STEP (ref)))
2088 disp1 = new hsa_op_reg (addrtype);
2089 hsa_insn_basic *insn = new hsa_insn_basic (3, BRIG_OPCODE_MUL,
2090 addrtype);
2092 /* As step must respect addrtype, we overwrite the type
2093 of an immediate value. */
2094 hsa_op_immed *step = new hsa_op_immed (TMR_STEP (ref));
2095 step->m_type = addrtype;
2097 insn->set_op (0, disp1);
2098 insn->set_op (1, idx);
2099 insn->set_op (2, step);
2100 hbb->append_insn (insn);
2102 else
2103 disp1 = as_a <hsa_op_reg *> (idx);
2104 reg = add_addr_regs_if_needed (reg, disp1, hbb);
2106 if (TMR_INDEX2 (ref))
2108 hsa_op_base *disp2 = hsa_cfun->reg_for_gimple_ssa
2109 (TMR_INDEX2 (ref))->get_in_type (addrtype, hbb);
2110 reg = add_addr_regs_if_needed (reg, as_a <hsa_op_reg *> (disp2), hbb);
2112 offset += wi::to_offset (TMR_OFFSET (ref));
2113 break;
2114 case FUNCTION_DECL:
2115 HSA_SORRY_AT (EXPR_LOCATION (origref),
2116 "support for HSA does not implement function pointers");
2117 goto out;
2118 default:
2119 HSA_SORRY_ATV (EXPR_LOCATION (origref), "support for HSA does "
2120 "not implement memory access to %E", origref);
2121 goto out;
2124 if (varoffset)
2126 if (TREE_CODE (varoffset) == INTEGER_CST)
2127 offset += wi::to_offset (varoffset);
2128 else
2130 hsa_op_base *off_op = gen_address_calculation (varoffset, hbb,
2131 addrtype);
2132 reg = add_addr_regs_if_needed (reg, as_a <hsa_op_reg *> (off_op),
2133 hbb);
2137 gcc_checking_assert ((symbol
2138 && addrtype
2139 == hsa_get_segment_addr_type (symbol->m_segment))
2140 || (!symbol
2141 && addrtype
2142 == hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT)));
2143 out:
2144 HOST_WIDE_INT hwi_offset = offset.to_shwi ();
2146 /* Calculate remaining bitsize offset (if presented). */
2147 bitpos %= BITS_PER_UNIT;
2148 /* If bitsize is a power of two that is greater or equal to BITS_PER_UNIT, it
2149 is not a reason to think this is a bit-field access. */
2150 if (bitpos == 0
2151 && (bitsize >= BITS_PER_UNIT)
2152 && !(bitsize & (bitsize - 1)))
2153 bitsize = 0;
2155 if ((bitpos || bitsize) && (output_bitpos == NULL || output_bitsize == NULL))
2156 HSA_SORRY_ATV (EXPR_LOCATION (origref), "support for HSA does not "
2157 "implement unhandled bit field reference such as %E", ref);
2159 if (output_bitsize != NULL && output_bitpos != NULL)
2161 *output_bitsize = bitsize;
2162 *output_bitpos = bitpos;
2165 return new hsa_op_address (symbol, reg, hwi_offset);
2168 /* Generate HSA address for a function call argument of given TYPE.
2169 INDEX is used to generate corresponding name of the arguments.
2170 Special value -1 represents fact that result value is created. */
2172 static hsa_op_address *
2173 gen_hsa_addr_for_arg (tree tree_type, int index)
2175 hsa_symbol *sym = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_ARG,
2176 BRIG_LINKAGE_ARG);
2177 sym->m_type = hsa_type_for_tree_type (tree_type, &sym->m_dim);
2179 if (index == -1) /* Function result. */
2180 sym->m_name = "res";
2181 else /* Function call arguments. */
2183 sym->m_name = NULL;
2184 sym->m_name_number = index;
2187 return new hsa_op_address (sym);
2190 /* Generate HSA instructions that calculate address of VAL including all
2191 necessary conversions to flat addressing and place the result into DEST.
2192 Instructions are appended to HBB. */
2194 static void
2195 gen_hsa_addr_insns (tree val, hsa_op_reg *dest, hsa_bb *hbb)
2197 /* Handle cases like tmp = NULL, where we just emit a move instruction
2198 to a register. */
2199 if (TREE_CODE (val) == INTEGER_CST)
2201 hsa_op_immed *c = new hsa_op_immed (val);
2202 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_MOV,
2203 dest->m_type, dest, c);
2204 hbb->append_insn (insn);
2205 return;
2208 hsa_op_address *addr;
2210 gcc_assert (dest->m_type == hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT));
2211 if (TREE_CODE (val) == ADDR_EXPR)
2212 val = TREE_OPERAND (val, 0);
2213 addr = gen_hsa_addr (val, hbb);
2214 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_LDA);
2215 insn->set_op (1, addr);
2216 if (addr->m_symbol && addr->m_symbol->m_segment != BRIG_SEGMENT_GLOBAL)
2218 /* LDA produces segment-relative address, we need to convert
2219 it to the flat one. */
2220 hsa_op_reg *tmp;
2221 tmp = new hsa_op_reg (hsa_get_segment_addr_type
2222 (addr->m_symbol->m_segment));
2223 hsa_insn_seg *seg;
2224 seg = new hsa_insn_seg (BRIG_OPCODE_STOF,
2225 hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT),
2226 tmp->m_type, addr->m_symbol->m_segment, dest,
2227 tmp);
2229 insn->set_op (0, tmp);
2230 insn->m_type = tmp->m_type;
2231 hbb->append_insn (insn);
2232 hbb->append_insn (seg);
2234 else
2236 insn->set_op (0, dest);
2237 insn->m_type = hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT);
2238 hbb->append_insn (insn);
2242 /* Return an HSA register or HSA immediate value operand corresponding to
2243 gimple operand OP. */
2245 static hsa_op_with_type *
2246 hsa_reg_or_immed_for_gimple_op (tree op, hsa_bb *hbb)
2248 hsa_op_reg *tmp;
2250 if (TREE_CODE (op) == SSA_NAME)
2251 tmp = hsa_cfun->reg_for_gimple_ssa (op);
2252 else if (!POINTER_TYPE_P (TREE_TYPE (op)))
2253 return new hsa_op_immed (op);
2254 else
2256 tmp = new hsa_op_reg (hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT));
2257 gen_hsa_addr_insns (op, tmp, hbb);
2259 return tmp;
2262 /* Create a simple movement instruction with register destination DEST and
2263 register or immediate source SRC and append it to the end of HBB. */
2265 void
2266 hsa_build_append_simple_mov (hsa_op_reg *dest, hsa_op_base *src, hsa_bb *hbb)
2268 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_MOV, dest->m_type,
2269 dest, src);
2270 if (hsa_op_reg *sreg = dyn_cast <hsa_op_reg *> (src))
2271 gcc_assert (hsa_type_bit_size (dest->m_type)
2272 == hsa_type_bit_size (sreg->m_type));
2273 else
2274 gcc_assert (hsa_type_bit_size (dest->m_type)
2275 == hsa_type_bit_size (as_a <hsa_op_immed *> (src)->m_type));
2277 hbb->append_insn (insn);
2280 /* Generate HSAIL instructions loading a bit field into register DEST.
2281 VALUE_REG is a register of a SSA name that is used in the bit field
2282 reference. To identify a bit field BITPOS is offset to the loaded memory
2283 and BITSIZE is number of bits of the bit field.
2284 Add instructions to HBB. */
2286 static void
2287 gen_hsa_insns_for_bitfield (hsa_op_reg *dest, hsa_op_reg *value_reg,
2288 HOST_WIDE_INT bitsize, HOST_WIDE_INT bitpos,
2289 hsa_bb *hbb)
2291 unsigned type_bitsize = hsa_type_bit_size (dest->m_type);
2292 unsigned left_shift = type_bitsize - (bitsize + bitpos);
2293 unsigned right_shift = left_shift + bitpos;
2295 if (left_shift)
2297 hsa_op_reg *value_reg_2 = new hsa_op_reg (dest->m_type);
2298 hsa_op_immed *c = new hsa_op_immed (left_shift, BRIG_TYPE_U32);
2300 hsa_insn_basic *lshift
2301 = new hsa_insn_basic (3, BRIG_OPCODE_SHL, value_reg_2->m_type,
2302 value_reg_2, value_reg, c);
2304 hbb->append_insn (lshift);
2306 value_reg = value_reg_2;
2309 if (right_shift)
2311 hsa_op_reg *value_reg_2 = new hsa_op_reg (dest->m_type);
2312 hsa_op_immed *c = new hsa_op_immed (right_shift, BRIG_TYPE_U32);
2314 hsa_insn_basic *rshift
2315 = new hsa_insn_basic (3, BRIG_OPCODE_SHR, value_reg_2->m_type,
2316 value_reg_2, value_reg, c);
2318 hbb->append_insn (rshift);
2320 value_reg = value_reg_2;
2323 hsa_insn_basic *assignment
2324 = new hsa_insn_basic (2, BRIG_OPCODE_MOV, dest->m_type, dest, value_reg);
2325 hbb->append_insn (assignment);
2329 /* Generate HSAIL instructions loading a bit field into register DEST. ADDR is
2330 prepared memory address which is used to load the bit field. To identify a
2331 bit field BITPOS is offset to the loaded memory and BITSIZE is number of
2332 bits of the bit field. Add instructions to HBB. Load must be performed in
2333 alignment ALIGN. */
2335 static void
2336 gen_hsa_insns_for_bitfield_load (hsa_op_reg *dest, hsa_op_address *addr,
2337 HOST_WIDE_INT bitsize, HOST_WIDE_INT bitpos,
2338 hsa_bb *hbb, BrigAlignment8_t align)
2340 hsa_op_reg *value_reg = new hsa_op_reg (dest->m_type);
2341 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, dest->m_type, value_reg,
2342 addr);
2343 mem->set_align (align);
2344 hbb->append_insn (mem);
2345 gen_hsa_insns_for_bitfield (dest, value_reg, bitsize, bitpos, hbb);
2348 /* Return the alignment of base memory accesses we issue to perform bit-field
2349 memory access REF. */
2351 static BrigAlignment8_t
2352 hsa_bitmemref_alignment (tree ref)
2354 unsigned HOST_WIDE_INT bit_offset = 0;
2356 while (true)
2358 if (TREE_CODE (ref) == BIT_FIELD_REF)
2360 if (!tree_fits_uhwi_p (TREE_OPERAND (ref, 2)))
2361 return BRIG_ALIGNMENT_1;
2362 bit_offset += tree_to_uhwi (TREE_OPERAND (ref, 2));
2364 else if (TREE_CODE (ref) == COMPONENT_REF
2365 && DECL_BIT_FIELD (TREE_OPERAND (ref, 1)))
2366 bit_offset += int_bit_position (TREE_OPERAND (ref, 1));
2367 else
2368 break;
2369 ref = TREE_OPERAND (ref, 0);
2372 unsigned HOST_WIDE_INT bits = bit_offset % BITS_PER_UNIT;
2373 unsigned HOST_WIDE_INT byte_bits = bit_offset - bits;
2374 BrigAlignment8_t base = hsa_alignment_encoding (get_object_alignment (ref));
2375 if (byte_bits == 0)
2376 return base;
2377 return MIN (base, hsa_alignment_encoding (byte_bits & -byte_bits));
2380 /* Generate HSAIL instructions loading something into register DEST. RHS is
2381 tree representation of the loaded data, which are loaded as type TYPE. Add
2382 instructions to HBB. */
2384 static void
2385 gen_hsa_insns_for_load (hsa_op_reg *dest, tree rhs, tree type, hsa_bb *hbb)
2387 /* The destination SSA name will give us the type. */
2388 if (TREE_CODE (rhs) == VIEW_CONVERT_EXPR)
2389 rhs = TREE_OPERAND (rhs, 0);
2391 if (TREE_CODE (rhs) == SSA_NAME)
2393 hsa_op_reg *src = hsa_cfun->reg_for_gimple_ssa (rhs);
2394 hsa_build_append_simple_mov (dest, src, hbb);
2396 else if (is_gimple_min_invariant (rhs)
2397 || TREE_CODE (rhs) == ADDR_EXPR)
2399 if (POINTER_TYPE_P (TREE_TYPE (rhs)))
2401 if (dest->m_type != hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT))
2403 HSA_SORRY_ATV (EXPR_LOCATION (rhs),
2404 "support for HSA does not implement conversion "
2405 "of %E to the requested non-pointer type.", rhs);
2406 return;
2409 gen_hsa_addr_insns (rhs, dest, hbb);
2411 else if (TREE_CODE (rhs) == COMPLEX_CST)
2413 hsa_op_immed *real_part = new hsa_op_immed (TREE_REALPART (rhs));
2414 hsa_op_immed *imag_part = new hsa_op_immed (TREE_IMAGPART (rhs));
2416 hsa_op_reg *real_part_reg
2417 = new hsa_op_reg (hsa_type_for_scalar_tree_type (TREE_TYPE (type),
2418 true));
2419 hsa_op_reg *imag_part_reg
2420 = new hsa_op_reg (hsa_type_for_scalar_tree_type (TREE_TYPE (type),
2421 true));
2423 hsa_build_append_simple_mov (real_part_reg, real_part, hbb);
2424 hsa_build_append_simple_mov (imag_part_reg, imag_part, hbb);
2426 BrigType16_t src_type = hsa_bittype_for_type (real_part_reg->m_type);
2428 hsa_insn_packed *insn
2429 = new hsa_insn_packed (3, BRIG_OPCODE_COMBINE, dest->m_type,
2430 src_type, dest, real_part_reg,
2431 imag_part_reg);
2432 hbb->append_insn (insn);
2434 else
2436 hsa_op_immed *imm = new hsa_op_immed (rhs);
2437 hsa_build_append_simple_mov (dest, imm, hbb);
2440 else if (TREE_CODE (rhs) == REALPART_EXPR || TREE_CODE (rhs) == IMAGPART_EXPR)
2442 tree pack_type = TREE_TYPE (TREE_OPERAND (rhs, 0));
2444 hsa_op_reg *packed_reg
2445 = new hsa_op_reg (hsa_type_for_scalar_tree_type (pack_type, true));
2447 tree complex_rhs = TREE_OPERAND (rhs, 0);
2448 gen_hsa_insns_for_load (packed_reg, complex_rhs, TREE_TYPE (complex_rhs),
2449 hbb);
2451 hsa_op_reg *real_reg
2452 = new hsa_op_reg (hsa_type_for_scalar_tree_type (type, true));
2454 hsa_op_reg *imag_reg
2455 = new hsa_op_reg (hsa_type_for_scalar_tree_type (type, true));
2457 BrigKind16_t brig_type = packed_reg->m_type;
2458 hsa_insn_packed *packed
2459 = new hsa_insn_packed (3, BRIG_OPCODE_EXPAND,
2460 hsa_bittype_for_type (real_reg->m_type),
2461 brig_type, real_reg, imag_reg, packed_reg);
2463 hbb->append_insn (packed);
2465 hsa_op_reg *source = TREE_CODE (rhs) == REALPART_EXPR ?
2466 real_reg : imag_reg;
2468 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_MOV,
2469 dest->m_type, dest, source);
2471 hbb->append_insn (insn);
2473 else if (TREE_CODE (rhs) == BIT_FIELD_REF
2474 && TREE_CODE (TREE_OPERAND (rhs, 0)) == SSA_NAME)
2476 tree ssa_name = TREE_OPERAND (rhs, 0);
2477 HOST_WIDE_INT bitsize = tree_to_uhwi (TREE_OPERAND (rhs, 1));
2478 HOST_WIDE_INT bitpos = tree_to_uhwi (TREE_OPERAND (rhs, 2));
2480 hsa_op_reg *imm_value = hsa_cfun->reg_for_gimple_ssa (ssa_name);
2481 gen_hsa_insns_for_bitfield (dest, imm_value, bitsize, bitpos, hbb);
2483 else if (DECL_P (rhs) || TREE_CODE (rhs) == MEM_REF
2484 || TREE_CODE (rhs) == TARGET_MEM_REF
2485 || handled_component_p (rhs))
2487 HOST_WIDE_INT bitsize, bitpos;
2489 /* Load from memory. */
2490 hsa_op_address *addr;
2491 addr = gen_hsa_addr (rhs, hbb, &bitsize, &bitpos);
2493 /* Handle load of a bit field. */
2494 if (bitsize > 64)
2496 HSA_SORRY_AT (EXPR_LOCATION (rhs),
2497 "support for HSA does not implement load from a bit "
2498 "field bigger than 64 bits");
2499 return;
2502 if (bitsize || bitpos)
2503 gen_hsa_insns_for_bitfield_load (dest, addr, bitsize, bitpos, hbb,
2504 hsa_bitmemref_alignment (rhs));
2505 else
2507 BrigType16_t mtype;
2508 /* Not dest->m_type, that's possibly extended. */
2509 mtype = mem_type_for_type (hsa_type_for_scalar_tree_type (type,
2510 false));
2511 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, mtype, dest,
2512 addr);
2513 mem->set_align (hsa_alignment_encoding (get_object_alignment (rhs)));
2514 hbb->append_insn (mem);
2517 else
2518 HSA_SORRY_ATV (EXPR_LOCATION (rhs),
2519 "support for HSA does not implement loading "
2520 "of expression %E",
2521 rhs);
2524 /* Return number of bits necessary for representation of a bit field,
2525 starting at BITPOS with size of BITSIZE. */
2527 static unsigned
2528 get_bitfield_size (unsigned bitpos, unsigned bitsize)
2530 unsigned s = bitpos + bitsize;
2531 unsigned sizes[] = {8, 16, 32, 64};
2533 for (unsigned i = 0; i < 4; i++)
2534 if (s <= sizes[i])
2535 return sizes[i];
2537 gcc_unreachable ();
2538 return 0;
2541 /* Generate HSAIL instructions storing into memory. LHS is the destination of
2542 the store, SRC is the source operand. Add instructions to HBB. */
2544 static void
2545 gen_hsa_insns_for_store (tree lhs, hsa_op_base *src, hsa_bb *hbb)
2547 HOST_WIDE_INT bitsize = 0, bitpos = 0;
2548 BrigAlignment8_t req_align;
2549 BrigType16_t mtype;
2550 mtype = mem_type_for_type (hsa_type_for_scalar_tree_type (TREE_TYPE (lhs),
2551 false));
2552 hsa_op_address *addr;
2553 addr = gen_hsa_addr (lhs, hbb, &bitsize, &bitpos);
2555 /* Handle store to a bit field. */
2556 if (bitsize > 64)
2558 HSA_SORRY_AT (EXPR_LOCATION (lhs),
2559 "support for HSA does not implement store to a bit field "
2560 "bigger than 64 bits");
2561 return;
2564 unsigned type_bitsize = get_bitfield_size (bitpos, bitsize);
2566 /* HSAIL does not support MOV insn with 16-bits integers. */
2567 if (type_bitsize < 32)
2568 type_bitsize = 32;
2570 if (bitpos || (bitsize && type_bitsize != bitsize))
2572 unsigned HOST_WIDE_INT mask = 0;
2573 BrigType16_t mem_type
2574 = get_integer_type_by_bytes (type_bitsize / BITS_PER_UNIT,
2575 !TYPE_UNSIGNED (TREE_TYPE (lhs)));
2577 for (unsigned i = 0; i < type_bitsize; i++)
2578 if (i < bitpos || i >= bitpos + bitsize)
2579 mask |= ((unsigned HOST_WIDE_INT)1 << i);
2581 hsa_op_reg *value_reg = new hsa_op_reg (mem_type);
2583 req_align = hsa_bitmemref_alignment (lhs);
2584 /* Load value from memory. */
2585 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, mem_type,
2586 value_reg, addr);
2587 mem->set_align (req_align);
2588 hbb->append_insn (mem);
2590 /* AND the loaded value with prepared mask. */
2591 hsa_op_reg *cleared_reg = new hsa_op_reg (mem_type);
2593 BrigType16_t t
2594 = get_integer_type_by_bytes (type_bitsize / BITS_PER_UNIT, false);
2595 hsa_op_immed *c = new hsa_op_immed (mask, t);
2597 hsa_insn_basic *clearing
2598 = new hsa_insn_basic (3, BRIG_OPCODE_AND, mem_type, cleared_reg,
2599 value_reg, c);
2600 hbb->append_insn (clearing);
2602 /* Shift to left a value that is going to be stored. */
2603 hsa_op_reg *new_value_reg = new hsa_op_reg (mem_type);
2605 hsa_insn_basic *basic = new hsa_insn_basic (2, BRIG_OPCODE_MOV, mem_type,
2606 new_value_reg, src);
2607 hbb->append_insn (basic);
2609 if (bitpos)
2611 hsa_op_reg *shifted_value_reg = new hsa_op_reg (mem_type);
2612 c = new hsa_op_immed (bitpos, BRIG_TYPE_U32);
2614 hsa_insn_basic *basic
2615 = new hsa_insn_basic (3, BRIG_OPCODE_SHL, mem_type,
2616 shifted_value_reg, new_value_reg, c);
2617 hbb->append_insn (basic);
2619 new_value_reg = shifted_value_reg;
2622 /* OR the prepared value with prepared chunk loaded from memory. */
2623 hsa_op_reg *prepared_reg= new hsa_op_reg (mem_type);
2624 basic = new hsa_insn_basic (3, BRIG_OPCODE_OR, mem_type, prepared_reg,
2625 new_value_reg, cleared_reg);
2626 hbb->append_insn (basic);
2628 src = prepared_reg;
2629 mtype = mem_type;
2631 else
2632 req_align = hsa_alignment_encoding (get_object_alignment (lhs));
2634 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, mtype, src, addr);
2635 mem->set_align (req_align);
2637 /* The HSAIL verifier has another constraint: if the source is an immediate
2638 then it must match the destination type. If it's a register the low bits
2639 will be used for sub-word stores. We're always allocating new operands so
2640 we can modify the above in place. */
2641 if (hsa_op_immed *imm = dyn_cast <hsa_op_immed *> (src))
2643 if ((imm->m_type & BRIG_TYPE_PACK_MASK) == BRIG_TYPE_PACK_NONE)
2644 imm->m_type = mem->m_type;
2645 else
2647 /* ...and all vector immediates apparently need to be vectors of
2648 unsigned bytes. */
2649 unsigned bs = hsa_type_bit_size (imm->m_type);
2650 gcc_assert (bs == hsa_type_bit_size (mem->m_type));
2651 switch (bs)
2653 case 32:
2654 imm->m_type = BRIG_TYPE_U8X4;
2655 break;
2656 case 64:
2657 imm->m_type = BRIG_TYPE_U8X8;
2658 break;
2659 case 128:
2660 imm->m_type = BRIG_TYPE_U8X16;
2661 break;
2662 default:
2663 gcc_unreachable ();
2668 hbb->append_insn (mem);
2671 /* Generate memory copy instructions that are going to be used
2672 for copying a HSA symbol SRC_SYMBOL (or SRC_REG) to TARGET memory,
2673 represented by pointer in a register. */
2675 static void
2676 gen_hsa_memory_copy (hsa_bb *hbb, hsa_op_address *target, hsa_op_address *src,
2677 unsigned size)
2679 hsa_op_address *addr;
2680 hsa_insn_mem *mem;
2682 unsigned offset = 0;
2684 while (size)
2686 unsigned s;
2687 if (size >= 8)
2688 s = 8;
2689 else if (size >= 4)
2690 s = 4;
2691 else if (size >= 2)
2692 s = 2;
2693 else
2694 s = 1;
2696 BrigType16_t t = get_integer_type_by_bytes (s, false);
2698 hsa_op_reg *tmp = new hsa_op_reg (t);
2699 addr = new hsa_op_address (src->m_symbol, src->m_reg,
2700 src->m_imm_offset + offset);
2701 mem = new hsa_insn_mem (BRIG_OPCODE_LD, t, tmp, addr);
2702 hbb->append_insn (mem);
2704 addr = new hsa_op_address (target->m_symbol, target->m_reg,
2705 target->m_imm_offset + offset);
2706 mem = new hsa_insn_mem (BRIG_OPCODE_ST, t, tmp, addr);
2707 hbb->append_insn (mem);
2708 offset += s;
2709 size -= s;
2713 /* Create a memset mask that is created by copying a CONSTANT byte value
2714 to an integer of BYTE_SIZE bytes. */
2716 static unsigned HOST_WIDE_INT
2717 build_memset_value (unsigned HOST_WIDE_INT constant, unsigned byte_size)
2719 if (constant == 0)
2720 return 0;
2722 HOST_WIDE_INT v = constant;
2724 for (unsigned i = 1; i < byte_size; i++)
2725 v |= constant << (8 * i);
2727 return v;
2730 /* Generate memory set instructions that are going to be used
2731 for setting a CONSTANT byte value to TARGET memory of SIZE bytes. */
2733 static void
2734 gen_hsa_memory_set (hsa_bb *hbb, hsa_op_address *target,
2735 unsigned HOST_WIDE_INT constant,
2736 unsigned size)
2738 hsa_op_address *addr;
2739 hsa_insn_mem *mem;
2741 unsigned offset = 0;
2743 while (size)
2745 unsigned s;
2746 if (size >= 8)
2747 s = 8;
2748 else if (size >= 4)
2749 s = 4;
2750 else if (size >= 2)
2751 s = 2;
2752 else
2753 s = 1;
2755 addr = new hsa_op_address (target->m_symbol, target->m_reg,
2756 target->m_imm_offset + offset);
2758 BrigType16_t t = get_integer_type_by_bytes (s, false);
2759 HOST_WIDE_INT c = build_memset_value (constant, s);
2761 mem = new hsa_insn_mem (BRIG_OPCODE_ST, t, new hsa_op_immed (c, t),
2762 addr);
2763 hbb->append_insn (mem);
2764 offset += s;
2765 size -= s;
2769 /* Generate HSAIL instructions for a single assignment
2770 of an empty constructor to an ADDR_LHS. Constructor is passed as a
2771 tree RHS and all instructions are appended to HBB. */
2773 void
2774 gen_hsa_ctor_assignment (hsa_op_address *addr_lhs, tree rhs, hsa_bb *hbb)
2776 if (vec_safe_length (CONSTRUCTOR_ELTS (rhs)))
2778 HSA_SORRY_AT (EXPR_LOCATION (rhs),
2779 "support for HSA does not implement load from constructor");
2780 return;
2783 unsigned size = tree_to_uhwi (TYPE_SIZE_UNIT (TREE_TYPE (rhs)));
2784 gen_hsa_memory_set (hbb, addr_lhs, 0, size);
2787 /* Generate HSA instructions for a single assignment of RHS to LHS.
2788 HBB is the basic block they will be appended to. */
2790 static void
2791 gen_hsa_insns_for_single_assignment (tree lhs, tree rhs, hsa_bb *hbb)
2793 if (TREE_CODE (lhs) == SSA_NAME)
2795 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
2796 if (hsa_seen_error ())
2797 return;
2799 gen_hsa_insns_for_load (dest, rhs, TREE_TYPE (lhs), hbb);
2801 else if (TREE_CODE (rhs) == SSA_NAME
2802 || (is_gimple_min_invariant (rhs) && TREE_CODE (rhs) != STRING_CST))
2804 /* Store to memory. */
2805 hsa_op_base *src = hsa_reg_or_immed_for_gimple_op (rhs, hbb);
2806 if (hsa_seen_error ())
2807 return;
2809 gen_hsa_insns_for_store (lhs, src, hbb);
2811 else
2813 hsa_op_address *addr_lhs = gen_hsa_addr (lhs, hbb);
2815 if (TREE_CODE (rhs) == CONSTRUCTOR)
2816 gen_hsa_ctor_assignment (addr_lhs, rhs, hbb);
2817 else
2819 hsa_op_address *addr_rhs = gen_hsa_addr (rhs, hbb);
2821 unsigned size = tree_to_uhwi (TYPE_SIZE_UNIT (TREE_TYPE (rhs)));
2822 gen_hsa_memory_copy (hbb, addr_lhs, addr_rhs, size);
2827 /* Prepend before INSN a load from spill symbol of SPILL_REG. Return the
2828 register into which we loaded. If this required another register to convert
2829 from a B1 type, return it in *PTMP2, otherwise store NULL into it. We
2830 assume we are out of SSA so the returned register does not have its
2831 definition set. */
2833 hsa_op_reg *
2834 hsa_spill_in (hsa_insn_basic *insn, hsa_op_reg *spill_reg, hsa_op_reg **ptmp2)
2836 hsa_symbol *spill_sym = spill_reg->m_spill_sym;
2837 hsa_op_reg *reg = new hsa_op_reg (spill_sym->m_type);
2838 hsa_op_address *addr = new hsa_op_address (spill_sym);
2840 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, spill_sym->m_type,
2841 reg, addr);
2842 hsa_insert_insn_before (mem, insn);
2844 *ptmp2 = NULL;
2845 if (spill_reg->m_type == BRIG_TYPE_B1)
2847 hsa_insn_basic *cvtinsn;
2848 *ptmp2 = reg;
2849 reg = new hsa_op_reg (spill_reg->m_type);
2851 cvtinsn = new hsa_insn_cvt (reg, *ptmp2);
2852 hsa_insert_insn_before (cvtinsn, insn);
2854 return reg;
2857 /* Append after INSN a store to spill symbol of SPILL_REG. Return the register
2858 from which we stored. If this required another register to convert to a B1
2859 type, return it in *PTMP2, otherwise store NULL into it. We assume we are
2860 out of SSA so the returned register does not have its use updated. */
2862 hsa_op_reg *
2863 hsa_spill_out (hsa_insn_basic *insn, hsa_op_reg *spill_reg, hsa_op_reg **ptmp2)
2865 hsa_symbol *spill_sym = spill_reg->m_spill_sym;
2866 hsa_op_reg *reg = new hsa_op_reg (spill_sym->m_type);
2867 hsa_op_address *addr = new hsa_op_address (spill_sym);
2868 hsa_op_reg *returnreg;
2870 *ptmp2 = NULL;
2871 returnreg = reg;
2872 if (spill_reg->m_type == BRIG_TYPE_B1)
2874 hsa_insn_basic *cvtinsn;
2875 *ptmp2 = new hsa_op_reg (spill_sym->m_type);
2876 reg->m_type = spill_reg->m_type;
2878 cvtinsn = new hsa_insn_cvt (*ptmp2, returnreg);
2879 hsa_append_insn_after (cvtinsn, insn);
2880 insn = cvtinsn;
2881 reg = *ptmp2;
2884 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, spill_sym->m_type, reg,
2885 addr);
2886 hsa_append_insn_after (mem, insn);
2887 return returnreg;
2890 /* Generate a comparison instruction that will compare LHS and RHS with
2891 comparison specified by CODE and put result into register DEST. DEST has to
2892 have its type set already but must not have its definition set yet.
2893 Generated instructions will be added to HBB. */
2895 static void
2896 gen_hsa_cmp_insn_from_gimple (enum tree_code code, tree lhs, tree rhs,
2897 hsa_op_reg *dest, hsa_bb *hbb)
2899 BrigCompareOperation8_t compare;
2901 switch (code)
2903 case LT_EXPR:
2904 compare = BRIG_COMPARE_LT;
2905 break;
2906 case LE_EXPR:
2907 compare = BRIG_COMPARE_LE;
2908 break;
2909 case GT_EXPR:
2910 compare = BRIG_COMPARE_GT;
2911 break;
2912 case GE_EXPR:
2913 compare = BRIG_COMPARE_GE;
2914 break;
2915 case EQ_EXPR:
2916 compare = BRIG_COMPARE_EQ;
2917 break;
2918 case NE_EXPR:
2919 compare = BRIG_COMPARE_NE;
2920 break;
2921 case UNORDERED_EXPR:
2922 compare = BRIG_COMPARE_NAN;
2923 break;
2924 case ORDERED_EXPR:
2925 compare = BRIG_COMPARE_NUM;
2926 break;
2927 case UNLT_EXPR:
2928 compare = BRIG_COMPARE_LTU;
2929 break;
2930 case UNLE_EXPR:
2931 compare = BRIG_COMPARE_LEU;
2932 break;
2933 case UNGT_EXPR:
2934 compare = BRIG_COMPARE_GTU;
2935 break;
2936 case UNGE_EXPR:
2937 compare = BRIG_COMPARE_GEU;
2938 break;
2939 case UNEQ_EXPR:
2940 compare = BRIG_COMPARE_EQU;
2941 break;
2942 case LTGT_EXPR:
2943 compare = BRIG_COMPARE_NEU;
2944 break;
2946 default:
2947 HSA_SORRY_ATV (EXPR_LOCATION (lhs),
2948 "support for HSA does not implement comparison tree "
2949 "code %s\n", get_tree_code_name (code));
2950 return;
2953 /* CMP instruction returns e.g. 0xffffffff (for a 32-bit with integer)
2954 as a result of comparison. */
2956 BrigType16_t dest_type = hsa_type_integer_p (dest->m_type)
2957 ? (BrigType16_t) BRIG_TYPE_B1 : dest->m_type;
2959 hsa_insn_cmp *cmp = new hsa_insn_cmp (compare, dest_type);
2960 cmp->set_op (1, hsa_reg_or_immed_for_gimple_op (lhs, hbb));
2961 cmp->set_op (2, hsa_reg_or_immed_for_gimple_op (rhs, hbb));
2963 hbb->append_insn (cmp);
2964 cmp->set_output_in_type (dest, 0, hbb);
2967 /* Generate an unary instruction with OPCODE and append it to a basic block
2968 HBB. The instruction uses DEST as a destination and OP1
2969 as a single operand. */
2971 static void
2972 gen_hsa_unary_operation (BrigOpcode opcode, hsa_op_reg *dest,
2973 hsa_op_with_type *op1, hsa_bb *hbb)
2975 gcc_checking_assert (dest);
2976 hsa_insn_basic *insn;
2978 if (opcode == BRIG_OPCODE_MOV && hsa_needs_cvt (dest->m_type, op1->m_type))
2979 insn = new hsa_insn_cvt (dest, op1);
2980 else if (opcode == BRIG_OPCODE_FIRSTBIT || opcode == BRIG_OPCODE_LASTBIT)
2981 insn = new hsa_insn_srctype (2, opcode, BRIG_TYPE_U32, op1->m_type, NULL,
2982 op1);
2983 else
2985 insn = new hsa_insn_basic (2, opcode, dest->m_type, dest, op1);
2987 if (opcode == BRIG_OPCODE_ABS || opcode == BRIG_OPCODE_NEG)
2989 /* ABS and NEG only exist in _s form :-/ */
2990 if (insn->m_type == BRIG_TYPE_U32)
2991 insn->m_type = BRIG_TYPE_S32;
2992 else if (insn->m_type == BRIG_TYPE_U64)
2993 insn->m_type = BRIG_TYPE_S64;
2997 hbb->append_insn (insn);
2999 if (opcode == BRIG_OPCODE_FIRSTBIT || opcode == BRIG_OPCODE_LASTBIT)
3000 insn->set_output_in_type (dest, 0, hbb);
3003 /* Generate a binary instruction with OPCODE and append it to a basic block
3004 HBB. The instruction uses DEST as a destination and operands OP1
3005 and OP2. */
3007 static void
3008 gen_hsa_binary_operation (int opcode, hsa_op_reg *dest,
3009 hsa_op_base *op1, hsa_op_base *op2, hsa_bb *hbb)
3011 gcc_checking_assert (dest);
3013 if ((opcode == BRIG_OPCODE_SHL || opcode == BRIG_OPCODE_SHR)
3014 && is_a <hsa_op_immed *> (op2))
3016 hsa_op_immed *i = dyn_cast <hsa_op_immed *> (op2);
3017 i->set_type (BRIG_TYPE_U32);
3019 if ((opcode == BRIG_OPCODE_OR
3020 || opcode == BRIG_OPCODE_XOR
3021 || opcode == BRIG_OPCODE_AND)
3022 && is_a <hsa_op_immed *> (op2))
3024 hsa_op_immed *i = dyn_cast <hsa_op_immed *> (op2);
3025 i->set_type (hsa_uint_for_bitsize (hsa_type_bit_size (i->m_type)));
3028 hsa_insn_basic *insn = new hsa_insn_basic (3, opcode, dest->m_type, dest,
3029 op1, op2);
3030 hbb->append_insn (insn);
3033 /* Generate HSA instructions for a single assignment. HBB is the basic block
3034 they will be appended to. */
3036 static void
3037 gen_hsa_insns_for_operation_assignment (gimple *assign, hsa_bb *hbb)
3039 tree_code code = gimple_assign_rhs_code (assign);
3040 gimple_rhs_class rhs_class = get_gimple_rhs_class (gimple_expr_code (assign));
3042 tree lhs = gimple_assign_lhs (assign);
3043 tree rhs1 = gimple_assign_rhs1 (assign);
3044 tree rhs2 = gimple_assign_rhs2 (assign);
3045 tree rhs3 = gimple_assign_rhs3 (assign);
3047 BrigOpcode opcode;
3049 switch (code)
3051 CASE_CONVERT:
3052 case FLOAT_EXPR:
3053 /* The opcode is changed to BRIG_OPCODE_CVT if BRIG types
3054 needs a conversion. */
3055 opcode = BRIG_OPCODE_MOV;
3056 break;
3058 case PLUS_EXPR:
3059 case POINTER_PLUS_EXPR:
3060 opcode = BRIG_OPCODE_ADD;
3061 break;
3062 case MINUS_EXPR:
3063 opcode = BRIG_OPCODE_SUB;
3064 break;
3065 case MULT_EXPR:
3066 opcode = BRIG_OPCODE_MUL;
3067 break;
3068 case MULT_HIGHPART_EXPR:
3069 opcode = BRIG_OPCODE_MULHI;
3070 break;
3071 case RDIV_EXPR:
3072 case TRUNC_DIV_EXPR:
3073 case EXACT_DIV_EXPR:
3074 opcode = BRIG_OPCODE_DIV;
3075 break;
3076 case CEIL_DIV_EXPR:
3077 case FLOOR_DIV_EXPR:
3078 case ROUND_DIV_EXPR:
3079 HSA_SORRY_AT (gimple_location (assign),
3080 "support for HSA does not implement CEIL_DIV_EXPR, "
3081 "FLOOR_DIV_EXPR or ROUND_DIV_EXPR");
3082 return;
3083 case TRUNC_MOD_EXPR:
3084 opcode = BRIG_OPCODE_REM;
3085 break;
3086 case CEIL_MOD_EXPR:
3087 case FLOOR_MOD_EXPR:
3088 case ROUND_MOD_EXPR:
3089 HSA_SORRY_AT (gimple_location (assign),
3090 "support for HSA does not implement CEIL_MOD_EXPR, "
3091 "FLOOR_MOD_EXPR or ROUND_MOD_EXPR");
3092 return;
3093 case NEGATE_EXPR:
3094 opcode = BRIG_OPCODE_NEG;
3095 break;
3096 case MIN_EXPR:
3097 opcode = BRIG_OPCODE_MIN;
3098 break;
3099 case MAX_EXPR:
3100 opcode = BRIG_OPCODE_MAX;
3101 break;
3102 case ABS_EXPR:
3103 opcode = BRIG_OPCODE_ABS;
3104 break;
3105 case LSHIFT_EXPR:
3106 opcode = BRIG_OPCODE_SHL;
3107 break;
3108 case RSHIFT_EXPR:
3109 opcode = BRIG_OPCODE_SHR;
3110 break;
3111 case LROTATE_EXPR:
3112 case RROTATE_EXPR:
3114 hsa_insn_basic *insn = NULL;
3115 int code1 = code == LROTATE_EXPR ? BRIG_OPCODE_SHL : BRIG_OPCODE_SHR;
3116 int code2 = code != LROTATE_EXPR ? BRIG_OPCODE_SHL : BRIG_OPCODE_SHR;
3117 BrigType16_t btype = hsa_type_for_scalar_tree_type (TREE_TYPE (lhs),
3118 true);
3120 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
3121 hsa_op_reg *op1 = new hsa_op_reg (btype);
3122 hsa_op_reg *op2 = new hsa_op_reg (btype);
3123 hsa_op_with_type *shift1 = hsa_reg_or_immed_for_gimple_op (rhs2, hbb);
3125 tree type = TREE_TYPE (rhs2);
3126 unsigned HOST_WIDE_INT bitsize = TREE_INT_CST_LOW (TYPE_SIZE (type));
3128 hsa_op_with_type *shift2 = NULL;
3129 if (TREE_CODE (rhs2) == INTEGER_CST)
3130 shift2 = new hsa_op_immed (bitsize - tree_to_uhwi (rhs2),
3131 BRIG_TYPE_U32);
3132 else if (TREE_CODE (rhs2) == SSA_NAME)
3134 hsa_op_reg *s = hsa_cfun->reg_for_gimple_ssa (rhs2);
3135 hsa_op_reg *d = new hsa_op_reg (s->m_type);
3136 hsa_op_immed *size_imm = new hsa_op_immed (bitsize, BRIG_TYPE_U32);
3138 insn = new hsa_insn_basic (3, BRIG_OPCODE_SUB, d->m_type,
3139 d, s, size_imm);
3140 hbb->append_insn (insn);
3142 shift2 = d;
3144 else
3145 gcc_unreachable ();
3147 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3148 gen_hsa_binary_operation (code1, op1, src, shift1, hbb);
3149 gen_hsa_binary_operation (code2, op2, src, shift2, hbb);
3150 gen_hsa_binary_operation (BRIG_OPCODE_OR, dest, op1, op2, hbb);
3152 return;
3154 case BIT_IOR_EXPR:
3155 opcode = BRIG_OPCODE_OR;
3156 break;
3157 case BIT_XOR_EXPR:
3158 opcode = BRIG_OPCODE_XOR;
3159 break;
3160 case BIT_AND_EXPR:
3161 opcode = BRIG_OPCODE_AND;
3162 break;
3163 case BIT_NOT_EXPR:
3164 opcode = BRIG_OPCODE_NOT;
3165 break;
3166 case FIX_TRUNC_EXPR:
3168 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3169 hsa_op_with_type *v = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
3171 if (hsa_needs_cvt (dest->m_type, v->m_type))
3173 hsa_op_reg *tmp = new hsa_op_reg (v->m_type);
3175 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_TRUNC,
3176 tmp->m_type, tmp, v);
3177 hbb->append_insn (insn);
3179 hsa_insn_basic *cvtinsn = new hsa_insn_cvt (dest, tmp);
3180 hbb->append_insn (cvtinsn);
3182 else
3184 hsa_insn_basic *insn = new hsa_insn_basic (2, BRIG_OPCODE_TRUNC,
3185 dest->m_type, dest, v);
3186 hbb->append_insn (insn);
3189 return;
3191 opcode = BRIG_OPCODE_TRUNC;
3192 break;
3194 case LT_EXPR:
3195 case LE_EXPR:
3196 case GT_EXPR:
3197 case GE_EXPR:
3198 case EQ_EXPR:
3199 case NE_EXPR:
3200 case UNORDERED_EXPR:
3201 case ORDERED_EXPR:
3202 case UNLT_EXPR:
3203 case UNLE_EXPR:
3204 case UNGT_EXPR:
3205 case UNGE_EXPR:
3206 case UNEQ_EXPR:
3207 case LTGT_EXPR:
3209 hsa_op_reg *dest
3210 = hsa_cfun->reg_for_gimple_ssa (gimple_assign_lhs (assign));
3212 gen_hsa_cmp_insn_from_gimple (code, rhs1, rhs2, dest, hbb);
3213 return;
3215 case COND_EXPR:
3217 hsa_op_reg *dest
3218 = hsa_cfun->reg_for_gimple_ssa (gimple_assign_lhs (assign));
3219 hsa_op_with_type *ctrl = NULL;
3220 tree cond = rhs1;
3222 if (CONSTANT_CLASS_P (cond) || TREE_CODE (cond) == SSA_NAME)
3223 ctrl = hsa_reg_or_immed_for_gimple_op (cond, hbb);
3224 else
3226 hsa_op_reg *r = new hsa_op_reg (BRIG_TYPE_B1);
3228 gen_hsa_cmp_insn_from_gimple (TREE_CODE (cond),
3229 TREE_OPERAND (cond, 0),
3230 TREE_OPERAND (cond, 1),
3231 r, hbb);
3233 ctrl = r;
3236 hsa_op_with_type *rhs2_reg = hsa_reg_or_immed_for_gimple_op (rhs2, hbb);
3237 hsa_op_with_type *rhs3_reg = hsa_reg_or_immed_for_gimple_op (rhs3, hbb);
3239 BrigType16_t btype = hsa_bittype_for_type (dest->m_type);
3240 hsa_op_reg *tmp = new hsa_op_reg (btype);
3242 rhs2_reg->m_type = btype;
3243 rhs3_reg->m_type = btype;
3245 hsa_insn_basic *insn
3246 = new hsa_insn_basic (4, BRIG_OPCODE_CMOV, tmp->m_type, tmp, ctrl,
3247 rhs2_reg, rhs3_reg);
3249 hbb->append_insn (insn);
3251 /* As operands of a CMOV insn must be Bx types, we have to emit
3252 a conversion insn. */
3253 hsa_insn_basic *mov = new hsa_insn_basic (2, BRIG_OPCODE_MOV,
3254 dest->m_type, dest, tmp);
3255 hbb->append_insn (mov);
3257 return;
3259 case COMPLEX_EXPR:
3261 hsa_op_reg *dest
3262 = hsa_cfun->reg_for_gimple_ssa (gimple_assign_lhs (assign));
3263 hsa_op_with_type *rhs1_reg = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
3264 hsa_op_with_type *rhs2_reg = hsa_reg_or_immed_for_gimple_op (rhs2, hbb);
3266 if (hsa_seen_error ())
3267 return;
3269 BrigType16_t src_type = hsa_bittype_for_type (rhs1_reg->m_type);
3270 rhs1_reg = rhs1_reg->get_in_type (src_type, hbb);
3271 rhs2_reg = rhs2_reg->get_in_type (src_type, hbb);
3273 hsa_insn_packed *insn
3274 = new hsa_insn_packed (3, BRIG_OPCODE_COMBINE, dest->m_type, src_type,
3275 dest, rhs1_reg, rhs2_reg);
3276 hbb->append_insn (insn);
3278 return;
3280 default:
3281 /* Implement others as we come across them. */
3282 HSA_SORRY_ATV (gimple_location (assign),
3283 "support for HSA does not implement operation %s",
3284 get_tree_code_name (code));
3285 return;
3289 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (gimple_assign_lhs (assign));
3291 hsa_op_with_type *op1 = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
3292 hsa_op_with_type *op2 = rhs2 != NULL_TREE ?
3293 hsa_reg_or_immed_for_gimple_op (rhs2, hbb) : NULL;
3295 if (hsa_seen_error ())
3296 return;
3298 switch (rhs_class)
3300 case GIMPLE_TERNARY_RHS:
3301 gcc_unreachable ();
3302 return;
3304 /* Fall through */
3305 case GIMPLE_BINARY_RHS:
3306 gen_hsa_binary_operation (opcode, dest, op1, op2, hbb);
3307 break;
3308 /* Fall through */
3309 case GIMPLE_UNARY_RHS:
3310 gen_hsa_unary_operation (opcode, dest, op1, hbb);
3311 break;
3312 default:
3313 gcc_unreachable ();
3317 /* Generate HSA instructions for a given gimple condition statement COND.
3318 Instructions will be appended to HBB, which also needs to be the
3319 corresponding structure to the basic_block of COND. */
3321 static void
3322 gen_hsa_insns_for_cond_stmt (gimple *cond, hsa_bb *hbb)
3324 hsa_op_reg *ctrl = new hsa_op_reg (BRIG_TYPE_B1);
3325 hsa_insn_br *cbr;
3327 gen_hsa_cmp_insn_from_gimple (gimple_cond_code (cond),
3328 gimple_cond_lhs (cond),
3329 gimple_cond_rhs (cond),
3330 ctrl, hbb);
3332 cbr = new hsa_insn_br (ctrl);
3333 hbb->append_insn (cbr);
3336 /* Maximum number of elements in a jump table for an HSA SBR instruction. */
3338 #define HSA_MAXIMUM_SBR_LABELS 16
3340 /* Return lowest value of a switch S that is handled in a non-default
3341 label. */
3343 static tree
3344 get_switch_low (gswitch *s)
3346 unsigned labels = gimple_switch_num_labels (s);
3347 gcc_checking_assert (labels >= 1);
3349 return CASE_LOW (gimple_switch_label (s, 1));
3352 /* Return highest value of a switch S that is handled in a non-default
3353 label. */
3355 static tree
3356 get_switch_high (gswitch *s)
3358 unsigned labels = gimple_switch_num_labels (s);
3360 /* Compare last label to maximum number of labels. */
3361 tree label = gimple_switch_label (s, labels - 1);
3362 tree low = CASE_LOW (label);
3363 tree high = CASE_HIGH (label);
3365 return high != NULL_TREE ? high : low;
3368 static tree
3369 get_switch_size (gswitch *s)
3371 return int_const_binop (MINUS_EXPR, get_switch_high (s), get_switch_low (s));
3374 /* Generate HSA instructions for a given gimple switch.
3375 Instructions will be appended to HBB. */
3377 static void
3378 gen_hsa_insns_for_switch_stmt (gswitch *s, hsa_bb *hbb)
3380 function *func = DECL_STRUCT_FUNCTION (current_function_decl);
3381 tree index_tree = gimple_switch_index (s);
3382 tree lowest = get_switch_low (s);
3384 hsa_op_reg *index = hsa_cfun->reg_for_gimple_ssa (index_tree);
3385 hsa_op_reg *sub_index = new hsa_op_reg (index->m_type);
3386 hbb->append_insn (new hsa_insn_basic (3, BRIG_OPCODE_SUB, sub_index->m_type,
3387 sub_index, index,
3388 new hsa_op_immed (lowest)));
3390 hsa_op_base *tmp = sub_index->get_in_type (BRIG_TYPE_U64, hbb);
3391 sub_index = as_a <hsa_op_reg *> (tmp);
3392 unsigned labels = gimple_switch_num_labels (s);
3393 unsigned HOST_WIDE_INT size = tree_to_uhwi (get_switch_size (s));
3395 hsa_insn_sbr *sbr = new hsa_insn_sbr (sub_index, size + 1);
3396 tree default_label = gimple_switch_default_label (s);
3397 basic_block default_label_bb = label_to_block_fn (func,
3398 CASE_LABEL (default_label));
3400 sbr->m_default_bb = default_label_bb;
3402 /* Prepare array with default label destination. */
3403 for (unsigned HOST_WIDE_INT i = 0; i <= size; i++)
3404 sbr->m_jump_table.safe_push (default_label_bb);
3406 /* Iterate all labels and fill up the jump table. */
3407 for (unsigned i = 1; i < labels; i++)
3409 tree label = gimple_switch_label (s, i);
3410 basic_block bb = label_to_block_fn (func, CASE_LABEL (label));
3412 unsigned HOST_WIDE_INT sub_low
3413 = tree_to_uhwi (int_const_binop (MINUS_EXPR, CASE_LOW (label), lowest));
3415 unsigned HOST_WIDE_INT sub_high = sub_low;
3416 tree high = CASE_HIGH (label);
3417 if (high != NULL)
3418 sub_high = tree_to_uhwi (int_const_binop (MINUS_EXPR, high, lowest));
3420 for (unsigned HOST_WIDE_INT j = sub_low; j <= sub_high; j++)
3421 sbr->m_jump_table[j] = bb;
3424 hbb->append_insn (sbr);
3427 /* Verify that the function DECL can be handled by HSA. */
3429 static void
3430 verify_function_arguments (tree decl)
3432 if (DECL_STATIC_CHAIN (decl))
3434 HSA_SORRY_ATV (EXPR_LOCATION (decl),
3435 "HSA does not support nested functions: %D", decl);
3436 return;
3438 else if (!TYPE_ARG_TYPES (TREE_TYPE (decl)))
3440 HSA_SORRY_ATV (EXPR_LOCATION (decl),
3441 "HSA does not support functions with variadic arguments "
3442 "(or unknown return type): %D", decl);
3443 return;
3447 /* Return BRIG type for FORMAL_ARG_TYPE. If the formal argument type is NULL,
3448 return ACTUAL_ARG_TYPE. */
3450 static BrigType16_t
3451 get_format_argument_type (tree formal_arg_type, BrigType16_t actual_arg_type)
3453 if (formal_arg_type == NULL)
3454 return actual_arg_type;
3456 BrigType16_t decl_type
3457 = hsa_type_for_scalar_tree_type (formal_arg_type, false);
3458 return mem_type_for_type (decl_type);
3461 /* Generate HSA instructions for a direct call instruction.
3462 Instructions will be appended to HBB, which also needs to be the
3463 corresponding structure to the basic_block of STMT. */
3465 static void
3466 gen_hsa_insns_for_direct_call (gimple *stmt, hsa_bb *hbb)
3468 tree decl = gimple_call_fndecl (stmt);
3469 verify_function_arguments (decl);
3470 if (hsa_seen_error ())
3471 return;
3473 hsa_insn_call *call_insn = new hsa_insn_call (decl);
3474 hsa_cfun->m_called_functions.safe_push (call_insn->m_called_function);
3476 /* Argument block start. */
3477 hsa_insn_arg_block *arg_start
3478 = new hsa_insn_arg_block (BRIG_KIND_DIRECTIVE_ARG_BLOCK_START, call_insn);
3479 hbb->append_insn (arg_start);
3481 tree parm_type_chain = TYPE_ARG_TYPES (gimple_call_fntype (stmt));
3483 /* Preparation of arguments that will be passed to function. */
3484 const unsigned args = gimple_call_num_args (stmt);
3485 for (unsigned i = 0; i < args; ++i)
3487 tree parm = gimple_call_arg (stmt, (int)i);
3488 tree parm_decl_type = parm_type_chain != NULL_TREE
3489 ? TREE_VALUE (parm_type_chain) : NULL_TREE;
3490 hsa_op_address *addr;
3492 if (AGGREGATE_TYPE_P (TREE_TYPE (parm)))
3494 addr = gen_hsa_addr_for_arg (TREE_TYPE (parm), i);
3495 hsa_op_address *src = gen_hsa_addr (parm, hbb);
3496 gen_hsa_memory_copy (hbb, addr, src,
3497 addr->m_symbol->total_byte_size ());
3499 else
3501 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (parm, hbb);
3503 if (parm_decl_type != NULL && AGGREGATE_TYPE_P (parm_decl_type))
3505 HSA_SORRY_AT (gimple_location (stmt),
3506 "support for HSA does not implement an aggregate "
3507 "formal argument in a function call, while actual "
3508 "argument is not an aggregate");
3509 return;
3512 BrigType16_t formal_arg_type
3513 = get_format_argument_type (parm_decl_type, src->m_type);
3514 if (hsa_seen_error ())
3515 return;
3517 if (src->m_type != formal_arg_type)
3518 src = src->get_in_type (formal_arg_type, hbb);
3520 addr
3521 = gen_hsa_addr_for_arg (parm_decl_type != NULL_TREE ?
3522 parm_decl_type: TREE_TYPE (parm), i);
3523 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, formal_arg_type,
3524 src, addr);
3526 hbb->append_insn (mem);
3529 call_insn->m_input_args.safe_push (addr->m_symbol);
3530 if (parm_type_chain)
3531 parm_type_chain = TREE_CHAIN (parm_type_chain);
3534 call_insn->m_args_code_list = new hsa_op_code_list (args);
3535 hbb->append_insn (call_insn);
3537 tree result_type = TREE_TYPE (TREE_TYPE (decl));
3539 tree result = gimple_call_lhs (stmt);
3540 hsa_insn_mem *result_insn = NULL;
3541 if (!VOID_TYPE_P (result_type))
3543 hsa_op_address *addr = gen_hsa_addr_for_arg (result_type, -1);
3545 /* Even if result of a function call is unused, we have to emit
3546 declaration for the result. */
3547 if (result)
3549 tree lhs_type = TREE_TYPE (result);
3551 if (hsa_seen_error ())
3552 return;
3554 if (AGGREGATE_TYPE_P (lhs_type))
3556 hsa_op_address *result_addr = gen_hsa_addr (result, hbb);
3557 gen_hsa_memory_copy (hbb, result_addr, addr,
3558 addr->m_symbol->total_byte_size ());
3560 else
3562 BrigType16_t mtype
3563 = mem_type_for_type (hsa_type_for_scalar_tree_type (lhs_type,
3564 false));
3566 hsa_op_reg *dst = hsa_cfun->reg_for_gimple_ssa (result);
3567 result_insn = new hsa_insn_mem (BRIG_OPCODE_LD, mtype, dst, addr);
3568 hbb->append_insn (result_insn);
3572 call_insn->m_output_arg = addr->m_symbol;
3573 call_insn->m_result_code_list = new hsa_op_code_list (1);
3575 else
3577 if (result)
3579 HSA_SORRY_AT (gimple_location (stmt),
3580 "support for HSA does not implement an assignment of "
3581 "return value from a void function");
3582 return;
3585 call_insn->m_result_code_list = new hsa_op_code_list (0);
3588 /* Argument block end. */
3589 hsa_insn_arg_block *arg_end
3590 = new hsa_insn_arg_block (BRIG_KIND_DIRECTIVE_ARG_BLOCK_END, call_insn);
3591 hbb->append_insn (arg_end);
3594 /* Generate HSA instructions for a direct call of an internal fn.
3595 Instructions will be appended to HBB, which also needs to be the
3596 corresponding structure to the basic_block of STMT. */
3598 static void
3599 gen_hsa_insns_for_call_of_internal_fn (gimple *stmt, hsa_bb *hbb)
3601 tree lhs = gimple_call_lhs (stmt);
3602 if (!lhs)
3603 return;
3605 tree lhs_type = TREE_TYPE (lhs);
3606 tree rhs1 = gimple_call_arg (stmt, 0);
3607 tree rhs1_type = TREE_TYPE (rhs1);
3608 enum internal_fn fn = gimple_call_internal_fn (stmt);
3609 hsa_internal_fn *ifn
3610 = new hsa_internal_fn (fn, tree_to_uhwi (TYPE_SIZE (rhs1_type)));
3611 hsa_insn_call *call_insn = new hsa_insn_call (ifn);
3613 gcc_checking_assert (FLOAT_TYPE_P (rhs1_type));
3615 if (!hsa_emitted_internal_decls->find (call_insn->m_called_internal_fn))
3616 hsa_cfun->m_called_internal_fns.safe_push (call_insn->m_called_internal_fn);
3618 hsa_insn_arg_block *arg_start
3619 = new hsa_insn_arg_block (BRIG_KIND_DIRECTIVE_ARG_BLOCK_START, call_insn);
3620 hbb->append_insn (arg_start);
3622 unsigned num_args = gimple_call_num_args (stmt);
3624 /* Function arguments. */
3625 for (unsigned i = 0; i < num_args; i++)
3627 tree parm = gimple_call_arg (stmt, (int)i);
3628 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (parm, hbb);
3630 hsa_op_address *addr = gen_hsa_addr_for_arg (TREE_TYPE (parm), i);
3631 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, src->m_type,
3632 src, addr);
3634 call_insn->m_input_args.safe_push (addr->m_symbol);
3635 hbb->append_insn (mem);
3638 call_insn->m_args_code_list = new hsa_op_code_list (num_args);
3639 hbb->append_insn (call_insn);
3641 /* Assign returned value. */
3642 hsa_op_address *addr = gen_hsa_addr_for_arg (lhs_type, -1);
3644 call_insn->m_output_arg = addr->m_symbol;
3645 call_insn->m_result_code_list = new hsa_op_code_list (1);
3647 /* Argument block end. */
3648 hsa_insn_arg_block *arg_end
3649 = new hsa_insn_arg_block (BRIG_KIND_DIRECTIVE_ARG_BLOCK_END, call_insn);
3650 hbb->append_insn (arg_end);
3653 /* Generate HSA instructions for a return value instruction.
3654 Instructions will be appended to HBB, which also needs to be the
3655 corresponding structure to the basic_block of STMT. */
3657 static void
3658 gen_hsa_insns_for_return (greturn *stmt, hsa_bb *hbb)
3660 tree retval = gimple_return_retval (stmt);
3661 if (retval)
3663 hsa_op_address *addr = new hsa_op_address (hsa_cfun->m_output_arg);
3665 if (AGGREGATE_TYPE_P (TREE_TYPE (retval)))
3667 hsa_op_address *retval_addr = gen_hsa_addr (retval, hbb);
3668 gen_hsa_memory_copy (hbb, addr, retval_addr,
3669 hsa_cfun->m_output_arg->total_byte_size ());
3671 else
3673 BrigType16_t t = hsa_type_for_scalar_tree_type (TREE_TYPE (retval),
3674 false);
3675 BrigType16_t mtype = mem_type_for_type (t);
3677 /* Store of return value. */
3678 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (retval, hbb);
3679 src = src->get_in_type (mtype, hbb);
3680 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, mtype, src,
3681 addr);
3682 hbb->append_insn (mem);
3686 /* HSAIL return instruction emission. */
3687 hsa_insn_basic *ret = new hsa_insn_basic (0, BRIG_OPCODE_RET);
3688 hbb->append_insn (ret);
3691 /* Set OP_INDEX-th operand of the instruction to DEST, as the DEST
3692 can have a different type, conversion instructions are possibly
3693 appended to HBB. */
3695 void
3696 hsa_insn_basic::set_output_in_type (hsa_op_reg *dest, unsigned op_index,
3697 hsa_bb *hbb)
3699 hsa_insn_basic *insn;
3700 gcc_checking_assert (op_output_p (op_index));
3702 if (dest->m_type == m_type)
3704 set_op (op_index, dest);
3705 return;
3708 hsa_op_reg *tmp = new hsa_op_reg (m_type);
3709 set_op (op_index, tmp);
3711 if (hsa_needs_cvt (dest->m_type, m_type))
3712 insn = new hsa_insn_cvt (dest, tmp);
3713 else
3714 insn = new hsa_insn_basic (2, BRIG_OPCODE_MOV, dest->m_type,
3715 dest, tmp->get_in_type (dest->m_type, hbb));
3717 hbb->append_insn (insn);
3720 /* Generate instruction OPCODE to query a property of HSA grid along the
3721 given DIMENSION. Store result into DEST and append the instruction to
3722 HBB. */
3724 static void
3725 query_hsa_grid (hsa_op_reg *dest, BrigType16_t opcode, int dimension,
3726 hsa_bb *hbb)
3728 /* We're using just one-dimensional kernels, so hard-coded
3729 dimension X. */
3730 hsa_op_immed *imm
3731 = new hsa_op_immed (dimension, (BrigKind16_t) BRIG_TYPE_U32);
3732 hsa_insn_basic *insn = new hsa_insn_basic (2, opcode, BRIG_TYPE_U32, NULL,
3733 imm);
3734 hbb->append_insn (insn);
3735 insn->set_output_in_type (dest, 0, hbb);
3738 /* Generate a special HSA-related instruction for gimple STMT.
3739 Instructions are appended to basic block HBB. */
3741 static void
3742 query_hsa_grid (gimple *stmt, BrigOpcode16_t opcode, int dimension,
3743 hsa_bb *hbb)
3745 tree lhs = gimple_call_lhs (dyn_cast <gcall *> (stmt));
3746 if (lhs == NULL_TREE)
3747 return;
3749 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3751 query_hsa_grid (dest, opcode, dimension, hbb);
3754 /* Emit instructions that set hsa_num_threads according to provided VALUE.
3755 Instructions are appended to basic block HBB. */
3757 static void
3758 gen_set_num_threads (tree value, hsa_bb *hbb)
3760 hbb->append_insn (new hsa_insn_comment ("omp_set_num_threads"));
3761 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (value, hbb);
3763 src = src->get_in_type (hsa_num_threads->m_type, hbb);
3764 hsa_op_address *addr = new hsa_op_address (hsa_num_threads);
3766 hsa_insn_basic *basic
3767 = new hsa_insn_mem (BRIG_OPCODE_ST, hsa_num_threads->m_type, src, addr);
3768 hbb->append_insn (basic);
3771 static GTY (()) tree hsa_kernel_dispatch_type = NULL;
3773 /* Return byte offset of a FIELD_NAME in GOMP_hsa_kernel_dispatch which
3774 is defined in plugin-hsa.c. */
3776 static HOST_WIDE_INT
3777 get_hsa_kernel_dispatch_offset (const char *field_name)
3779 if (hsa_kernel_dispatch_type == NULL)
3781 /* Collection of information needed for a dispatch of a kernel from a
3782 kernel. Keep in sync with libgomp's plugin-hsa.c. */
3784 hsa_kernel_dispatch_type = make_node (RECORD_TYPE);
3785 tree id_f1 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3786 get_identifier ("queue"), ptr_type_node);
3787 DECL_CHAIN (id_f1) = NULL_TREE;
3788 tree id_f2 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3789 get_identifier ("omp_data_memory"),
3790 ptr_type_node);
3791 DECL_CHAIN (id_f2) = id_f1;
3792 tree id_f3 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3793 get_identifier ("kernarg_address"),
3794 ptr_type_node);
3795 DECL_CHAIN (id_f3) = id_f2;
3796 tree id_f4 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3797 get_identifier ("object"),
3798 uint64_type_node);
3799 DECL_CHAIN (id_f4) = id_f3;
3800 tree id_f5 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3801 get_identifier ("signal"),
3802 uint64_type_node);
3803 DECL_CHAIN (id_f5) = id_f4;
3804 tree id_f6 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3805 get_identifier ("private_segment_size"),
3806 uint32_type_node);
3807 DECL_CHAIN (id_f6) = id_f5;
3808 tree id_f7 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3809 get_identifier ("group_segment_size"),
3810 uint32_type_node);
3811 DECL_CHAIN (id_f7) = id_f6;
3812 tree id_f8 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3813 get_identifier ("kernel_dispatch_count"),
3814 uint64_type_node);
3815 DECL_CHAIN (id_f8) = id_f7;
3816 tree id_f9 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3817 get_identifier ("debug"),
3818 uint64_type_node);
3819 DECL_CHAIN (id_f9) = id_f8;
3820 tree id_f10 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3821 get_identifier ("omp_level"),
3822 uint64_type_node);
3823 DECL_CHAIN (id_f10) = id_f9;
3824 tree id_f11 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3825 get_identifier ("children_dispatches"),
3826 ptr_type_node);
3827 DECL_CHAIN (id_f11) = id_f10;
3828 tree id_f12 = build_decl (BUILTINS_LOCATION, FIELD_DECL,
3829 get_identifier ("omp_num_threads"),
3830 uint32_type_node);
3831 DECL_CHAIN (id_f12) = id_f11;
3834 finish_builtin_struct (hsa_kernel_dispatch_type, "__hsa_kernel_dispatch",
3835 id_f12, NULL_TREE);
3836 TYPE_ARTIFICIAL (hsa_kernel_dispatch_type) = 1;
3839 for (tree chain = TYPE_FIELDS (hsa_kernel_dispatch_type);
3840 chain != NULL_TREE; chain = TREE_CHAIN (chain))
3841 if (strcmp (field_name, IDENTIFIER_POINTER (DECL_NAME (chain))) == 0)
3842 return int_byte_position (chain);
3844 gcc_unreachable ();
3847 /* Return an HSA register that will contain number of threads for
3848 a future dispatched kernel. Instructions are added to HBB. */
3850 static hsa_op_reg *
3851 gen_num_threads_for_dispatch (hsa_bb *hbb)
3853 /* Step 1) Assign to number of threads:
3854 MIN (HSA_DEFAULT_NUM_THREADS, hsa_num_threads). */
3855 hsa_op_reg *threads = new hsa_op_reg (hsa_num_threads->m_type);
3856 hsa_op_address *addr = new hsa_op_address (hsa_num_threads);
3858 hbb->append_insn (new hsa_insn_mem (BRIG_OPCODE_LD, threads->m_type,
3859 threads, addr));
3861 hsa_op_immed *limit = new hsa_op_immed (HSA_DEFAULT_NUM_THREADS,
3862 BRIG_TYPE_U32);
3863 hsa_op_reg *r = new hsa_op_reg (BRIG_TYPE_B1);
3864 hsa_insn_cmp * cmp
3865 = new hsa_insn_cmp (BRIG_COMPARE_LT, r->m_type, r, threads, limit);
3866 hbb->append_insn (cmp);
3868 BrigType16_t btype = hsa_bittype_for_type (threads->m_type);
3869 hsa_op_reg *tmp = new hsa_op_reg (threads->m_type);
3871 hbb->append_insn (new hsa_insn_basic (4, BRIG_OPCODE_CMOV, btype, tmp, r,
3872 threads, limit));
3874 /* Step 2) If the number is equal to zero,
3875 return shadow->omp_num_threads. */
3876 hsa_op_reg *shadow_reg_ptr = hsa_cfun->get_shadow_reg ();
3878 hsa_op_reg *shadow_thread_count = new hsa_op_reg (BRIG_TYPE_U32);
3879 addr
3880 = new hsa_op_address (shadow_reg_ptr,
3881 get_hsa_kernel_dispatch_offset ("omp_num_threads"));
3882 hsa_insn_basic *basic
3883 = new hsa_insn_mem (BRIG_OPCODE_LD, shadow_thread_count->m_type,
3884 shadow_thread_count, addr);
3885 hbb->append_insn (basic);
3887 hsa_op_reg *tmp2 = new hsa_op_reg (threads->m_type);
3888 r = new hsa_op_reg (BRIG_TYPE_B1);
3889 hsa_op_immed *imm = new hsa_op_immed (0, shadow_thread_count->m_type);
3890 hbb->append_insn (new hsa_insn_cmp (BRIG_COMPARE_EQ, r->m_type, r, tmp, imm));
3891 hbb->append_insn (new hsa_insn_basic (4, BRIG_OPCODE_CMOV, btype, tmp2, r,
3892 shadow_thread_count, tmp));
3894 hsa_op_base *dest = tmp2->get_in_type (BRIG_TYPE_U16, hbb);
3896 return as_a <hsa_op_reg *> (dest);
3900 /* Emit instructions that assign number of teams to lhs of gimple STMT.
3901 Instructions are appended to basic block HBB. */
3903 static void
3904 gen_get_num_teams (gimple *stmt, hsa_bb *hbb)
3906 if (gimple_call_lhs (stmt) == NULL_TREE)
3907 return;
3909 hbb->append_insn (new hsa_insn_comment ("omp_get_num_teams"));
3911 tree lhs = gimple_call_lhs (stmt);
3912 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3913 hsa_op_immed *one = new hsa_op_immed (1, dest->m_type);
3915 hsa_insn_basic *basic
3916 = new hsa_insn_basic (2, BRIG_OPCODE_MOV, dest->m_type, dest, one);
3918 hbb->append_insn (basic);
3921 /* Emit instructions that assign a team number to lhs of gimple STMT.
3922 Instructions are appended to basic block HBB. */
3924 static void
3925 gen_get_team_num (gimple *stmt, hsa_bb *hbb)
3927 if (gimple_call_lhs (stmt) == NULL_TREE)
3928 return;
3930 hbb->append_insn (new hsa_insn_comment ("omp_get_team_num"));
3932 tree lhs = gimple_call_lhs (stmt);
3933 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3934 hsa_op_immed *zero = new hsa_op_immed (0, dest->m_type);
3936 hsa_insn_basic *basic
3937 = new hsa_insn_basic (2, BRIG_OPCODE_MOV, dest->m_type, dest, zero);
3939 hbb->append_insn (basic);
3942 /* Emit instructions that get levels-var ICV to lhs of gimple STMT.
3943 Instructions are appended to basic block HBB. */
3945 static void
3946 gen_get_level (gimple *stmt, hsa_bb *hbb)
3948 if (gimple_call_lhs (stmt) == NULL_TREE)
3949 return;
3951 hbb->append_insn (new hsa_insn_comment ("omp_get_level"));
3953 tree lhs = gimple_call_lhs (stmt);
3954 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3956 hsa_op_reg *shadow_reg_ptr = hsa_cfun->get_shadow_reg ();
3957 if (shadow_reg_ptr == NULL)
3959 HSA_SORRY_AT (gimple_location (stmt),
3960 "support for HSA does not implement omp_get_level called "
3961 "from a function not being inlined within a kernel");
3962 return;
3965 hsa_op_address *addr
3966 = new hsa_op_address (shadow_reg_ptr,
3967 get_hsa_kernel_dispatch_offset ("omp_level"));
3969 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, BRIG_TYPE_U64,
3970 (hsa_op_base *) NULL, addr);
3971 hbb->append_insn (mem);
3972 mem->set_output_in_type (dest, 0, hbb);
3975 /* Emit instruction that implement omp_get_max_threads of gimple STMT. */
3977 static void
3978 gen_get_max_threads (gimple *stmt, hsa_bb *hbb)
3980 tree lhs = gimple_call_lhs (stmt);
3981 if (!lhs)
3982 return;
3984 hbb->append_insn (new hsa_insn_comment ("omp_get_max_threads"));
3986 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
3987 hsa_op_with_type *num_theads_reg = gen_num_threads_for_dispatch (hbb)
3988 ->get_in_type (dest->m_type, hbb);
3989 hsa_build_append_simple_mov (dest, num_theads_reg, hbb);
3992 /* Emit instructions that implement alloca builtin gimple STMT.
3993 Instructions are appended to basic block HBB. */
3995 static void
3996 gen_hsa_alloca (gcall *call, hsa_bb *hbb)
3998 tree lhs = gimple_call_lhs (call);
3999 if (lhs == NULL_TREE)
4000 return;
4002 built_in_function fn = DECL_FUNCTION_CODE (gimple_call_fndecl (call));
4004 gcc_checking_assert (fn == BUILT_IN_ALLOCA
4005 || fn == BUILT_IN_ALLOCA_WITH_ALIGN);
4007 unsigned bit_alignment = 0;
4009 if (fn == BUILT_IN_ALLOCA_WITH_ALIGN)
4011 tree alignment_tree = gimple_call_arg (call, 1);
4012 if (TREE_CODE (alignment_tree) != INTEGER_CST)
4014 HSA_SORRY_ATV (gimple_location (call),
4015 "support for HSA does not implement "
4016 "__builtin_alloca_with_align with a non-constant "
4017 "alignment: %E", alignment_tree);
4020 bit_alignment = tree_to_uhwi (alignment_tree);
4023 tree rhs1 = gimple_call_arg (call, 0);
4024 hsa_op_with_type *size = hsa_reg_or_immed_for_gimple_op (rhs1, hbb)
4025 ->get_in_type (BRIG_TYPE_U32, hbb);
4026 hsa_op_with_type *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4028 hsa_op_reg *tmp
4029 = new hsa_op_reg (hsa_get_segment_addr_type (BRIG_SEGMENT_PRIVATE));
4030 hsa_insn_alloca *a = new hsa_insn_alloca (tmp, size, bit_alignment);
4031 hbb->append_insn (a);
4033 hsa_insn_seg *seg
4034 = new hsa_insn_seg (BRIG_OPCODE_STOF,
4035 hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT),
4036 tmp->m_type, BRIG_SEGMENT_PRIVATE, dest, tmp);
4037 hbb->append_insn (seg);
4040 /* Emit instructions that implement clrsb builtin STMT:
4041 Returns the number of leading redundant sign bits in x, i.e. the number
4042 of bits following the most significant bit that are identical to it.
4043 There are no special cases for 0 or other values.
4044 Instructions are appended to basic block HBB. */
4046 static void
4047 gen_hsa_clrsb (gcall *call, hsa_bb *hbb)
4049 tree lhs = gimple_call_lhs (call);
4050 if (lhs == NULL_TREE)
4051 return;
4053 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4054 tree rhs1 = gimple_call_arg (call, 0);
4055 hsa_op_with_type *arg = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
4056 BrigType16_t bittype = hsa_bittype_for_type (arg->m_type);
4057 unsigned bitsize = tree_to_uhwi (TYPE_SIZE (TREE_TYPE (rhs1)));
4058 gcc_checking_assert (bitsize >= 32);
4060 /* Set true to MOST_SIG if the most significant bit is set to one. */
4061 hsa_op_immed *c = new hsa_op_immed (1ul << (bitsize - 1),
4062 hsa_uint_for_bitsize (bitsize));
4064 hsa_op_reg *and_reg = new hsa_op_reg (bittype);
4065 gen_hsa_binary_operation (BRIG_OPCODE_AND, and_reg, arg, c, hbb);
4067 hsa_op_reg *most_sign = new hsa_op_reg (BRIG_TYPE_B1);
4068 hsa_insn_cmp *cmp
4069 = new hsa_insn_cmp (BRIG_COMPARE_EQ, most_sign->m_type, most_sign,
4070 and_reg, c);
4071 hbb->append_insn (cmp);
4073 /* If the most significant bit is one, negate the input. Otherwise
4074 shift the input value to left by one bit. */
4075 hsa_op_reg *arg_neg = new hsa_op_reg (arg->m_type);
4076 gen_hsa_unary_operation (BRIG_OPCODE_NEG, arg_neg, arg, hbb);
4078 hsa_op_reg *shifted_arg = new hsa_op_reg (arg->m_type);
4079 gen_hsa_binary_operation (BRIG_OPCODE_SHL, shifted_arg, arg,
4080 new hsa_op_immed (1, BRIG_TYPE_U64), hbb);
4082 /* Assign the value that can be used for FIRSTBIT instruction according
4083 to the most significant bit. */
4084 hsa_op_reg *tmp = new hsa_op_reg (bittype);
4085 hsa_insn_basic *cmov
4086 = new hsa_insn_basic (4, BRIG_OPCODE_CMOV, bittype, tmp, most_sign,
4087 arg_neg, shifted_arg);
4088 hbb->append_insn (cmov);
4090 hsa_op_reg *leading_bits = new hsa_op_reg (BRIG_TYPE_S32);
4091 gen_hsa_unary_operation (BRIG_OPCODE_FIRSTBIT, leading_bits,
4092 tmp->get_in_type (hsa_uint_for_bitsize (bitsize),
4093 hbb), hbb);
4095 /* Set flag if the input value is equal to zero. */
4096 hsa_op_reg *is_zero = new hsa_op_reg (BRIG_TYPE_B1);
4097 cmp = new hsa_insn_cmp (BRIG_COMPARE_EQ, is_zero->m_type, is_zero, arg,
4098 new hsa_op_immed (0, arg->m_type));
4099 hbb->append_insn (cmp);
4101 /* Return the number of leading bits, or 31 if the input value is zero. */
4102 cmov = new hsa_insn_basic (4, BRIG_OPCODE_CMOV, BRIG_TYPE_B32, NULL, is_zero,
4103 new hsa_op_immed (31, BRIG_TYPE_U32),
4104 leading_bits->get_in_type (BRIG_TYPE_B32, hbb));
4105 hbb->append_insn (cmov);
4106 cmov->set_output_in_type (dest, 0, hbb);
4109 /* Emit instructions that implement ffs builtin STMT:
4110 Returns one plus the index of the least significant 1-bit of x,
4111 or if x is zero, returns zero.
4112 Instructions are appended to basic block HBB. */
4114 static void
4115 gen_hsa_ffs (gcall *call, hsa_bb *hbb)
4117 tree lhs = gimple_call_lhs (call);
4118 if (lhs == NULL_TREE)
4119 return;
4121 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4123 tree rhs1 = gimple_call_arg (call, 0);
4124 hsa_op_with_type *arg = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
4126 hsa_op_reg *tmp = new hsa_op_reg (BRIG_TYPE_U32);
4127 hsa_insn_srctype *insn = new hsa_insn_srctype (2, BRIG_OPCODE_LASTBIT,
4128 tmp->m_type, arg->m_type,
4129 tmp, arg);
4130 hbb->append_insn (insn);
4132 hsa_insn_basic *addition
4133 = new hsa_insn_basic (3, BRIG_OPCODE_ADD, tmp->m_type, NULL, tmp,
4134 new hsa_op_immed (1, tmp->m_type));
4135 hbb->append_insn (addition);
4136 addition->set_output_in_type (dest, 0, hbb);
4139 static void
4140 gen_hsa_popcount_to_dest (hsa_op_reg *dest, hsa_op_with_type *arg, hsa_bb *hbb)
4142 gcc_checking_assert (hsa_type_integer_p (arg->m_type));
4144 if (hsa_type_bit_size (arg->m_type) < 32)
4145 arg = arg->get_in_type (BRIG_TYPE_B32, hbb);
4147 if (!hsa_btype_p (arg->m_type))
4148 arg = arg->get_in_type (hsa_bittype_for_type (arg->m_type), hbb);
4150 hsa_insn_srctype *popcount
4151 = new hsa_insn_srctype (2, BRIG_OPCODE_POPCOUNT, BRIG_TYPE_U32,
4152 arg->m_type, NULL, arg);
4153 hbb->append_insn (popcount);
4154 popcount->set_output_in_type (dest, 0, hbb);
4157 /* Emit instructions that implement parity builtin STMT:
4158 Returns the parity of x, i.e. the number of 1-bits in x modulo 2.
4159 Instructions are appended to basic block HBB. */
4161 static void
4162 gen_hsa_parity (gcall *call, hsa_bb *hbb)
4164 tree lhs = gimple_call_lhs (call);
4165 if (lhs == NULL_TREE)
4166 return;
4168 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4169 tree rhs1 = gimple_call_arg (call, 0);
4170 hsa_op_with_type *arg = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
4172 hsa_op_reg *popcount = new hsa_op_reg (BRIG_TYPE_U32);
4173 gen_hsa_popcount_to_dest (popcount, arg, hbb);
4175 hsa_insn_basic *insn
4176 = new hsa_insn_basic (3, BRIG_OPCODE_REM, popcount->m_type, NULL, popcount,
4177 new hsa_op_immed (2, popcount->m_type));
4178 hbb->append_insn (insn);
4179 insn->set_output_in_type (dest, 0, hbb);
4182 /* Emit instructions that implement popcount builtin STMT.
4183 Instructions are appended to basic block HBB. */
4185 static void
4186 gen_hsa_popcount (gcall *call, hsa_bb *hbb)
4188 tree lhs = gimple_call_lhs (call);
4189 if (lhs == NULL_TREE)
4190 return;
4192 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4193 tree rhs1 = gimple_call_arg (call, 0);
4194 hsa_op_with_type *arg = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
4196 gen_hsa_popcount_to_dest (dest, arg, hbb);
4199 /* Set VALUE to a shadow kernel debug argument and append a new instruction
4200 to HBB basic block. */
4202 static void
4203 set_debug_value (hsa_bb *hbb, hsa_op_with_type *value)
4205 hsa_op_reg *shadow_reg_ptr = hsa_cfun->get_shadow_reg ();
4206 if (shadow_reg_ptr == NULL)
4207 return;
4209 hsa_op_address *addr
4210 = new hsa_op_address (shadow_reg_ptr,
4211 get_hsa_kernel_dispatch_offset ("debug"));
4212 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_ST, BRIG_TYPE_U64, value,
4213 addr);
4214 hbb->append_insn (mem);
4217 void
4218 omp_simple_builtin::generate (gimple *stmt, hsa_bb *hbb)
4220 if (m_sorry)
4222 if (m_warning_message)
4223 HSA_SORRY_AT (gimple_location (stmt), m_warning_message)
4224 else
4225 HSA_SORRY_ATV (gimple_location (stmt),
4226 "Support for HSA does not implement calls to %s\n",
4227 m_name)
4229 else if (m_warning_message != NULL)
4230 warning_at (gimple_location (stmt), OPT_Whsa, m_warning_message);
4232 if (m_return_value != NULL)
4234 tree lhs = gimple_call_lhs (stmt);
4235 if (!lhs)
4236 return;
4238 hbb->append_insn (new hsa_insn_comment (m_name));
4240 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4241 hsa_op_with_type *op = m_return_value->get_in_type (dest->m_type, hbb);
4242 hsa_build_append_simple_mov (dest, op, hbb);
4246 /* If STMT is a call of a known library function, generate code to perform
4247 it and return true. */
4249 static bool
4250 gen_hsa_insns_for_known_library_call (gimple *stmt, hsa_bb *hbb)
4252 bool handled = false;
4253 const char *name = hsa_get_declaration_name (gimple_call_fndecl (stmt));
4255 char *copy = NULL;
4256 size_t len = strlen (name);
4257 if (len > 0 && name[len - 1] == '_')
4259 copy = XNEWVEC (char, len + 1);
4260 strcpy (copy, name);
4261 copy[len - 1] = '\0';
4262 name = copy;
4265 /* Handle omp_* routines. */
4266 if (strstr (name, "omp_") == name)
4268 hsa_init_simple_builtins ();
4269 omp_simple_builtin *builtin = omp_simple_builtins->get (name);
4270 if (builtin)
4272 builtin->generate (stmt, hbb);
4273 return true;
4276 handled = true;
4277 if (strcmp (name, "omp_set_num_threads") == 0)
4278 gen_set_num_threads (gimple_call_arg (stmt, 0), hbb);
4279 else if (strcmp (name, "omp_get_thread_num") == 0)
4281 hbb->append_insn (new hsa_insn_comment (name));
4282 query_hsa_grid (stmt, BRIG_OPCODE_WORKITEMABSID, 0, hbb);
4284 else if (strcmp (name, "omp_get_num_threads") == 0)
4286 hbb->append_insn (new hsa_insn_comment (name));
4287 query_hsa_grid (stmt, BRIG_OPCODE_GRIDSIZE, 0, hbb);
4289 else if (strcmp (name, "omp_get_num_teams") == 0)
4290 gen_get_num_teams (stmt, hbb);
4291 else if (strcmp (name, "omp_get_team_num") == 0)
4292 gen_get_team_num (stmt, hbb);
4293 else if (strcmp (name, "omp_get_level") == 0)
4294 gen_get_level (stmt, hbb);
4295 else if (strcmp (name, "omp_get_active_level") == 0)
4296 gen_get_level (stmt, hbb);
4297 else if (strcmp (name, "omp_in_parallel") == 0)
4298 gen_get_level (stmt, hbb);
4299 else if (strcmp (name, "omp_get_max_threads") == 0)
4300 gen_get_max_threads (stmt, hbb);
4301 else
4302 handled = false;
4304 if (handled)
4306 if (copy)
4307 free (copy);
4308 return true;
4312 if (strcmp (name, "__hsa_set_debug_value") == 0)
4314 handled = true;
4315 if (hsa_cfun->has_shadow_reg_p ())
4317 tree rhs1 = gimple_call_arg (stmt, 0);
4318 hsa_op_with_type *src = hsa_reg_or_immed_for_gimple_op (rhs1, hbb);
4320 src = src->get_in_type (BRIG_TYPE_U64, hbb);
4321 set_debug_value (hbb, src);
4325 if (copy)
4326 free (copy);
4327 return handled;
4330 /* Helper functions to create a single unary HSA operations out of calls to
4331 builtins. OPCODE is the HSA operation to be generated. STMT is a gimple
4332 call to a builtin. HBB is the HSA BB to which the instruction should be
4333 added. Note that nothing will be created if STMT does not have a LHS. */
4335 static void
4336 gen_hsa_unaryop_for_builtin (BrigOpcode opcode, gimple *stmt, hsa_bb *hbb)
4338 tree lhs = gimple_call_lhs (stmt);
4339 if (!lhs)
4340 return;
4341 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4342 hsa_op_with_type *op
4343 = hsa_reg_or_immed_for_gimple_op (gimple_call_arg (stmt, 0), hbb);
4344 gen_hsa_unary_operation (opcode, dest, op, hbb);
4347 /* Helper functions to create a call to standard library if LHS of the
4348 STMT is used. HBB is the HSA BB to which the instruction should be
4349 added. */
4351 static void
4352 gen_hsa_unaryop_builtin_call (gimple *stmt, hsa_bb *hbb)
4354 tree lhs = gimple_call_lhs (stmt);
4355 if (!lhs)
4356 return;
4358 if (gimple_call_internal_p (stmt))
4359 gen_hsa_insns_for_call_of_internal_fn (stmt, hbb);
4360 else
4361 gen_hsa_insns_for_direct_call (stmt, hbb);
4364 /* Helper functions to create a single unary HSA operations out of calls to
4365 builtins (if unsafe math optimizations are enable). Otherwise, create
4366 a call to standard library function.
4367 OPCODE is the HSA operation to be generated. STMT is a gimple
4368 call to a builtin. HBB is the HSA BB to which the instruction should be
4369 added. Note that nothing will be created if STMT does not have a LHS. */
4371 static void
4372 gen_hsa_unaryop_or_call_for_builtin (BrigOpcode opcode, gimple *stmt,
4373 hsa_bb *hbb)
4375 if (flag_unsafe_math_optimizations)
4376 gen_hsa_unaryop_for_builtin (opcode, stmt, hbb);
4377 else
4378 gen_hsa_unaryop_builtin_call (stmt, hbb);
4381 /* Generate HSA address corresponding to a value VAL (as opposed to a memory
4382 reference tree), for example an SSA_NAME or an ADDR_EXPR. HBB is the HSA BB
4383 to which the instruction should be added. */
4385 static hsa_op_address *
4386 get_address_from_value (tree val, hsa_bb *hbb)
4388 switch (TREE_CODE (val))
4390 case SSA_NAME:
4392 BrigType16_t addrtype = hsa_get_segment_addr_type (BRIG_SEGMENT_FLAT);
4393 hsa_op_base *reg
4394 = hsa_cfun->reg_for_gimple_ssa (val)->get_in_type (addrtype, hbb);
4395 return new hsa_op_address (NULL, as_a <hsa_op_reg *> (reg), 0);
4397 case ADDR_EXPR:
4398 return gen_hsa_addr (TREE_OPERAND (val, 0), hbb);
4400 case INTEGER_CST:
4401 if (tree_fits_shwi_p (val))
4402 return new hsa_op_address (NULL, NULL, tree_to_shwi (val));
4403 /* Otherwise fall-through */
4405 default:
4406 HSA_SORRY_ATV (EXPR_LOCATION (val),
4407 "support for HSA does not implement memory access to %E",
4408 val);
4409 return new hsa_op_address (NULL, NULL, 0);
4413 /* Return string for MEMMODEL. */
4415 static const char *
4416 get_memory_order_name (unsigned memmodel)
4418 switch (memmodel & MEMMODEL_BASE_MASK)
4420 case MEMMODEL_RELAXED:
4421 return "relaxed";
4422 case MEMMODEL_CONSUME:
4423 return "consume";
4424 case MEMMODEL_ACQUIRE:
4425 return "acquire";
4426 case MEMMODEL_RELEASE:
4427 return "release";
4428 case MEMMODEL_ACQ_REL:
4429 return "acq_rel";
4430 case MEMMODEL_SEQ_CST:
4431 return "seq_cst";
4432 default:
4433 return NULL;
4437 /* Return memory order according to predefined __atomic memory model
4438 constants. LOCATION is provided to locate the problematic statement. */
4440 static BrigMemoryOrder
4441 get_memory_order (unsigned memmodel, location_t location)
4443 switch (memmodel & MEMMODEL_BASE_MASK)
4445 case MEMMODEL_RELAXED:
4446 return BRIG_MEMORY_ORDER_RELAXED;
4447 case MEMMODEL_CONSUME:
4448 /* HSA does not have an equivalent, but we can use the slightly stronger
4449 ACQUIRE. */
4450 case MEMMODEL_ACQUIRE:
4451 return BRIG_MEMORY_ORDER_SC_ACQUIRE;
4452 case MEMMODEL_RELEASE:
4453 return BRIG_MEMORY_ORDER_SC_RELEASE;
4454 case MEMMODEL_ACQ_REL:
4455 case MEMMODEL_SEQ_CST:
4456 /* Callers implementing a simple load or store need to remove the release
4457 or acquire part respectively. */
4458 return BRIG_MEMORY_ORDER_SC_ACQUIRE_RELEASE;
4459 default:
4461 const char *mmname = get_memory_order_name (memmodel);
4462 HSA_SORRY_ATV (location,
4463 "support for HSA does not implement the specified "
4464 " memory model%s %s",
4465 mmname ? ": " : "", mmname ? mmname : "");
4466 return BRIG_MEMORY_ORDER_NONE;
4471 /* Helper function to create an HSA atomic binary operation instruction out of
4472 calls to atomic builtins. RET_ORIG is true if the built-in is the variant
4473 that return s the value before applying operation, and false if it should
4474 return the value after applying the operation (if it returns value at all).
4475 ACODE is the atomic operation code, STMT is a gimple call to a builtin. HBB
4476 is the HSA BB to which the instruction should be added. */
4478 static void
4479 gen_hsa_ternary_atomic_for_builtin (bool ret_orig,
4480 enum BrigAtomicOperation acode,
4481 gimple *stmt,
4482 hsa_bb *hbb)
4484 tree lhs = gimple_call_lhs (stmt);
4486 tree type = TREE_TYPE (gimple_call_arg (stmt, 1));
4487 BrigType16_t hsa_type = hsa_type_for_scalar_tree_type (type, false);
4488 BrigType16_t mtype = mem_type_for_type (hsa_type);
4489 tree model = gimple_call_arg (stmt, 2);
4491 if (!tree_fits_uhwi_p (model))
4493 HSA_SORRY_ATV (gimple_location (stmt),
4494 "support for HSA does not implement memory model %E",
4495 model);
4496 return;
4499 unsigned HOST_WIDE_INT mmodel = tree_to_uhwi (model);
4501 BrigMemoryOrder memorder = get_memory_order (mmodel, gimple_location (stmt));
4503 /* Certain atomic insns must have Bx memory types. */
4504 switch (acode)
4506 case BRIG_ATOMIC_LD:
4507 case BRIG_ATOMIC_ST:
4508 case BRIG_ATOMIC_AND:
4509 case BRIG_ATOMIC_OR:
4510 case BRIG_ATOMIC_XOR:
4511 case BRIG_ATOMIC_EXCH:
4512 mtype = hsa_bittype_for_type (mtype);
4513 break;
4514 default:
4515 break;
4518 hsa_op_reg *dest;
4519 int nops, opcode;
4520 if (lhs)
4522 if (ret_orig)
4523 dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4524 else
4525 dest = new hsa_op_reg (hsa_type);
4526 opcode = BRIG_OPCODE_ATOMIC;
4527 nops = 3;
4529 else
4531 dest = NULL;
4532 opcode = BRIG_OPCODE_ATOMICNORET;
4533 nops = 2;
4536 if (acode == BRIG_ATOMIC_ST)
4538 if (memorder == BRIG_MEMORY_ORDER_SC_ACQUIRE_RELEASE)
4539 memorder = BRIG_MEMORY_ORDER_SC_RELEASE;
4541 if (memorder != BRIG_MEMORY_ORDER_RELAXED
4542 && memorder != BRIG_MEMORY_ORDER_SC_RELEASE
4543 && memorder != BRIG_MEMORY_ORDER_NONE)
4545 HSA_SORRY_ATV (gimple_location (stmt),
4546 "support for HSA does not implement memory model for "
4547 "ATOMIC_ST: %s", get_memory_order_name (mmodel));
4548 return;
4552 hsa_insn_atomic *atominsn = new hsa_insn_atomic (nops, opcode, acode, mtype,
4553 memorder);
4555 hsa_op_address *addr;
4556 addr = get_address_from_value (gimple_call_arg (stmt, 0), hbb);
4557 /* TODO: Warn if addr has private segment, because the finalizer will not
4558 accept that (and it does not make much sense). */
4559 hsa_op_base *op = hsa_reg_or_immed_for_gimple_op (gimple_call_arg (stmt, 1),
4560 hbb);
4562 if (lhs)
4564 atominsn->set_op (0, dest);
4565 atominsn->set_op (1, addr);
4566 atominsn->set_op (2, op);
4568 else
4570 atominsn->set_op (0, addr);
4571 atominsn->set_op (1, op);
4574 hbb->append_insn (atominsn);
4576 /* HSA does not natively support the variants that return the modified value,
4577 so re-do the operation again non-atomically if that is what was
4578 requested. */
4579 if (lhs && !ret_orig)
4581 int arith;
4582 switch (acode)
4584 case BRIG_ATOMIC_ADD:
4585 arith = BRIG_OPCODE_ADD;
4586 break;
4587 case BRIG_ATOMIC_AND:
4588 arith = BRIG_OPCODE_AND;
4589 break;
4590 case BRIG_ATOMIC_OR:
4591 arith = BRIG_OPCODE_OR;
4592 break;
4593 case BRIG_ATOMIC_SUB:
4594 arith = BRIG_OPCODE_SUB;
4595 break;
4596 case BRIG_ATOMIC_XOR:
4597 arith = BRIG_OPCODE_XOR;
4598 break;
4599 default:
4600 gcc_unreachable ();
4602 hsa_op_reg *real_dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4603 gen_hsa_binary_operation (arith, real_dest, dest, op, hbb);
4607 /* Generate HSA instructions for an internal fn.
4608 Instructions will be appended to HBB, which also needs to be the
4609 corresponding structure to the basic_block of STMT. */
4611 static void
4612 gen_hsa_insn_for_internal_fn_call (gcall *stmt, hsa_bb *hbb)
4614 gcc_checking_assert (gimple_call_internal_fn (stmt));
4615 internal_fn fn = gimple_call_internal_fn (stmt);
4617 bool is_float_type_p = false;
4618 if (gimple_call_lhs (stmt) != NULL
4619 && TREE_TYPE (gimple_call_lhs (stmt)) == float_type_node)
4620 is_float_type_p = true;
4622 switch (fn)
4624 case IFN_CEIL:
4625 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_CEIL, stmt, hbb);
4626 break;
4628 case IFN_FLOOR:
4629 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_FLOOR, stmt, hbb);
4630 break;
4632 case IFN_RINT:
4633 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_RINT, stmt, hbb);
4634 break;
4636 case IFN_SQRT:
4637 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_SQRT, stmt, hbb);
4638 break;
4640 case IFN_TRUNC:
4641 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_TRUNC, stmt, hbb);
4642 break;
4644 case IFN_COS:
4646 if (is_float_type_p)
4647 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NCOS, stmt, hbb);
4648 else
4649 gen_hsa_unaryop_builtin_call (stmt, hbb);
4651 break;
4653 case IFN_EXP2:
4655 if (is_float_type_p)
4656 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NEXP2, stmt, hbb);
4657 else
4658 gen_hsa_unaryop_builtin_call (stmt, hbb);
4660 break;
4663 case IFN_LOG2:
4665 if (is_float_type_p)
4666 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NLOG2, stmt, hbb);
4667 else
4668 gen_hsa_unaryop_builtin_call (stmt, hbb);
4670 break;
4673 case IFN_SIN:
4675 if (is_float_type_p)
4676 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NSIN, stmt, hbb);
4677 else
4678 gen_hsa_unaryop_builtin_call (stmt, hbb);
4679 break;
4682 case IFN_CLRSB:
4683 gen_hsa_clrsb (stmt, hbb);
4684 break;
4686 case IFN_CLZ:
4687 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_FIRSTBIT, stmt, hbb);
4688 break;
4690 case IFN_CTZ:
4691 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_LASTBIT, stmt, hbb);
4692 break;
4694 case IFN_FFS:
4695 gen_hsa_ffs (stmt, hbb);
4696 break;
4698 case IFN_PARITY:
4699 gen_hsa_parity (stmt, hbb);
4700 break;
4702 case IFN_POPCOUNT:
4703 gen_hsa_popcount (stmt, hbb);
4704 break;
4706 case IFN_ACOS:
4707 case IFN_ASIN:
4708 case IFN_ATAN:
4709 case IFN_EXP:
4710 case IFN_EXP10:
4711 case IFN_EXPM1:
4712 case IFN_LOG:
4713 case IFN_LOG10:
4714 case IFN_LOG1P:
4715 case IFN_LOGB:
4716 case IFN_SIGNIFICAND:
4717 case IFN_TAN:
4718 case IFN_NEARBYINT:
4719 case IFN_ROUND:
4720 case IFN_ATAN2:
4721 case IFN_COPYSIGN:
4722 case IFN_FMOD:
4723 case IFN_POW:
4724 case IFN_REMAINDER:
4725 case IFN_SCALB:
4726 case IFN_FMIN:
4727 case IFN_FMAX:
4728 gen_hsa_insns_for_call_of_internal_fn (stmt, hbb);
4730 default:
4731 HSA_SORRY_ATV (gimple_location (stmt),
4732 "support for HSA does not implement internal function: %s",
4733 internal_fn_name (fn));
4734 break;
4738 #define HSA_MEMORY_BUILTINS_LIMIT 128
4740 /* Generate HSA instructions for the given call statement STMT. Instructions
4741 will be appended to HBB. */
4743 static void
4744 gen_hsa_insns_for_call (gimple *stmt, hsa_bb *hbb)
4746 gcall *call = as_a <gcall *> (stmt);
4747 tree lhs = gimple_call_lhs (stmt);
4748 hsa_op_reg *dest;
4750 if (gimple_call_internal_p (stmt))
4752 gen_hsa_insn_for_internal_fn_call (call, hbb);
4753 return;
4756 if (!gimple_call_builtin_p (stmt, BUILT_IN_NORMAL))
4758 tree function_decl = gimple_call_fndecl (stmt);
4759 if (function_decl == NULL_TREE)
4761 HSA_SORRY_AT (gimple_location (stmt),
4762 "support for HSA does not implement indirect calls");
4763 return;
4766 if (hsa_callable_function_p (function_decl))
4767 gen_hsa_insns_for_direct_call (stmt, hbb);
4768 else if (!gen_hsa_insns_for_known_library_call (stmt, hbb))
4769 HSA_SORRY_AT (gimple_location (stmt),
4770 "HSA supports only calls of functions marked with pragma "
4771 "omp declare target");
4772 return;
4775 tree fndecl = gimple_call_fndecl (stmt);
4776 enum built_in_function builtin = DECL_FUNCTION_CODE (fndecl);
4777 switch (builtin)
4779 case BUILT_IN_FABS:
4780 case BUILT_IN_FABSF:
4781 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_ABS, stmt, hbb);
4782 break;
4784 case BUILT_IN_CEIL:
4785 case BUILT_IN_CEILF:
4786 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_CEIL, stmt, hbb);
4787 break;
4789 case BUILT_IN_FLOOR:
4790 case BUILT_IN_FLOORF:
4791 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_FLOOR, stmt, hbb);
4792 break;
4794 case BUILT_IN_RINT:
4795 case BUILT_IN_RINTF:
4796 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_RINT, stmt, hbb);
4797 break;
4799 case BUILT_IN_SQRT:
4800 case BUILT_IN_SQRTF:
4801 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_SQRT, stmt, hbb);
4802 break;
4804 case BUILT_IN_TRUNC:
4805 case BUILT_IN_TRUNCF:
4806 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_TRUNC, stmt, hbb);
4807 break;
4809 case BUILT_IN_COS:
4810 case BUILT_IN_SIN:
4811 case BUILT_IN_EXP2:
4812 case BUILT_IN_LOG2:
4813 /* HSAIL does not provide an instruction for double argument type. */
4814 gen_hsa_unaryop_builtin_call (stmt, hbb);
4815 break;
4817 case BUILT_IN_COSF:
4818 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NCOS, stmt, hbb);
4819 break;
4821 case BUILT_IN_EXP2F:
4822 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NEXP2, stmt, hbb);
4823 break;
4825 case BUILT_IN_LOG2F:
4826 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NLOG2, stmt, hbb);
4827 break;
4829 case BUILT_IN_SINF:
4830 gen_hsa_unaryop_or_call_for_builtin (BRIG_OPCODE_NSIN, stmt, hbb);
4831 break;
4833 case BUILT_IN_CLRSB:
4834 case BUILT_IN_CLRSBL:
4835 case BUILT_IN_CLRSBLL:
4836 gen_hsa_clrsb (call, hbb);
4837 break;
4839 case BUILT_IN_CLZ:
4840 case BUILT_IN_CLZL:
4841 case BUILT_IN_CLZLL:
4842 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_FIRSTBIT, stmt, hbb);
4843 break;
4845 case BUILT_IN_CTZ:
4846 case BUILT_IN_CTZL:
4847 case BUILT_IN_CTZLL:
4848 gen_hsa_unaryop_for_builtin (BRIG_OPCODE_LASTBIT, stmt, hbb);
4849 break;
4851 case BUILT_IN_FFS:
4852 case BUILT_IN_FFSL:
4853 case BUILT_IN_FFSLL:
4854 gen_hsa_ffs (call, hbb);
4855 break;
4857 case BUILT_IN_PARITY:
4858 case BUILT_IN_PARITYL:
4859 case BUILT_IN_PARITYLL:
4860 gen_hsa_parity (call, hbb);
4861 break;
4863 case BUILT_IN_POPCOUNT:
4864 case BUILT_IN_POPCOUNTL:
4865 case BUILT_IN_POPCOUNTLL:
4866 gen_hsa_popcount (call, hbb);
4867 break;
4869 case BUILT_IN_ATOMIC_LOAD_1:
4870 case BUILT_IN_ATOMIC_LOAD_2:
4871 case BUILT_IN_ATOMIC_LOAD_4:
4872 case BUILT_IN_ATOMIC_LOAD_8:
4873 case BUILT_IN_ATOMIC_LOAD_16:
4875 BrigType16_t mtype;
4876 hsa_op_address *addr;
4877 addr = get_address_from_value (gimple_call_arg (stmt, 0), hbb);
4878 tree model = gimple_call_arg (stmt, 1);
4879 if (!tree_fits_uhwi_p (model))
4881 HSA_SORRY_ATV (gimple_location (stmt),
4882 "support for HSA does not implement "
4883 "memory model: %E",
4884 model);
4885 return;
4888 unsigned HOST_WIDE_INT mmodel = tree_to_uhwi (model);
4889 BrigMemoryOrder memorder = get_memory_order (mmodel,
4890 gimple_location (stmt));
4892 if (memorder == BRIG_MEMORY_ORDER_SC_ACQUIRE_RELEASE)
4893 memorder = BRIG_MEMORY_ORDER_SC_ACQUIRE;
4895 if (memorder != BRIG_MEMORY_ORDER_RELAXED
4896 && memorder != BRIG_MEMORY_ORDER_SC_ACQUIRE
4897 && memorder != BRIG_MEMORY_ORDER_NONE)
4899 HSA_SORRY_ATV (gimple_location (stmt),
4900 "support for HSA does not implement "
4901 "memory model for ATOMIC_LD: %s",
4902 get_memory_order_name (mmodel));
4903 return;
4906 if (lhs)
4908 BrigType16_t t = hsa_type_for_scalar_tree_type (TREE_TYPE (lhs),
4909 false);
4910 mtype = mem_type_for_type (t);
4911 mtype = hsa_bittype_for_type (mtype);
4912 dest = hsa_cfun->reg_for_gimple_ssa (lhs);
4914 else
4916 mtype = BRIG_TYPE_B64;
4917 dest = new hsa_op_reg (mtype);
4920 hsa_insn_atomic *atominsn
4921 = new hsa_insn_atomic (2, BRIG_OPCODE_ATOMIC, BRIG_ATOMIC_LD, mtype,
4922 memorder, dest, addr);
4924 hbb->append_insn (atominsn);
4925 break;
4928 case BUILT_IN_ATOMIC_EXCHANGE_1:
4929 case BUILT_IN_ATOMIC_EXCHANGE_2:
4930 case BUILT_IN_ATOMIC_EXCHANGE_4:
4931 case BUILT_IN_ATOMIC_EXCHANGE_8:
4932 case BUILT_IN_ATOMIC_EXCHANGE_16:
4933 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_EXCH, stmt, hbb);
4934 break;
4936 case BUILT_IN_ATOMIC_FETCH_ADD_1:
4937 case BUILT_IN_ATOMIC_FETCH_ADD_2:
4938 case BUILT_IN_ATOMIC_FETCH_ADD_4:
4939 case BUILT_IN_ATOMIC_FETCH_ADD_8:
4940 case BUILT_IN_ATOMIC_FETCH_ADD_16:
4941 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_ADD, stmt, hbb);
4942 break;
4944 case BUILT_IN_ATOMIC_FETCH_SUB_1:
4945 case BUILT_IN_ATOMIC_FETCH_SUB_2:
4946 case BUILT_IN_ATOMIC_FETCH_SUB_4:
4947 case BUILT_IN_ATOMIC_FETCH_SUB_8:
4948 case BUILT_IN_ATOMIC_FETCH_SUB_16:
4949 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_SUB, stmt, hbb);
4950 break;
4952 case BUILT_IN_ATOMIC_FETCH_AND_1:
4953 case BUILT_IN_ATOMIC_FETCH_AND_2:
4954 case BUILT_IN_ATOMIC_FETCH_AND_4:
4955 case BUILT_IN_ATOMIC_FETCH_AND_8:
4956 case BUILT_IN_ATOMIC_FETCH_AND_16:
4957 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_AND, stmt, hbb);
4958 break;
4960 case BUILT_IN_ATOMIC_FETCH_XOR_1:
4961 case BUILT_IN_ATOMIC_FETCH_XOR_2:
4962 case BUILT_IN_ATOMIC_FETCH_XOR_4:
4963 case BUILT_IN_ATOMIC_FETCH_XOR_8:
4964 case BUILT_IN_ATOMIC_FETCH_XOR_16:
4965 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_XOR, stmt, hbb);
4966 break;
4968 case BUILT_IN_ATOMIC_FETCH_OR_1:
4969 case BUILT_IN_ATOMIC_FETCH_OR_2:
4970 case BUILT_IN_ATOMIC_FETCH_OR_4:
4971 case BUILT_IN_ATOMIC_FETCH_OR_8:
4972 case BUILT_IN_ATOMIC_FETCH_OR_16:
4973 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_OR, stmt, hbb);
4974 break;
4976 case BUILT_IN_ATOMIC_STORE_1:
4977 case BUILT_IN_ATOMIC_STORE_2:
4978 case BUILT_IN_ATOMIC_STORE_4:
4979 case BUILT_IN_ATOMIC_STORE_8:
4980 case BUILT_IN_ATOMIC_STORE_16:
4981 /* Since there cannot be any LHS, the first parameter is meaningless. */
4982 gen_hsa_ternary_atomic_for_builtin (true, BRIG_ATOMIC_ST, stmt, hbb);
4983 break;
4985 case BUILT_IN_ATOMIC_ADD_FETCH_1:
4986 case BUILT_IN_ATOMIC_ADD_FETCH_2:
4987 case BUILT_IN_ATOMIC_ADD_FETCH_4:
4988 case BUILT_IN_ATOMIC_ADD_FETCH_8:
4989 case BUILT_IN_ATOMIC_ADD_FETCH_16:
4990 gen_hsa_ternary_atomic_for_builtin (false, BRIG_ATOMIC_ADD, stmt, hbb);
4991 break;
4993 case BUILT_IN_ATOMIC_SUB_FETCH_1:
4994 case BUILT_IN_ATOMIC_SUB_FETCH_2:
4995 case BUILT_IN_ATOMIC_SUB_FETCH_4:
4996 case BUILT_IN_ATOMIC_SUB_FETCH_8:
4997 case BUILT_IN_ATOMIC_SUB_FETCH_16:
4998 gen_hsa_ternary_atomic_for_builtin (false, BRIG_ATOMIC_SUB, stmt, hbb);
4999 break;
5001 case BUILT_IN_ATOMIC_AND_FETCH_1:
5002 case BUILT_IN_ATOMIC_AND_FETCH_2:
5003 case BUILT_IN_ATOMIC_AND_FETCH_4:
5004 case BUILT_IN_ATOMIC_AND_FETCH_8:
5005 case BUILT_IN_ATOMIC_AND_FETCH_16:
5006 gen_hsa_ternary_atomic_for_builtin (false, BRIG_ATOMIC_AND, stmt, hbb);
5007 break;
5009 case BUILT_IN_ATOMIC_XOR_FETCH_1:
5010 case BUILT_IN_ATOMIC_XOR_FETCH_2:
5011 case BUILT_IN_ATOMIC_XOR_FETCH_4:
5012 case BUILT_IN_ATOMIC_XOR_FETCH_8:
5013 case BUILT_IN_ATOMIC_XOR_FETCH_16:
5014 gen_hsa_ternary_atomic_for_builtin (false, BRIG_ATOMIC_XOR, stmt, hbb);
5015 break;
5017 case BUILT_IN_ATOMIC_OR_FETCH_1:
5018 case BUILT_IN_ATOMIC_OR_FETCH_2:
5019 case BUILT_IN_ATOMIC_OR_FETCH_4:
5020 case BUILT_IN_ATOMIC_OR_FETCH_8:
5021 case BUILT_IN_ATOMIC_OR_FETCH_16:
5022 gen_hsa_ternary_atomic_for_builtin (false, BRIG_ATOMIC_OR, stmt, hbb);
5023 break;
5025 case BUILT_IN_SYNC_VAL_COMPARE_AND_SWAP_1:
5026 case BUILT_IN_SYNC_VAL_COMPARE_AND_SWAP_2:
5027 case BUILT_IN_SYNC_VAL_COMPARE_AND_SWAP_4:
5028 case BUILT_IN_SYNC_VAL_COMPARE_AND_SWAP_8:
5029 case BUILT_IN_SYNC_VAL_COMPARE_AND_SWAP_16:
5031 /* TODO: Use the appropriate memory model for now. */
5032 tree type = TREE_TYPE (gimple_call_arg (stmt, 1));
5034 BrigType16_t atype
5035 = hsa_bittype_for_type (hsa_type_for_scalar_tree_type (type, false));
5037 hsa_insn_atomic *atominsn
5038 = new hsa_insn_atomic (4, BRIG_OPCODE_ATOMIC, BRIG_ATOMIC_CAS, atype,
5039 BRIG_MEMORY_ORDER_SC_ACQUIRE_RELEASE);
5040 hsa_op_address *addr;
5041 addr = get_address_from_value (gimple_call_arg (stmt, 0), hbb);
5043 if (lhs != NULL)
5044 dest = hsa_cfun->reg_for_gimple_ssa (lhs);
5045 else
5046 dest = new hsa_op_reg (atype);
5048 /* Should check what the memory scope is. */
5049 atominsn->m_memoryscope = BRIG_MEMORY_SCOPE_WORKGROUP;
5050 atominsn->set_op (0, dest);
5051 atominsn->set_op (1, addr);
5053 hsa_op_with_type *op
5054 = hsa_reg_or_immed_for_gimple_op (gimple_call_arg (stmt, 1), hbb);
5055 atominsn->set_op (2, op);
5056 op = hsa_reg_or_immed_for_gimple_op (gimple_call_arg (stmt, 2), hbb);
5057 atominsn->set_op (3, op);
5059 hbb->append_insn (atominsn);
5060 break;
5062 case BUILT_IN_GOMP_PARALLEL:
5063 HSA_SORRY_AT (gimple_location (stmt),
5064 "support for HSA does not implement non-gridified "
5065 "OpenMP parallel constructs.");
5066 break;
5067 case BUILT_IN_OMP_GET_THREAD_NUM:
5069 query_hsa_grid (stmt, BRIG_OPCODE_WORKITEMABSID, 0, hbb);
5070 break;
5073 case BUILT_IN_OMP_GET_NUM_THREADS:
5075 query_hsa_grid (stmt, BRIG_OPCODE_GRIDSIZE, 0, hbb);
5076 break;
5078 case BUILT_IN_GOMP_TEAMS:
5080 gen_set_num_threads (gimple_call_arg (stmt, 1), hbb);
5081 break;
5083 case BUILT_IN_OMP_GET_NUM_TEAMS:
5085 gen_get_num_teams (stmt, hbb);
5086 break;
5088 case BUILT_IN_OMP_GET_TEAM_NUM:
5090 gen_get_team_num (stmt, hbb);
5091 break;
5093 case BUILT_IN_MEMCPY:
5094 case BUILT_IN_MEMPCPY:
5096 tree byte_size = gimple_call_arg (stmt, 2);
5098 if (!tree_fits_uhwi_p (byte_size))
5100 gen_hsa_insns_for_direct_call (stmt, hbb);
5101 return;
5104 unsigned n = tree_to_uhwi (byte_size);
5106 if (n > HSA_MEMORY_BUILTINS_LIMIT)
5108 gen_hsa_insns_for_direct_call (stmt, hbb);
5109 return;
5112 tree dst = gimple_call_arg (stmt, 0);
5113 tree src = gimple_call_arg (stmt, 1);
5115 hsa_op_address *dst_addr = get_address_from_value (dst, hbb);
5116 hsa_op_address *src_addr = get_address_from_value (src, hbb);
5118 gen_hsa_memory_copy (hbb, dst_addr, src_addr, n);
5120 tree lhs = gimple_call_lhs (stmt);
5121 if (lhs)
5123 hsa_op_reg *lhs_reg = hsa_cfun->reg_for_gimple_ssa (lhs);
5124 hsa_op_with_type *dst_reg = hsa_reg_or_immed_for_gimple_op (dst,
5125 hbb);
5126 hsa_op_with_type *tmp;
5128 if (builtin == BUILT_IN_MEMPCPY)
5130 tmp = new hsa_op_reg (dst_reg->m_type);
5131 hsa_insn_basic *add
5132 = new hsa_insn_basic (3, BRIG_OPCODE_ADD, tmp->m_type,
5133 tmp, dst_reg,
5134 new hsa_op_immed (n, dst_reg->m_type));
5135 hbb->append_insn (add);
5137 else
5138 tmp = dst_reg;
5140 hsa_build_append_simple_mov (lhs_reg, tmp, hbb);
5143 break;
5145 case BUILT_IN_MEMSET:
5147 tree dst = gimple_call_arg (stmt, 0);
5148 tree c = gimple_call_arg (stmt, 1);
5150 if (TREE_CODE (c) != INTEGER_CST)
5152 gen_hsa_insns_for_direct_call (stmt, hbb);
5153 return;
5156 tree byte_size = gimple_call_arg (stmt, 2);
5158 if (!tree_fits_uhwi_p (byte_size))
5160 gen_hsa_insns_for_direct_call (stmt, hbb);
5161 return;
5164 unsigned n = tree_to_uhwi (byte_size);
5166 if (n > HSA_MEMORY_BUILTINS_LIMIT)
5168 gen_hsa_insns_for_direct_call (stmt, hbb);
5169 return;
5172 hsa_op_address *dst_addr;
5173 dst_addr = get_address_from_value (dst, hbb);
5174 unsigned HOST_WIDE_INT constant
5175 = tree_to_uhwi (fold_convert (unsigned_char_type_node, c));
5177 gen_hsa_memory_set (hbb, dst_addr, constant, n);
5179 tree lhs = gimple_call_lhs (stmt);
5180 if (lhs)
5181 gen_hsa_insns_for_single_assignment (lhs, dst, hbb);
5183 break;
5185 case BUILT_IN_BZERO:
5187 tree dst = gimple_call_arg (stmt, 0);
5188 tree byte_size = gimple_call_arg (stmt, 1);
5190 if (!tree_fits_uhwi_p (byte_size))
5192 gen_hsa_insns_for_direct_call (stmt, hbb);
5193 return;
5196 unsigned n = tree_to_uhwi (byte_size);
5198 if (n > HSA_MEMORY_BUILTINS_LIMIT)
5200 gen_hsa_insns_for_direct_call (stmt, hbb);
5201 return;
5204 hsa_op_address *dst_addr;
5205 dst_addr = get_address_from_value (dst, hbb);
5207 gen_hsa_memory_set (hbb, dst_addr, 0, n);
5209 break;
5211 case BUILT_IN_ALLOCA:
5212 case BUILT_IN_ALLOCA_WITH_ALIGN:
5214 gen_hsa_alloca (call, hbb);
5215 break;
5217 default:
5219 gen_hsa_insns_for_direct_call (stmt, hbb);
5220 return;
5225 /* Generate HSA instructions for a given gimple statement. Instructions will be
5226 appended to HBB. */
5228 static void
5229 gen_hsa_insns_for_gimple_stmt (gimple *stmt, hsa_bb *hbb)
5231 switch (gimple_code (stmt))
5233 case GIMPLE_ASSIGN:
5234 if (gimple_clobber_p (stmt))
5235 break;
5237 if (gimple_assign_single_p (stmt))
5239 tree lhs = gimple_assign_lhs (stmt);
5240 tree rhs = gimple_assign_rhs1 (stmt);
5241 gen_hsa_insns_for_single_assignment (lhs, rhs, hbb);
5243 else
5244 gen_hsa_insns_for_operation_assignment (stmt, hbb);
5245 break;
5246 case GIMPLE_RETURN:
5247 gen_hsa_insns_for_return (as_a <greturn *> (stmt), hbb);
5248 break;
5249 case GIMPLE_COND:
5250 gen_hsa_insns_for_cond_stmt (stmt, hbb);
5251 break;
5252 case GIMPLE_CALL:
5253 gen_hsa_insns_for_call (stmt, hbb);
5254 break;
5255 case GIMPLE_DEBUG:
5256 /* ??? HSA supports some debug facilities. */
5257 break;
5258 case GIMPLE_LABEL:
5260 tree label = gimple_label_label (as_a <glabel *> (stmt));
5261 if (FORCED_LABEL (label))
5262 HSA_SORRY_AT (gimple_location (stmt),
5263 "support for HSA does not implement gimple label with "
5264 "address taken");
5266 break;
5268 case GIMPLE_NOP:
5270 hbb->append_insn (new hsa_insn_basic (0, BRIG_OPCODE_NOP));
5271 break;
5273 case GIMPLE_SWITCH:
5275 gen_hsa_insns_for_switch_stmt (as_a <gswitch *> (stmt), hbb);
5276 break;
5278 default:
5279 HSA_SORRY_ATV (gimple_location (stmt),
5280 "support for HSA does not implement gimple statement %s",
5281 gimple_code_name[(int) gimple_code (stmt)]);
5285 /* Generate a HSA PHI from a gimple PHI. */
5287 static void
5288 gen_hsa_phi_from_gimple_phi (gimple *phi_stmt, hsa_bb *hbb)
5290 hsa_insn_phi *hphi;
5291 unsigned count = gimple_phi_num_args (phi_stmt);
5293 hsa_op_reg *dest
5294 = hsa_cfun->reg_for_gimple_ssa (gimple_phi_result (phi_stmt));
5295 hphi = new hsa_insn_phi (count, dest);
5296 hphi->m_bb = hbb->m_bb;
5298 tree lhs = gimple_phi_result (phi_stmt);
5300 for (unsigned i = 0; i < count; i++)
5302 tree op = gimple_phi_arg_def (phi_stmt, i);
5304 if (TREE_CODE (op) == SSA_NAME)
5306 hsa_op_reg *hreg = hsa_cfun->reg_for_gimple_ssa (op);
5307 hphi->set_op (i, hreg);
5309 else
5311 gcc_assert (is_gimple_min_invariant (op));
5312 tree t = TREE_TYPE (op);
5313 if (!POINTER_TYPE_P (t)
5314 || (TREE_CODE (op) == STRING_CST
5315 && TREE_CODE (TREE_TYPE (t)) == INTEGER_TYPE))
5316 hphi->set_op (i, new hsa_op_immed (op));
5317 else if (POINTER_TYPE_P (TREE_TYPE (lhs))
5318 && TREE_CODE (op) == INTEGER_CST)
5320 /* Handle assignment of NULL value to a pointer type. */
5321 hphi->set_op (i, new hsa_op_immed (op));
5323 else if (TREE_CODE (op) == ADDR_EXPR)
5325 edge e = gimple_phi_arg_edge (as_a <gphi *> (phi_stmt), i);
5326 hsa_bb *hbb_src = hsa_init_new_bb (split_edge (e));
5327 hsa_op_address *addr = gen_hsa_addr (TREE_OPERAND (op, 0),
5328 hbb_src);
5330 hsa_op_reg *dest = new hsa_op_reg (BRIG_TYPE_U64);
5331 hsa_insn_basic *insn
5332 = new hsa_insn_basic (2, BRIG_OPCODE_LDA, BRIG_TYPE_U64,
5333 dest, addr);
5334 hbb_src->append_insn (insn);
5336 hphi->set_op (i, dest);
5338 else
5340 HSA_SORRY_AT (gimple_location (phi_stmt),
5341 "support for HSA does not handle PHI nodes with "
5342 "constant address operands");
5343 return;
5348 hphi->m_prev = hbb->m_last_phi;
5349 hphi->m_next = NULL;
5350 if (hbb->m_last_phi)
5351 hbb->m_last_phi->m_next = hphi;
5352 hbb->m_last_phi = hphi;
5353 if (!hbb->m_first_phi)
5354 hbb->m_first_phi = hphi;
5357 /* Constructor of class containing HSA-specific information about a basic
5358 block. CFG_BB is the CFG BB this HSA BB is associated with. IDX is the new
5359 index of this BB (so that the constructor does not attempt to use
5360 hsa_cfun during its construction). */
5362 hsa_bb::hsa_bb (basic_block cfg_bb, int idx)
5363 : m_bb (cfg_bb), m_first_insn (NULL), m_last_insn (NULL), m_first_phi (NULL),
5364 m_last_phi (NULL), m_index (idx), m_liveout (BITMAP_ALLOC (NULL)),
5365 m_livein (BITMAP_ALLOC (NULL))
5367 gcc_assert (!cfg_bb->aux);
5368 cfg_bb->aux = this;
5371 /* Constructor of class containing HSA-specific information about a basic
5372 block. CFG_BB is the CFG BB this HSA BB is associated with. */
5374 hsa_bb::hsa_bb (basic_block cfg_bb)
5375 : m_bb (cfg_bb), m_first_insn (NULL), m_last_insn (NULL), m_first_phi (NULL),
5376 m_last_phi (NULL), m_index (hsa_cfun->m_hbb_count++),
5377 m_liveout (BITMAP_ALLOC (NULL)), m_livein (BITMAP_ALLOC (NULL))
5379 gcc_assert (!cfg_bb->aux);
5380 cfg_bb->aux = this;
5383 /* Destructor of class representing HSA BB. */
5385 hsa_bb::~hsa_bb ()
5387 BITMAP_FREE (m_livein);
5388 BITMAP_FREE (m_liveout);
5391 /* Create and initialize and return a new hsa_bb structure for a given CFG
5392 basic block BB. */
5394 hsa_bb *
5395 hsa_init_new_bb (basic_block bb)
5397 return new (*hsa_allocp_bb) hsa_bb (bb);
5400 /* Initialize OMP in an HSA basic block PROLOGUE. */
5402 static void
5403 init_prologue (void)
5405 if (!hsa_cfun->m_kern_p)
5406 return;
5408 hsa_bb *prologue = hsa_bb_for_bb (ENTRY_BLOCK_PTR_FOR_FN (cfun));
5410 /* Create a magic number that is going to be printed by libgomp. */
5411 unsigned index = hsa_get_number_decl_kernel_mappings ();
5413 /* Emit store to debug argument. */
5414 if (PARAM_VALUE (PARAM_HSA_GEN_DEBUG_STORES) > 0)
5415 set_debug_value (prologue, new hsa_op_immed (1000 + index, BRIG_TYPE_U64));
5418 /* Initialize hsa_num_threads to a default value. */
5420 static void
5421 init_hsa_num_threads (void)
5423 hsa_bb *prologue = hsa_bb_for_bb (ENTRY_BLOCK_PTR_FOR_FN (cfun));
5425 /* Save the default value to private variable hsa_num_threads. */
5426 hsa_insn_basic *basic
5427 = new hsa_insn_mem (BRIG_OPCODE_ST, hsa_num_threads->m_type,
5428 new hsa_op_immed (0, hsa_num_threads->m_type),
5429 new hsa_op_address (hsa_num_threads));
5430 prologue->append_insn (basic);
5433 /* Go over gimple representation and generate our internal HSA one. */
5435 static void
5436 gen_body_from_gimple ()
5438 basic_block bb;
5440 /* Verify CFG for complex edges we are unable to handle. */
5441 edge_iterator ei;
5442 edge e;
5444 FOR_EACH_BB_FN (bb, cfun)
5446 FOR_EACH_EDGE (e, ei, bb->succs)
5448 /* Verify all unsupported flags for edges that point
5449 to the same basic block. */
5450 if (e->flags & EDGE_EH)
5452 HSA_SORRY_AT (UNKNOWN_LOCATION,
5453 "support for HSA does not implement exception "
5454 "handling");
5455 return;
5460 FOR_EACH_BB_FN (bb, cfun)
5462 gimple_stmt_iterator gsi;
5463 hsa_bb *hbb = hsa_bb_for_bb (bb);
5464 if (hbb)
5465 continue;
5467 hbb = hsa_init_new_bb (bb);
5469 for (gsi = gsi_start_bb (bb); !gsi_end_p (gsi); gsi_next (&gsi))
5471 gen_hsa_insns_for_gimple_stmt (gsi_stmt (gsi), hbb);
5472 if (hsa_seen_error ())
5473 return;
5477 FOR_EACH_BB_FN (bb, cfun)
5479 gimple_stmt_iterator gsi;
5480 hsa_bb *hbb = hsa_bb_for_bb (bb);
5481 gcc_assert (hbb != NULL);
5483 for (gsi = gsi_start_phis (bb); !gsi_end_p (gsi); gsi_next (&gsi))
5484 if (!virtual_operand_p (gimple_phi_result (gsi_stmt (gsi))))
5485 gen_hsa_phi_from_gimple_phi (gsi_stmt (gsi), hbb);
5488 if (dump_file)
5490 fprintf (dump_file, "------- Generated SSA form -------\n");
5491 dump_hsa_cfun (dump_file);
5495 static void
5496 gen_function_decl_parameters (hsa_function_representation *f,
5497 tree decl)
5499 tree parm;
5500 unsigned i;
5502 for (parm = TYPE_ARG_TYPES (TREE_TYPE (decl)), i = 0;
5503 parm;
5504 parm = TREE_CHAIN (parm), i++)
5506 /* Result type if last in the tree list. */
5507 if (TREE_CHAIN (parm) == NULL)
5508 break;
5510 tree v = TREE_VALUE (parm);
5512 hsa_symbol *arg = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_ARG,
5513 BRIG_LINKAGE_NONE);
5514 arg->m_type = hsa_type_for_tree_type (v, &arg->m_dim);
5515 arg->m_name_number = i;
5517 f->m_input_args.safe_push (arg);
5520 tree result_type = TREE_TYPE (TREE_TYPE (decl));
5521 if (!VOID_TYPE_P (result_type))
5523 f->m_output_arg = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_ARG,
5524 BRIG_LINKAGE_NONE);
5525 f->m_output_arg->m_type
5526 = hsa_type_for_tree_type (result_type, &f->m_output_arg->m_dim);
5527 f->m_output_arg->m_name = "res";
5531 /* Generate the vector of parameters of the HSA representation of the current
5532 function. This also includes the output parameter representing the
5533 result. */
5535 static void
5536 gen_function_def_parameters ()
5538 tree parm;
5540 hsa_bb *prologue = hsa_bb_for_bb (ENTRY_BLOCK_PTR_FOR_FN (cfun));
5542 for (parm = DECL_ARGUMENTS (cfun->decl); parm;
5543 parm = DECL_CHAIN (parm))
5545 struct hsa_symbol **slot;
5547 hsa_symbol *arg
5548 = new hsa_symbol (BRIG_TYPE_NONE, hsa_cfun->m_kern_p
5549 ? BRIG_SEGMENT_KERNARG : BRIG_SEGMENT_ARG,
5550 BRIG_LINKAGE_FUNCTION);
5551 arg->fillup_for_decl (parm);
5553 hsa_cfun->m_input_args.safe_push (arg);
5555 if (hsa_seen_error ())
5556 return;
5558 arg->m_name = hsa_get_declaration_name (parm);
5560 /* Copy all input arguments and create corresponding private symbols
5561 for them. */
5562 hsa_symbol *private_arg;
5563 hsa_op_address *parm_addr = new hsa_op_address (arg);
5565 if (TREE_ADDRESSABLE (parm)
5566 || (!is_gimple_reg (parm) && !TREE_READONLY (parm)))
5568 private_arg = hsa_cfun->create_hsa_temporary (arg->m_type);
5569 private_arg->fillup_for_decl (parm);
5571 hsa_op_address *private_arg_addr = new hsa_op_address (private_arg);
5572 gen_hsa_memory_copy (prologue, private_arg_addr, parm_addr,
5573 arg->total_byte_size ());
5575 else
5576 private_arg = arg;
5578 slot = hsa_cfun->m_local_symbols->find_slot (private_arg, INSERT);
5579 gcc_assert (!*slot);
5580 *slot = private_arg;
5582 if (is_gimple_reg (parm))
5584 tree ddef = ssa_default_def (cfun, parm);
5585 if (ddef && !has_zero_uses (ddef))
5587 BrigType16_t t = hsa_type_for_scalar_tree_type (TREE_TYPE (ddef),
5588 false);
5589 BrigType16_t mtype = mem_type_for_type (t);
5590 hsa_op_reg *dest = hsa_cfun->reg_for_gimple_ssa (ddef);
5591 hsa_insn_mem *mem = new hsa_insn_mem (BRIG_OPCODE_LD, mtype,
5592 dest, parm_addr);
5593 gcc_assert (!parm_addr->m_reg);
5594 prologue->append_insn (mem);
5599 if (!VOID_TYPE_P (TREE_TYPE (TREE_TYPE (cfun->decl))))
5601 struct hsa_symbol **slot;
5603 hsa_cfun->m_output_arg = new hsa_symbol (BRIG_TYPE_NONE, BRIG_SEGMENT_ARG,
5604 BRIG_LINKAGE_FUNCTION);
5605 hsa_cfun->m_output_arg->fillup_for_decl (DECL_RESULT (cfun->decl));
5607 if (hsa_seen_error ())
5608 return;
5610 hsa_cfun->m_output_arg->m_name = "res";
5611 slot = hsa_cfun->m_local_symbols->find_slot (hsa_cfun->m_output_arg,
5612 INSERT);
5613 gcc_assert (!*slot);
5614 *slot = hsa_cfun->m_output_arg;
5618 /* Generate function representation that corresponds to
5619 a function declaration. */
5621 hsa_function_representation *
5622 hsa_generate_function_declaration (tree decl)
5624 hsa_function_representation *fun
5625 = new hsa_function_representation (decl, false, 0);
5627 fun->m_declaration_p = true;
5628 fun->m_name = get_brig_function_name (decl);
5629 gen_function_decl_parameters (fun, decl);
5631 return fun;
5635 /* Generate function representation that corresponds to
5636 an internal FN. */
5638 hsa_function_representation *
5639 hsa_generate_internal_fn_decl (hsa_internal_fn *fn)
5641 hsa_function_representation *fun = new hsa_function_representation (fn);
5643 fun->m_name = fn->name ();
5645 for (unsigned i = 0; i < fn->get_arity (); i++)
5647 hsa_symbol *arg
5648 = new hsa_symbol (fn->get_argument_type (i), BRIG_SEGMENT_ARG,
5649 BRIG_LINKAGE_NONE);
5650 arg->m_name_number = i;
5651 fun->m_input_args.safe_push (arg);
5654 fun->m_output_arg = new hsa_symbol (fn->get_argument_type (-1),
5655 BRIG_SEGMENT_ARG, BRIG_LINKAGE_NONE);
5656 fun->m_output_arg->m_name = "res";
5658 return fun;
5661 /* Return true if switch statement S can be transformed
5662 to a SBR instruction in HSAIL. */
5664 static bool
5665 transformable_switch_to_sbr_p (gswitch *s)
5667 /* Identify if a switch statement can be transformed to
5668 SBR instruction, like:
5670 sbr_u32 $s1 [@label1, @label2, @label3];
5673 tree size = get_switch_size (s);
5674 if (!tree_fits_uhwi_p (size))
5675 return false;
5677 if (tree_to_uhwi (size) > HSA_MAXIMUM_SBR_LABELS)
5678 return false;
5680 return true;
5683 /* Structure hold connection between PHI nodes and immediate
5684 values hold by there nodes. */
5686 struct phi_definition
5688 phi_definition (unsigned phi_i, unsigned label_i, tree imm):
5689 phi_index (phi_i), label_index (label_i), phi_value (imm)
5692 unsigned phi_index;
5693 unsigned label_index;
5694 tree phi_value;
5697 /* Sum slice of a vector V, starting from index START and ending
5698 at the index END - 1. */
5700 template <typename T>
5701 static
5702 T sum_slice (const auto_vec <T> &v, unsigned start, unsigned end)
5704 T s = 0;
5706 for (unsigned i = start; i < end; i++)
5707 s += v[i];
5709 return s;
5712 /* Function transforms GIMPLE SWITCH statements to a series of IF statements.
5713 Let's assume following example:
5716 switch (index)
5717 case C1:
5718 L1: hard_work_1 ();
5719 break;
5720 case C2..C3:
5721 L2: hard_work_2 ();
5722 break;
5723 default:
5724 LD: hard_work_3 ();
5725 break;
5727 The transformation encompasses following steps:
5728 1) all immediate values used by edges coming from the switch basic block
5729 are saved
5730 2) all these edges are removed
5731 3) the switch statement (in L0) is replaced by:
5732 if (index == C1)
5733 goto L1;
5734 else
5735 goto L1';
5737 4) newly created basic block Lx' is used for generation of
5738 a next condition
5739 5) else branch of the last condition goes to LD
5740 6) fix all immediate values in PHI nodes that were propagated though
5741 edges that were removed in step 2
5743 Note: if a case is made by a range C1..C2, then process
5744 following transformation:
5746 switch_cond_op1 = C1 <= index;
5747 switch_cond_op2 = index <= C2;
5748 switch_cond_and = switch_cond_op1 & switch_cond_op2;
5749 if (switch_cond_and != 0)
5750 goto Lx;
5751 else
5752 goto Ly;
5756 static void
5757 convert_switch_statements ()
5759 function *func = DECL_STRUCT_FUNCTION (current_function_decl);
5760 basic_block bb;
5762 bool need_update = false;
5764 FOR_EACH_BB_FN (bb, func)
5766 gimple_stmt_iterator gsi = gsi_last_bb (bb);
5767 if (gsi_end_p (gsi))
5768 continue;
5770 gimple *stmt = gsi_stmt (gsi);
5772 if (gimple_code (stmt) == GIMPLE_SWITCH)
5774 gswitch *s = as_a <gswitch *> (stmt);
5776 /* If the switch can utilize SBR insn, skip the statement. */
5777 if (transformable_switch_to_sbr_p (s))
5778 continue;
5780 need_update = true;
5782 unsigned labels = gimple_switch_num_labels (s);
5783 tree index = gimple_switch_index (s);
5784 tree index_type = TREE_TYPE (index);
5785 tree default_label = gimple_switch_default_label (s);
5786 basic_block default_label_bb
5787 = label_to_block_fn (func, CASE_LABEL (default_label));
5788 basic_block cur_bb = bb;
5790 auto_vec <edge> new_edges;
5791 auto_vec <phi_definition *> phi_todo_list;
5792 auto_vec <gcov_type> edge_counts;
5793 auto_vec <int> edge_probabilities;
5795 /* Investigate all labels that and PHI nodes in these edges which
5796 should be fixed after we add new collection of edges. */
5797 for (unsigned i = 0; i < labels; i++)
5799 tree label = gimple_switch_label (s, i);
5800 basic_block label_bb = label_to_block_fn (func, CASE_LABEL (label));
5801 edge e = find_edge (bb, label_bb);
5802 edge_counts.safe_push (e->count);
5803 edge_probabilities.safe_push (e->probability);
5804 gphi_iterator phi_gsi;
5806 /* Save PHI definitions that will be destroyed because of an edge
5807 is going to be removed. */
5808 unsigned phi_index = 0;
5809 for (phi_gsi = gsi_start_phis (e->dest);
5810 !gsi_end_p (phi_gsi); gsi_next (&phi_gsi))
5812 gphi *phi = phi_gsi.phi ();
5813 for (unsigned j = 0; j < gimple_phi_num_args (phi); j++)
5815 if (gimple_phi_arg_edge (phi, j) == e)
5817 tree imm = gimple_phi_arg_def (phi, j);
5818 phi_definition *p = new phi_definition (phi_index, i,
5819 imm);
5820 phi_todo_list.safe_push (p);
5821 break;
5824 phi_index++;
5828 /* Remove all edges for the current basic block. */
5829 for (int i = EDGE_COUNT (bb->succs) - 1; i >= 0; i--)
5831 edge e = EDGE_SUCC (bb, i);
5832 remove_edge (e);
5835 /* Iterate all non-default labels. */
5836 for (unsigned i = 1; i < labels; i++)
5838 tree label = gimple_switch_label (s, i);
5839 tree low = CASE_LOW (label);
5840 tree high = CASE_HIGH (label);
5842 if (!useless_type_conversion_p (TREE_TYPE (low), index_type))
5843 low = fold_convert (index_type, low);
5845 gimple_stmt_iterator cond_gsi = gsi_last_bb (cur_bb);
5846 gimple *c = NULL;
5847 if (high)
5849 tree tmp1 = make_temp_ssa_name (boolean_type_node, NULL,
5850 "switch_cond_op1");
5852 gimple *assign1 = gimple_build_assign (tmp1, LE_EXPR, low,
5853 index);
5855 tree tmp2 = make_temp_ssa_name (boolean_type_node, NULL,
5856 "switch_cond_op2");
5858 if (!useless_type_conversion_p (TREE_TYPE (high), index_type))
5859 high = fold_convert (index_type, high);
5860 gimple *assign2 = gimple_build_assign (tmp2, LE_EXPR, index,
5861 high);
5863 tree tmp3 = make_temp_ssa_name (boolean_type_node, NULL,
5864 "switch_cond_and");
5865 gimple *assign3 = gimple_build_assign (tmp3, BIT_AND_EXPR, tmp1,
5866 tmp2);
5868 gsi_insert_before (&cond_gsi, assign1, GSI_SAME_STMT);
5869 gsi_insert_before (&cond_gsi, assign2, GSI_SAME_STMT);
5870 gsi_insert_before (&cond_gsi, assign3, GSI_SAME_STMT);
5872 tree b = constant_boolean_node (false, boolean_type_node);
5873 c = gimple_build_cond (NE_EXPR, tmp3, b, NULL, NULL);
5875 else
5876 c = gimple_build_cond (EQ_EXPR, index, low, NULL, NULL);
5878 gimple_set_location (c, gimple_location (stmt));
5880 gsi_insert_before (&cond_gsi, c, GSI_SAME_STMT);
5882 basic_block label_bb
5883 = label_to_block_fn (func, CASE_LABEL (label));
5884 edge new_edge = make_edge (cur_bb, label_bb, EDGE_TRUE_VALUE);
5885 int prob_sum = sum_slice <int> (edge_probabilities, i, labels) +
5886 edge_probabilities[0];
5888 if (prob_sum)
5889 new_edge->probability
5890 = RDIV (REG_BR_PROB_BASE * edge_probabilities[i], prob_sum);
5892 new_edge->count = edge_counts[i];
5893 new_edges.safe_push (new_edge);
5895 if (i < labels - 1)
5897 /* Prepare another basic block that will contain
5898 next condition. */
5899 basic_block next_bb = create_empty_bb (cur_bb);
5900 if (current_loops)
5902 add_bb_to_loop (next_bb, cur_bb->loop_father);
5903 loops_state_set (LOOPS_NEED_FIXUP);
5906 edge next_edge = make_edge (cur_bb, next_bb, EDGE_FALSE_VALUE);
5907 next_edge->probability
5908 = inverse_probability (new_edge->probability);
5909 next_edge->count = edge_counts[0]
5910 + sum_slice <gcov_type> (edge_counts, i, labels);
5911 next_bb->frequency = EDGE_FREQUENCY (next_edge);
5912 cur_bb = next_bb;
5914 else /* Link last IF statement and default label
5915 of the switch. */
5917 edge e = make_edge (cur_bb, default_label_bb, EDGE_FALSE_VALUE);
5918 e->probability = inverse_probability (new_edge->probability);
5919 e->count = edge_counts[0];
5920 new_edges.safe_insert (0, e);
5924 /* Restore original PHI immediate value. */
5925 for (unsigned i = 0; i < phi_todo_list.length (); i++)
5927 phi_definition *phi_def = phi_todo_list[i];
5928 edge new_edge = new_edges[phi_def->label_index];
5930 gphi_iterator it = gsi_start_phis (new_edge->dest);
5931 for (unsigned i = 0; i < phi_def->phi_index; i++)
5932 gsi_next (&it);
5934 gphi *phi = it.phi ();
5935 add_phi_arg (phi, phi_def->phi_value, new_edge, UNKNOWN_LOCATION);
5936 delete phi_def;
5939 /* Remove the original GIMPLE switch statement. */
5940 gsi_remove (&gsi, true);
5944 if (dump_file)
5945 dump_function_to_file (current_function_decl, dump_file, TDF_DETAILS);
5947 if (need_update)
5949 free_dominance_info (CDI_DOMINATORS);
5950 calculate_dominance_info (CDI_DOMINATORS);
5954 /* Expand builtins that can't be handled by HSA back-end. */
5956 static void
5957 expand_builtins ()
5959 function *func = DECL_STRUCT_FUNCTION (current_function_decl);
5960 basic_block bb;
5962 FOR_EACH_BB_FN (bb, func)
5964 for (gimple_stmt_iterator gsi = gsi_start_bb (bb); !gsi_end_p (gsi);
5965 gsi_next (&gsi))
5967 gimple *stmt = gsi_stmt (gsi);
5969 if (gimple_code (stmt) != GIMPLE_CALL)
5970 continue;
5972 gcall *call = as_a <gcall *> (stmt);
5974 if (!gimple_call_builtin_p (call, BUILT_IN_NORMAL))
5975 continue;
5977 tree fndecl = gimple_call_fndecl (stmt);
5978 enum built_in_function fn = DECL_FUNCTION_CODE (fndecl);
5979 switch (fn)
5981 case BUILT_IN_CEXPF:
5982 case BUILT_IN_CEXPIF:
5983 case BUILT_IN_CEXPI:
5985 /* Similar to builtins.c (expand_builtin_cexpi), the builtin
5986 can be transformed to: cexp(I * z) = ccos(z) + I * csin(z). */
5987 tree lhs = gimple_call_lhs (stmt);
5988 tree rhs = gimple_call_arg (stmt, 0);
5989 tree rhs_type = TREE_TYPE (rhs);
5990 bool float_type_p = rhs_type == float_type_node;
5991 tree real_part = make_temp_ssa_name (rhs_type, NULL,
5992 "cexp_real_part");
5993 tree imag_part = make_temp_ssa_name (rhs_type, NULL,
5994 "cexp_imag_part");
5996 tree cos_fndecl
5997 = mathfn_built_in (rhs_type, fn == float_type_p
5998 ? BUILT_IN_COSF : BUILT_IN_COS);
5999 gcall *cos = gimple_build_call (cos_fndecl, 1, rhs);
6000 gimple_call_set_lhs (cos, real_part);
6001 gsi_insert_before (&gsi, cos, GSI_SAME_STMT);
6003 tree sin_fndecl
6004 = mathfn_built_in (rhs_type, fn == float_type_p
6005 ? BUILT_IN_SINF : BUILT_IN_SIN);
6006 gcall *sin = gimple_build_call (sin_fndecl, 1, rhs);
6007 gimple_call_set_lhs (sin, imag_part);
6008 gsi_insert_before (&gsi, sin, GSI_SAME_STMT);
6011 gassign *assign = gimple_build_assign (lhs, COMPLEX_EXPR,
6012 real_part, imag_part);
6013 gsi_insert_before (&gsi, assign, GSI_SAME_STMT);
6014 gsi_remove (&gsi, true);
6016 break;
6018 default:
6019 break;
6025 /* Emit HSA module variables that are global for the entire module. */
6027 static void
6028 emit_hsa_module_variables (void)
6030 hsa_num_threads = new hsa_symbol (BRIG_TYPE_U32, BRIG_SEGMENT_PRIVATE,
6031 BRIG_LINKAGE_MODULE, true);
6033 hsa_num_threads->m_name = "hsa_num_threads";
6035 hsa_brig_emit_omp_symbols ();
6038 /* Generate HSAIL representation of the current function and write into a
6039 special section of the output file. If KERNEL is set, the function will be
6040 considered an HSA kernel callable from the host, otherwise it will be
6041 compiled as an HSA function callable from other HSA code. */
6043 static void
6044 generate_hsa (bool kernel)
6046 hsa_init_data_for_cfun ();
6048 if (hsa_num_threads == NULL)
6049 emit_hsa_module_variables ();
6051 /* Initialize hsa_cfun. */
6052 hsa_cfun = new hsa_function_representation (cfun->decl, kernel,
6053 SSANAMES (cfun)->length ());
6054 hsa_cfun->init_extra_bbs ();
6056 if (flag_tm)
6058 HSA_SORRY_AT (UNKNOWN_LOCATION,
6059 "support for HSA does not implement transactional memory");
6060 goto fail;
6063 verify_function_arguments (cfun->decl);
6064 if (hsa_seen_error ())
6065 goto fail;
6067 hsa_cfun->m_name = get_brig_function_name (cfun->decl);
6069 gen_function_def_parameters ();
6070 if (hsa_seen_error ())
6071 goto fail;
6073 init_prologue ();
6075 gen_body_from_gimple ();
6076 if (hsa_seen_error ())
6077 goto fail;
6079 if (hsa_cfun->m_kernel_dispatch_count)
6080 init_hsa_num_threads ();
6082 if (hsa_cfun->m_kern_p)
6084 hsa_function_summary *s
6085 = hsa_summaries->get (cgraph_node::get (hsa_cfun->m_decl));
6086 hsa_add_kern_decl_mapping (current_function_decl, hsa_cfun->m_name,
6087 hsa_cfun->m_maximum_omp_data_size,
6088 s->m_gridified_kernel_p);
6091 #ifdef ENABLE_CHECKING
6092 for (unsigned i = 0; i < hsa_cfun->m_ssa_map.length (); i++)
6093 if (hsa_cfun->m_ssa_map[i])
6094 hsa_cfun->m_ssa_map[i]->verify_ssa ();
6096 basic_block bb;
6097 FOR_EACH_BB_FN (bb, cfun)
6099 hsa_bb *hbb = hsa_bb_for_bb (bb);
6101 for (hsa_insn_basic *insn = hbb->m_first_insn; insn; insn = insn->m_next)
6102 insn->verify ();
6105 #endif
6107 hsa_regalloc ();
6108 hsa_brig_emit_function ();
6110 fail:
6111 hsa_deinit_data_for_cfun ();
6114 namespace {
6116 const pass_data pass_data_gen_hsail =
6118 GIMPLE_PASS,
6119 "hsagen", /* name */
6120 OPTGROUP_NONE, /* optinfo_flags */
6121 TV_NONE, /* tv_id */
6122 PROP_cfg | PROP_ssa, /* properties_required */
6123 0, /* properties_provided */
6124 0, /* properties_destroyed */
6125 0, /* todo_flags_start */
6126 0 /* todo_flags_finish */
6129 class pass_gen_hsail : public gimple_opt_pass
6131 public:
6132 pass_gen_hsail (gcc::context *ctxt)
6133 : gimple_opt_pass(pass_data_gen_hsail, ctxt)
6136 /* opt_pass methods: */
6137 bool gate (function *);
6138 unsigned int execute (function *);
6140 }; // class pass_gen_hsail
6142 /* Determine whether or not to run generation of HSAIL. */
6144 bool
6145 pass_gen_hsail::gate (function *f)
6147 return hsa_gen_requested_p ()
6148 && hsa_gpu_implementation_p (f->decl);
6151 unsigned int
6152 pass_gen_hsail::execute (function *)
6154 hsa_function_summary *s
6155 = hsa_summaries->get (cgraph_node::get_create (current_function_decl));
6157 convert_switch_statements ();
6158 expand_builtins ();
6159 generate_hsa (s->m_kind == HSA_KERNEL);
6160 TREE_ASM_WRITTEN (current_function_decl) = 1;
6161 return TODO_discard_function;
6164 } // anon namespace
6166 /* Create the instance of hsa gen pass. */
6168 gimple_opt_pass *
6169 make_pass_gen_hsail (gcc::context *ctxt)
6171 return new pass_gen_hsail (ctxt);