gcc/brig/brigfrontend/brig-cmp-inst-handler.cc

   1 /* brig-cmp-inst-handler.cc -- brig cmp instruction handling
   2    Copyright (C) 2016-2019 Free Software Foundation, Inc.
   3    Contributed by Pekka Jaaskelainen <pekka.jaaskelainen@parmance.com>
   4    for General Processor Tech.
   5
   6 This file is part of GCC.
   7
   8 GCC is free software; you can redistribute it and/or modify it under
   9 the terms of the GNU General Public License as published by the Free
  10 Software Foundation; either version 3, or (at your option) any later
  11 version.
  12
  13 GCC is distributed in the hope that it will be useful, but WITHOUT ANY
  14 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  15 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  16 for more details.
  17
  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/>.  */
  21
  22 #include "brig-code-entry-handler.h"
  23 #include "diagnostic.h"
  24 #include "tree-pretty-print.h"
  25 #include "print-tree.h"
  26 #include "brig-util.h"
  27 #include "convert.h"
  28
  29 size_t
  30 brig_cmp_inst_handler::operator () (const BrigBase *base)
  31 {
  32   const BrigInstBase *inst_base = (const BrigInstBase *) base;
  33   const BrigInstCmp *inst = (const BrigInstCmp *) base;
  34
  35   tree cmp_type = get_tree_expr_type_for_hsa_type (inst->sourceType);
  36
  37   /* The destination type to convert the comparison result to.  */
  38   tree dest_type = gccbrig_tree_type_for_hsa_type (inst_base->type);
  39
  40   const bool is_fp16_dest
  41     = (inst_base->type & BRIG_TYPE_BASE_MASK) == BRIG_TYPE_F16;
  42   const bool is_boolean_dest
  43     = (inst_base->type & BRIG_TYPE_BASE_MASK) == BRIG_TYPE_B1;
  44
  45   bool is_int_cmp = VECTOR_TYPE_P (cmp_type)
  46                       ? INTEGRAL_TYPE_P (TREE_TYPE (cmp_type))
  47                       : INTEGRAL_TYPE_P (cmp_type);
  48
  49   /* The type for the GENERIC comparison.  It should match the
  50      input operand width for vector comparisons, a boolean
  51      otherwise.  */
  52   tree result_type = get_comparison_result_type (cmp_type);
  53
  54   /* Save the result as a boolean and extend/convert it to the
  55      wanted destination type.  */
  56   tree expr = NULL_TREE;
  57
  58   std::vector<tree> operands = build_operands (*inst_base);
  59
  60   switch (inst->compare)
  61     {
  62     case BRIG_COMPARE_SEQ:
  63     case BRIG_COMPARE_EQ:
  64       expr = build2 (EQ_EXPR, result_type, operands[1], operands[2]);
  65       break;
  66     case BRIG_COMPARE_SNE:
  67     case BRIG_COMPARE_NE:
  68       expr = build2 (NE_EXPR, result_type, operands[1], operands[2]);
  69
  70       if (!is_int_cmp)
  71         expr = build2 (BIT_AND_EXPR, TREE_TYPE (expr),
  72                        expr,
  73                        build2 (ORDERED_EXPR, result_type, operands[1],
  74                                operands[2]));
  75       break;
  76     case BRIG_COMPARE_SLT:
  77     case BRIG_COMPARE_LT:
  78       expr = build2 (LT_EXPR, result_type, operands[1], operands[2]);
  79       break;
  80     case BRIG_COMPARE_SLE:
  81     case BRIG_COMPARE_LE:
  82       expr = build2 (LE_EXPR, result_type, operands[1], operands[2]);
  83       break;
  84     case BRIG_COMPARE_SGT:
  85     case BRIG_COMPARE_GT:
  86       expr = build2 (GT_EXPR, result_type, operands[1], operands[2]);
  87       break;
  88     case BRIG_COMPARE_SGE:
  89     case BRIG_COMPARE_GE:
  90       expr = build2 (GE_EXPR, result_type, operands[1], operands[2]);
  91       break;
  92     case BRIG_COMPARE_SEQU:
  93     case BRIG_COMPARE_EQU:
  94       expr = build2 (UNEQ_EXPR, result_type, operands[1], operands[2]);
  95       break;
  96     case BRIG_COMPARE_SNEU:
  97     case BRIG_COMPARE_NEU:
  98       expr = build2 (NE_EXPR, result_type, operands[1], operands[2]);
  99       break;
 100     case BRIG_COMPARE_SLTU:
 101     case BRIG_COMPARE_LTU:
 102       expr = build2 (UNLT_EXPR, result_type, operands[1], operands[2]);
 103       break;
 104     case BRIG_COMPARE_SLEU:
 105     case BRIG_COMPARE_LEU:
 106       expr = build2 (UNLE_EXPR, result_type, operands[1], operands[2]);
 107       break;
 108     case BRIG_COMPARE_SGTU:
 109     case BRIG_COMPARE_GTU:
 110       expr = build2 (UNGT_EXPR, result_type, operands[1], operands[2]);
 111       break;
 112     case BRIG_COMPARE_SGEU:
 113     case BRIG_COMPARE_GEU:
 114       expr = build2 (UNGE_EXPR, result_type, operands[1], operands[2]);
 115       break;
 116     case BRIG_COMPARE_SNUM:
 117     case BRIG_COMPARE_NUM:
 118       expr = build2 (ORDERED_EXPR, result_type, operands[1], operands[2]);
 119       break;
 120     case BRIG_COMPARE_SNAN:
 121     case BRIG_COMPARE_NAN:
 122       expr = build2 (UNORDERED_EXPR, result_type, operands[1], operands[2]);
 123       break;
 124     default:
 125       break;
 126     }
 127
 128   if (expr == NULL_TREE)
 129     gcc_unreachable ();
 130
 131   if (is_fp16_dest)
 132     {
 133       expr = convert_to_real (brig_to_generic::s_fp32_type, expr);
 134     }
 135   else if (VECTOR_TYPE_P (dest_type) && ANY_INTEGRAL_TYPE_P (dest_type)
 136            && !is_boolean_dest
 137            && (inst->sourceType & BRIG_TYPE_BASE_MASK) != BRIG_TYPE_F16)
 138     {
 139       /* In later gcc versions, the output of comparison is not
 140          all ones for vectors like still in 4.9.1.  We need to use
 141          an additional VEC_COND_EXPR to produce the all ones 'true' value
 142          required by HSA.
 143          VEC_COND_EXPR <a == b, { -1, -1, -1, -1 }, { 0, 0, 0, 0 }>; */
 144
 145       tree all_ones
 146         = build_vector_from_val (dest_type,
 147                                build_minus_one_cst (TREE_TYPE (dest_type)));
 148       tree all_zeroes
 149         = build_vector_from_val (dest_type,
 150                                build_zero_cst (TREE_TYPE (dest_type)));
 151       expr = build3 (VEC_COND_EXPR, dest_type, expr, all_ones, all_zeroes);
 152     }
 153   else if (INTEGRAL_TYPE_P (dest_type) && !is_boolean_dest)
 154     {
 155       /* We need to produce the all-ones pattern for the width of the whole
 156          resulting integer type.  Use back and forth shifts for propagating
 157          the lower 1.  */
 158       tree signed_type = signed_type_for (dest_type);
 159       tree signed_result = convert_to_integer (signed_type, expr);
 160
 161       size_t result_width = int_size_in_bytes (dest_type) * BITS_PER_UNIT;
 162
 163       tree shift_amount_cst
 164         = build_int_cstu (signed_type, result_width - 1);
 165
 166       tree shift_left_result
 167         = build2 (LSHIFT_EXPR, signed_type, signed_result, shift_amount_cst);
 168
 169       expr = build2 (RSHIFT_EXPR, signed_type, shift_left_result,
 170                      shift_amount_cst);
 171     }
 172   else if (SCALAR_FLOAT_TYPE_P (dest_type))
 173     {
 174       expr = convert_to_real (dest_type, expr);
 175     }
 176   else if (VECTOR_TYPE_P (dest_type)
 177            && (inst->sourceType & BRIG_TYPE_BASE_MASK) == BRIG_TYPE_F16)
 178     {
 179       /* Because F16 comparison is emulated as an F32 comparison with S32
 180          results, we must now truncate the result vector to S16s so it
 181          fits to the destination register.  We can build the target vector
 182          type from the f16 storage type (unsigned ints).  */
 183       expr = m_parent.m_cf->add_temp_var ("wide_cmp_result", expr);
 184       tree_stl_vec wide_elements;
 185       tree_stl_vec shrunk_elements;
 186       m_parent.m_cf->unpack (expr, wide_elements);
 187       for (size_t i = 0; i < wide_elements.size (); ++i)
 188         {
 189           tree wide = wide_elements.at (i);
 190           shrunk_elements.push_back
 191             (convert_to_integer (short_integer_type_node, wide));
 192         }
 193       expr = m_parent.m_cf->pack (shrunk_elements);
 194     }
 195   build_output_assignment (*inst_base, operands[0], expr);
 196
 197   return base->byteCount;
 198 }