clang/utils/convert_arm_neon.py

   1 #!/usr/bin/env python3
   2
   3 # This script was committed on 20/11/2019 and it would probably make sense to remove
   4 # it after the next release branches.
   5
   6 # This script is pipe based and converts an arm_neon.td (or arm_fp16.td) file
   7 # using the old single-char type modifiers to an equivalent new-style form where
   8 # each modifier is orthogonal and they can be composed.
   9 #
  10 # It was used to directly generate the .td files on main, so if you have any
  11 # local additions I would suggest implementing any modifiers here, and running
  12 # it over your entire pre-merge .td files rather than trying to resolve any
  13 # conflicts manually.
  14
  15 import re, sys
  16
  17 MOD_MAP = {
  18     "v": "v",
  19     "x": "S",
  20     "u": "U",
  21     "d": ".",
  22     "g": "q",
  23     "j": "Q",
  24     "w": ">Q",
  25     "n": ">",
  26     "h": "<",
  27     "q": "<Q",
  28     "e": "<U",
  29     "m": "<q",
  30     "i": "I",
  31     "l": "IU>",
  32     "s": "1",
  33     "z": "1<",
  34     "r": "1>",
  35     "b": "1U",
  36     "$": "1S",
  37     "k": "Q",
  38     "2": "2",
  39     "3": "3",
  40     "4": "4",
  41     "B": "2Q",
  42     "C": "3Q",
  43     "D": "4Q",
  44     "p": "*",
  45     "c": "c*",
  46     "7": "<<q",
  47     "8": "<<",
  48     "9": "<<Q",
  49     "t": "p",
  50 }
  51
  52
  53 def typespec_elt_size(typespec):
  54     if "c" in typespec:
  55         return 8
  56     elif "s" in typespec or "h" in typespec:
  57         return 16
  58     elif "i" in typespec or "f" in typespec:
  59         return 32
  60     elif "l" in typespec or "d" in typespec:
  61         return 64
  62     elif "k" in typespec:
  63         return 128
  64
  65
  66 def get_resize(cur, desired):
  67     res = ""
  68     while cur < desired:
  69         res += ">"
  70         cur *= 2
  71     while cur > desired:
  72         res += "<"
  73         cur /= 2
  74     return res
  75
  76
  77 def remap_protocol(proto, typespec, name):
  78     key_type = 0
  79
  80     # Conversions like to see the integer type so they know signedness.
  81     if (
  82         "vcvt" in name
  83         and "_f" in name
  84         and name != "vcvt_f32_f64"
  85         and name != "vcvt_f64_f32"
  86     ):
  87         key_type = 1
  88     default_width = typespec_elt_size(typespec)
  89     inconsistent_width = False
  90     for elt in typespec:
  91         new_width = typespec_elt_size(elt)
  92         if new_width and new_width != default_width:
  93             inconsistent_width = True
  94
  95     res = ""
  96     for i, c in enumerate(proto):
  97         # void and pointers make for bad discriminators in CGBuiltin.cpp.
  98         if c in "vcp":
  99             key_type += 1
 100
 101         if c in MOD_MAP:
 102             cur_mod = MOD_MAP[c]
 103         elif inconsistent_width:
 104             # Otherwise it's a fixed output width modifier.
 105             sys.stderr.write(
 106                 f"warning: {name} uses fixed output size but has inconsistent input widths: {proto} {typespec}\n"
 107             )
 108
 109         if c == "Y":
 110             # y: scalar of half float
 111             resize = get_resize(default_width, 16)
 112             cur_mod = f"1F{resize}"
 113         elif c == "y":
 114             # y: scalar of float
 115             resize = get_resize(default_width, 32)
 116             cur_mod = f"1F{resize}"
 117         elif c == "o":
 118             # o: scalar of double
 119             resize = get_resize(default_width, 64)
 120             cur_mod = f"1F{resize}"
 121         elif c == "I":
 122             # I: scalar of 32-bit signed
 123             resize = get_resize(default_width, 32)
 124             cur_mod = f"1S{resize}"
 125         elif c == "L":
 126             # L: scalar of 64-bit signed
 127             resize = get_resize(default_width, 64)
 128             cur_mod = f"1S{resize}"
 129         elif c == "U":
 130             # I: scalar of 32-bit unsigned
 131             resize = get_resize(default_width, 32)
 132             cur_mod = f"1U{resize}"
 133         elif c == "O":
 134             # O: scalar of 64-bit unsigned
 135             resize = get_resize(default_width, 64)
 136             cur_mod = f"1U{resize}"
 137         elif c == "f":
 138             # f: float (int args)
 139             resize = get_resize(default_width, 32)
 140             cur_mod = f"F{resize}"
 141         elif c == "F":
 142             # F: double (int args)
 143             resize = get_resize(default_width, 64)
 144             cur_mod = f"F{resize}"
 145         elif c == "H":
 146             # H: half (int args)
 147             resize = get_resize(default_width, 16)
 148             cur_mod = f"F{resize}"
 149         elif c == "0":
 150             # 0: half (int args), ignore 'Q' size modifier.
 151             resize = get_resize(default_width, 16)
 152             cur_mod = f"Fq{resize}"
 153         elif c == "1":
 154             # 1: half (int args), force 'Q' size modifier.
 155             resize = get_resize(default_width, 16)
 156             cur_mod = f"FQ{resize}"
 157
 158         if len(cur_mod) == 0:
 159             raise Exception(f"WTF: {c} in {name}")
 160
 161         if key_type != 0 and key_type == i:
 162             cur_mod += "!"
 163
 164         if len(cur_mod) == 1:
 165             res += cur_mod
 166         else:
 167             res += "(" + cur_mod + ")"
 168
 169     return res
 170
 171
 172 def replace_insts(m):
 173     start, end = m.span("proto")
 174     start -= m.start()
 175     end -= m.start()
 176     new_proto = remap_protocol(m["proto"], m["kinds"], m["name"])
 177     return m.group()[:start] + new_proto + m.group()[end:]
 178
 179
 180 INST = re.compile(r'Inst<"(?P<name>.*?)",\s*"(?P<proto>.*?)",\s*"(?P<kinds>.*?)"')
 181
 182 new_td = INST.sub(replace_insts, sys.stdin.read())
 183 sys.stdout.write(new_td)