mlir/utils/tree-sitter-mlir/queries/highlights.scm

   1 [
   2   "ins"
   3   "outs"
   4   "else"
   5   "do"
   6   "loc"
   7   "attributes"
   8   "into"
   9   "to"
  10   "from"
  11   "step"
  12   "low"
  13   "high"
  14   "iter_args"
  15   "padding_value"
  16   "inner_tiles"
  17   "gather_dims"
  18   "scatter_dims"
  19   "outer_dims_perm"
  20   "inner_dims_pos"
  21   "shared_outs"
  22   "default"
  23
  24   "eq"
  25   "ne"
  26   "oeq"
  27   "olt"
  28   "ole"
  29   "ogt"
  30   "oge"
  31   "slt"
  32   "sle"
  33   "sgt"
  34   "sge"
  35   "ult"
  36   "ule"
  37   "ugt"
  38   "uge"
  39 ] @keyword
  40
  41 [
  42   "module"
  43   "unrealized_conversion_cast"
  44
  45   "func.call"
  46   "call"
  47   "func.call_indirect"
  48   "call_indirect"
  49   "func.constant"
  50   "constant"
  51   "func.func"
  52   "func.return"
  53   "return"
  54
  55   "llvm.func"
  56   "llvm.return"
  57
  58   "cf.assert"
  59   "cf.br"
  60   "cf.cond_br"
  61   "cf.switch"
  62
  63   "scf.condition"
  64   "scf.execute_region"
  65   "scf.if"
  66   "scf.index_switch"
  67   "scf.for"
  68   "scf.forall"
  69   "scf.forall.in_parallel"
  70   "scf.parallel"
  71   "scf.reduce"
  72   "scf.reduce.return"
  73   "scf.while"
  74   "scf.yield"
  75
  76   "arith.constant"
  77   "arith.addi"
  78   "arith.subi"
  79   "arith.divsi"
  80   "arith.divui"
  81   "arith.ceildivsi"
  82   "arith.ceildivui"
  83   "arith.floordivsi"
  84   "arith.remsi"
  85   "arith.remui"
  86   "arith.muli"
  87   "arith.mulsi_extended"
  88   "arith.mului_extended"
  89   "arith.andi"
  90   "arith.ori"
  91   "arith.xori"
  92   "arith.maxsi"
  93   "arith.maxui"
  94   "arith.minsi"
  95   "arith.minui"
  96   "arith.shli"
  97   "arith.shrsi"
  98   "arith.shrui"
  99   "arith.addui_extended"
 100   "arith.addf"
 101   "arith.divf"
 102   "arith.maximumf"
 103   "arith.minimumf"
 104   "arith.mulf"
 105   "arith.remf"
 106   "arith.subf"
 107   "arith.negf"
 108   "arith.cmpi"
 109   "arith.cmpf"
 110   "arith.extf"
 111   "arith.extsi"
 112   "arith.extui"
 113   "arith.fptosi"
 114   "arith.fptoui"
 115   "arith.index_cast"
 116   "arith.index_castui"
 117   "arith.sitofp"
 118   "arith.uitofp"
 119   "arith.bitcast"
 120   "arith.truncf"
 121   "arith.select"
 122
 123   "math.absf"
 124   "math.atan"
 125   "math.cbrt"
 126   "math.ceil"
 127   "math.cos"
 128   "math.erf"
 129   "math.exp"
 130   "math.exp2"
 131   "math.expm1"
 132   "math.floor"
 133   "math.log"
 134   "math.log10"
 135   "math.log1p"
 136   "math.log2"
 137   "math.round"
 138   "math.roundeven"
 139   "math.rsqrt"
 140   "math.sin"
 141   "math.sqrt"
 142   "math.tan"
 143   "math.tanh"
 144   "math.trunc"
 145   "math.absi"
 146   "math.ctlz"
 147   "math.cttz"
 148   "math.ctpop"
 149   "math.atan2"
 150   "math.copysign"
 151   "math.fpowi"
 152   "math.powf"
 153   "math.ipowi"
 154   "math.fma"
 155
 156   "memref.alloc"
 157   "memref.cast"
 158   "memref.copy"
 159   "memref.collapse_shape"
 160   "memref.expand_shape"
 161   "memref.prefetch"
 162   "memref.rank"
 163   "memref.realloc"
 164   "memref.view"
 165
 166   "vector.bitcast"
 167   "vector.broadcast"
 168   "vector.shape_cast"
 169   "vector.type_cast"
 170   "vector.constant_mask"
 171   "vector.create_mask"
 172   "vector.extract"
 173   "vector.load"
 174   "vector.scalable.extract"
 175   "vector.fma"
 176   "vector.flat_transpose"
 177   "vector.insert"
 178   "vector.scalable.insert"
 179   "vector.shuffle"
 180   "vector.store"
 181   "vector.insert_strided_slice"
 182   "vector.matrix_multiply"
 183   "vector.print"
 184   "vector.splat"
 185   "vector.transfer_read"
 186   "vector.transfer_write"
 187   "vector.yield"
 188
 189   "tensor.empty"
 190   "tensor.cast"
 191   "tensor.dim"
 192   "tensor.collapse_shape"
 193   "tensor.expand_shape"
 194   "tensor.extract"
 195   "tensor.insert"
 196   "tensor.extract_slice"
 197   "tensor.insert_slice"
 198   "tensor.parallel_insert_slice"
 199   "tensor.from_elements"
 200   "tensor.gather"
 201   "tensor.scatter"
 202   "tensor.pad"
 203   "tensor.reshape"
 204   "tensor.splat"
 205   "tensor.pack"
 206   "tensor.unpack"
 207   "tensor.generate"
 208   "tensor.rank"
 209   "tensor.yield"
 210
 211   "bufferization.alloc_tensor"
 212   "bufferization.to_memref"
 213   "bufferization.to_tensor"
 214
 215   "linalg.batch_matmul"
 216   "linalg.batch_matmul_transpose_b"
 217   "linalg.batch_matvec"
 218   "linalg.batch_reduce_matmul"
 219   "linalg.broadcast"
 220   "linalg.conv_1d_ncw_fcw"
 221   "linalg.conv_1d_nwc_wcf"
 222   "linalg.conv_1d"
 223   "linalg.conv_2d_nchw_fchw"
 224   "linalg.conv_2d_ngchw_fgchw"
 225   "linalg.conv_2d_nhwc_fhwc"
 226   "linalg.conv_2d_nhwc_hwcf"
 227   "linalg.conv_2d_nhwc_hwcf_q"
 228   "linalg.conv_2d"
 229   "linalg.conv_3d_ndhwc_dhwcf"
 230   "linalg.conv_3d_ndhwc_dhwcf_q"
 231   "linalg.conv_3d"
 232   "linalg.copy"
 233   "linalg.depthwise_conv_1d_nwc_wcm"
 234   "linalg.depthwise_conv_2d_nchw_chw"
 235   "linalg.depthwise_conv_2d_nhwc_hwc"
 236   "linalg.depthwise_conv_2d_nhwc_hwc_q"
 237   "linalg.depthwise_conv_2d_nhwc_hwcm"
 238   "linalg.depthwise_conv_2d_nhwc_hwcm_q"
 239   "linalg.depthwise_conv_3d_ndhwc_dhwc"
 240   "linalg.depthwise_conv_3d_ndhwc_dhwcm"
 241   "linalg.dot"
 242   "linalg.elemwise_binary"
 243   "linalg.elemwise_unary"
 244   "linalg.fill"
 245   "linalg.fill_rng_2d"
 246   "linalg.matmul"
 247   "linalg.matmul_transpose_b"
 248   "linalg.matmul_unsigned"
 249   "linalg.matvec"
 250   "linalg.mmt4d"
 251   "linalg.pooling_nchw_max"
 252   "linalg.pooling_nchw_sum"
 253   "linalg.pooling_ncw_max"
 254   "linalg.pooling_ncw_sum"
 255   "linalg.pooling_ndhwc_max"
 256   "linalg.pooling_ndhwc_min"
 257   "linalg.pooling_ndhwc_sum"
 258   "linalg.pooling_nhwc_max"
 259   "linalg.pooling_nhwc_max_unsigned"
 260   "linalg.pooling_nhwc_min"
 261   "linalg.pooling_nhwc_min_unsigned"
 262   "linalg.pooling_nhwc_sum"
 263   "linalg.pooling_nwc_max"
 264   "linalg.pooling_nwc_max_unsigned"
 265   "linalg.pooling_nwc_min"
 266   "linalg.pooling_nwc_min_unsigned"
 267   "linalg.pooling_nwc_sum"
 268   "linalg.quantized_batch_matmul"
 269   "linalg.quantized_matmul"
 270   "linalg.vecmat"
 271   "linalg.generic"
 272   "linalg.index"
 273   "linalg.map"
 274   "linalg.yield"
 275 ] @function.builtin
 276
 277 (generic_operation) @function
 278
 279 (builtin_type) @type.builtin
 280
 281 [
 282   (type_alias)
 283   (dialect_type)
 284   (type_alias_def)
 285 ] @type
 286
 287 [
 288   (integer_literal)
 289   (float_literal)
 290   (complex_literal)
 291 ] @number
 292
 293 [
 294   (bool_literal)
 295   (tensor_literal)
 296   (array_literal)
 297   (unit_literal)
 298 ] @constant.builtin
 299
 300 (string_literal) @string
 301
 302 [
 303   (attribute_alias_def)
 304   (attribute_alias)
 305   (bare_attribute_entry)
 306   (attribute)
 307   (fastmath_attr)
 308   (scatter_dims_attr)
 309   (gather_dims_attr)
 310   (outer_dims_perm_attr)
 311   (inner_dims_pos_attr)
 312   (inner_tiles_attr)
 313   (unique_attr)
 314   (nofold_attr)
 315   (isWrite_attr)
 316   (localityHint_attr)
 317   (isDataCache_attr)
 318   (restrict_attr)
 319   (writable_attr)
 320 ] @attribute
 321
 322 [
 323   "("
 324   ")"
 325   "{"
 326   "}"
 327   "["
 328   "]"
 329 ] @punctuation.bracket
 330
 331 [
 332   ":"
 333   ","
 334 ] @punctuation.delimeter
 335
 336 [
 337   "="
 338   "->"
 339 ] @operator
 340
 341 (func_dialect name: (symbol_ref_id) @function)
 342 (llvm_dialect name: (symbol_ref_id) @function)
 343
 344 (func_arg_list (value_use) @variable.parameter)
 345 (block_arg_list (value_use) @variable.parameter)
 346
 347 (caret_id) @tag
 348 (value_use) @variable
 349 (comment) @comment