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 MOD_MAP = {
  17     'v': 'v',
  18     'x': 'S',
  19     'u': 'U',
  20     'd': '.',
  21     'g': 'q',
  22     'j': 'Q',
  23     'w': '>Q',
  24     'n': '>',
  25     'h': '<',
  26     'q': '<Q',
  27     'e': '<U',
  28     'm': '<q',
  29     'i': 'I',
  30     'l': 'IU>',
  31     's': '1',
  32     'z': '1<',
  33     'r': '1>',
  34     'b': '1U',
  35     '$': '1S',
  36     'k': 'Q',
  37     '2': '2',
  38     '3': '3',
  39     '4': '4',
  40     'B': '2Q',
  41     'C': '3Q',
  42     'D': '4Q',
  43     'p': '*',
  44     'c': 'c*',
  45     '7': '<<q',
  46     '8': '<<',
  47     '9': '<<Q',
  48     't': 'p'
  49     }
  50
  51
  52 def typespec_elt_size(typespec):
  53     if 'c' in typespec:
  54         return 8
  55     elif 's' in typespec or 'h' in typespec:
  56         return 16
  57     elif 'i' in typespec or 'f' in typespec:
  58         return 32
  59     elif 'l' in typespec or 'd' in typespec:
  60         return 64
  61     elif 'k' in typespec:
  62         return 128
  63
  64 def get_resize(cur, desired):
  65     res = ''
  66     while cur < desired:
  67         res += '>'
  68         cur *= 2
  69     while cur > desired:
  70         res += '<'
  71         cur /= 2
  72     return res
  73
  74
  75 def remap_protocol(proto, typespec, name):
  76     key_type = 0
  77
  78     # Conversions like to see the integer type so they know signedness.
  79     if 'vcvt' in name and '_f' in name and name != 'vcvt_f32_f64' and name != 'vcvt_f64_f32':
  80         key_type = 1
  81     default_width = typespec_elt_size(typespec)
  82     inconsistent_width = False
  83     for elt in typespec:
  84         new_width = typespec_elt_size(elt)
  85         if new_width and new_width != default_width:
  86             inconsistent_width = True
  87
  88     res = ''
  89     for i, c in enumerate(proto):
  90         # void and pointers make for bad discriminators in CGBuiltin.cpp.
  91         if c in 'vcp':
  92                 key_type += 1
  93
  94         if c in MOD_MAP:
  95             cur_mod = MOD_MAP[c]
  96         elif inconsistent_width:
  97             # Otherwise it's a fixed output width modifier.
  98             sys.stderr.write(f'warning: {name} uses fixed output size but has inconsistent input widths: {proto} {typespec}\n')
  99
 100         if c == 'Y':
 101             # y: scalar of half float
 102             resize = get_resize(default_width, 16)
 103             cur_mod = f'1F{resize}'
 104         elif c == 'y':
 105             # y: scalar of float
 106             resize = get_resize(default_width, 32)
 107             cur_mod = f'1F{resize}'
 108         elif c == 'o':
 109             # o: scalar of double
 110             resize = get_resize(default_width, 64)
 111             cur_mod = f'1F{resize}'
 112         elif c == 'I':
 113             # I: scalar of 32-bit signed
 114             resize = get_resize(default_width, 32)
 115             cur_mod = f'1S{resize}'
 116         elif c == 'L':
 117             # L: scalar of 64-bit signed
 118             resize = get_resize(default_width, 64)
 119             cur_mod = f'1S{resize}'
 120         elif c == 'U':
 121             # I: scalar of 32-bit unsigned
 122             resize = get_resize(default_width, 32)
 123             cur_mod = f'1U{resize}'
 124         elif c == 'O':
 125             # O: scalar of 64-bit unsigned
 126             resize = get_resize(default_width, 64)
 127             cur_mod = f'1U{resize}'
 128         elif c == 'f':
 129             # f: float (int args)
 130             resize = get_resize(default_width, 32)
 131             cur_mod = f'F{resize}'
 132         elif c == 'F':
 133             # F: double (int args)
 134             resize = get_resize(default_width, 64)
 135             cur_mod = f'F{resize}'
 136         elif c == 'H':
 137             # H: half (int args)
 138             resize = get_resize(default_width, 16)
 139             cur_mod = f'F{resize}'
 140         elif c == '0':
 141             # 0: half (int args), ignore 'Q' size modifier.
 142             resize = get_resize(default_width, 16)
 143             cur_mod = f'Fq{resize}'
 144         elif c == '1':
 145             # 1: half (int args), force 'Q' size modifier.
 146             resize = get_resize(default_width, 16)
 147             cur_mod = f'FQ{resize}'
 148
 149         if len(cur_mod) == 0:
 150             raise Exception(f'WTF: {c} in {name}')
 151
 152         if key_type != 0 and key_type == i:
 153             cur_mod += '!'
 154
 155         if len(cur_mod) == 1:
 156             res += cur_mod
 157         else:
 158             res += '(' + cur_mod + ')'
 159
 160     return res
 161
 162 def replace_insts(m):
 163     start, end = m.span('proto')
 164     start -= m.start()
 165     end -= m.start()
 166     new_proto = remap_protocol(m['proto'], m['kinds'], m['name'])
 167     return m.group()[:start] + new_proto + m.group()[end:]
 168
 169 INST = re.compile(r'Inst<"(?P<name>.*?)",\s*"(?P<proto>.*?)",\s*"(?P<kinds>.*?)"')
 170
 171 new_td = INST.sub(replace_insts, sys.stdin.read())
 172 sys.stdout.write(new_td)