49 lines · plain
1// RUN: mlir-opt %s2 3/// Schedule to lower to LLVM.4module @lower_module_to_llvm attributes { transform.with_named_sequence } {5 6transform.named_sequence @lower_to_llvm(7 %module: !transform.any_op {transform.readonly}) -> !transform.any_op {8 9 %func = transform.structured.match ops{["func.func"]} in %module : (!transform.any_op) -> !transform.any_op10 %f = transform.apply_registered_pass "convert-vector-to-scf" to %func : (!transform.any_op) -> !transform.any_op11 %f2 = transform.apply_registered_pass "convert-linalg-to-loops" to %f : (!transform.any_op) -> !transform.any_op12 %f3 = transform.apply_registered_pass "convert-scf-to-cf" to %f2 : (!transform.any_op) -> !transform.any_op13 %f4 = transform.apply_registered_pass "expand-strided-metadata" to %f3 : (!transform.any_op) -> !transform.any_op14 %f5 = transform.apply_registered_pass "lower-affine" to %f4 : (!transform.any_op) -> !transform.any_op15 16 transform.apply_conversion_patterns to %f5 {17 transform.apply_conversion_patterns.dialect_to_llvm "math"18 transform.apply_conversion_patterns.vector.vector_to_llvm19 transform.apply_conversion_patterns.dialect_to_llvm "memref"20 transform.apply_conversion_patterns.func.func_to_llvm21 transform.apply_conversion_patterns.dialect_to_llvm "index"22 transform.apply_conversion_patterns.dialect_to_llvm "arith"23 transform.apply_conversion_patterns.dialect_to_llvm "cf"24 } with type_converter {25 transform.apply_conversion_patterns.memref.memref_to_llvm_type_converter26 {index_bitwidth = 64,27 use_bare_ptr = false,28 use_bare_ptr_memref_call_conv = false,29 use_opaque_pointers = true}30 } {31 legal_dialects = ["llvm"],32 partial_conversion33 } : !transform.any_op34 35 // Need to rematch here because:36 // 1. applying reconcile-unrealized-casts on the whole module yields the37 // transform applies to transform, when called from a named sequence, at38 // this time.39 // 2. apply_conversion patterns consumes the func but does not produce 40 // a new llvm.func.41 %f6 = transform.structured.match ops{["llvm.func"]} in %module 42 : (!transform.any_op) -> !transform.any_op43 %f7 = transform.apply_registered_pass "reconcile-unrealized-casts" to %f644 : (!transform.any_op) -> !transform.any_op45 transform.yield %module : !transform.any_op46}47 48} // transform module49