mlir/utils/tree-sitter-mlir/test/highlight/downstream.mlir

   1 func.func @sort_memref(%input1: memref<?x?xf32>, %input2: memref<?x?xi32>,
   2 // <- function.builtin
   3 //        ^ function
   4 //                     ^ variable.parameter
   5 //                              ^ type.builtin
   6                        %init1: memref<?x?xf32>, %init2: memref<?x?xi32>) {
   7   thlo.sort
   8       ins(%input1: memref<?x?xf32>, %input2: memref<?x?xi32>)
   9 //    ^ keyword
  10 //                                  ^ variable.parameter
  11       outs(%init1: memref<?x?xf32>, %init2: memref<?x?xi32>)
  12 //    ^ keyword
  13 //                                  ^ variable.parameter
  14       { dimension = 0 : i64, is_stable = true }
  15 //                                       ^ constant.builtin
  16       (%e11: f32, %e12: f32, %e21: i32, %e22: i32) {
  17         %gt = arith.cmpf ogt, %e11, %e12: f32
  18 //            ^ function.builtin
  19 //                       ^ keyword
  20 //                            ^ variable
  21 //                                  ^ variable
  22 //                                        ^ type.builtin
  23         thlo.yield %gt : i1
  24       }
  25   func.return
  26 // ^ function.builtin
  27 }