xref: /llvm-project/mlir/utils/tree-sitter-mlir/test/highlight/downstream.mlir (revision 10f8be19e7562624c0e63872e28cc963e03e56a5)
1func.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}
28