28 lines · plain
1func.func @sort_memref(%input1: memref<?x?xf32>, %input2: memref<?x?xi32>,2// <- function.builtin3// ^ function4// ^ variable.parameter5// ^ type.builtin6 %init1: memref<?x?xf32>, %init2: memref<?x?xi32>) {7 thlo.sort8 ins(%input1: memref<?x?xf32>, %input2: memref<?x?xi32>)9// ^ keyword10// ^ variable.parameter11 outs(%init1: memref<?x?xf32>, %init2: memref<?x?xi32>)12// ^ keyword13// ^ variable.parameter14 { dimension = 0 : i64, is_stable = true }15// ^ constant.builtin16 (%e11: f32, %e12: f32, %e21: i32, %e22: i32) {17 %gt = arith.cmpf ogt, %e11, %e12: f3218// ^ function.builtin19// ^ keyword20// ^ variable21// ^ variable22// ^ type.builtin23 thlo.yield %gt : i124 }25 func.return26// ^ function.builtin27}28