xref: /llvm-project/mlir/test/Dialect/Transform/include/Library/lower-to-llvm.mlir (revision 3517b67ef021055d201234384359fa313f3312f7)
1// RUN: mlir-opt %s
2// No need to check anything else than parsing here, this is being used by another test as data.
3
4/// Schedule to lower to LLVM.
5module @lower_module_to_llvm attributes { transform.with_named_sequence } {
6
7transform.named_sequence @lower_to_llvm(
8    %module: !transform.any_op {transform.readonly}) -> !transform.any_op {
9
10  %func = transform.structured.match ops{["func.func"]} in %module : (!transform.any_op) -> !transform.any_op
11  %f = transform.apply_registered_pass "convert-vector-to-scf" to %func : (!transform.any_op) -> !transform.any_op
12  %f2 = transform.apply_registered_pass "convert-linalg-to-loops" to %f : (!transform.any_op) -> !transform.any_op
13  %f3 = transform.apply_registered_pass "convert-scf-to-cf" to %f2 : (!transform.any_op) -> !transform.any_op
14  %f4 = transform.apply_registered_pass "expand-strided-metadata" to %f3 : (!transform.any_op) -> !transform.any_op
15  %f5 = transform.apply_registered_pass "lower-affine" to %f4 : (!transform.any_op) -> !transform.any_op
16
17  transform.apply_conversion_patterns to %f5 {
18    transform.apply_conversion_patterns.dialect_to_llvm "math"
19    transform.apply_conversion_patterns.vector.vector_to_llvm
20    transform.apply_conversion_patterns.dialect_to_llvm "memref"
21    transform.apply_conversion_patterns.func.func_to_llvm
22    transform.apply_conversion_patterns.dialect_to_llvm "index"
23    transform.apply_conversion_patterns.dialect_to_llvm "arith"
24    transform.apply_conversion_patterns.dialect_to_llvm "cf"
25  } with type_converter {
26    transform.apply_conversion_patterns.memref.memref_to_llvm_type_converter
27      {index_bitwidth = 64,
28       use_bare_ptr = false,
29       use_bare_ptr_memref_call_conv = false,
30       use_opaque_pointers = true}
31  } {
32    legal_dialects = ["llvm"],
33    partial_conversion
34  } : !transform.any_op
35
36  // Need to rematch here because:
37  //   1. applying reconcile-unrealized-casts on the whole module yields the
38  //      transform applies to transform, when called from a named sequence, at
39  //      this time.
40  //   2. apply_conversion patterns consumes the func but does not produce
41  //      a new llvm.func.
42  %f6 = transform.structured.match ops{["llvm.func"]} in %module
43    : (!transform.any_op) -> !transform.any_op
44  %f7 = transform.apply_registered_pass "reconcile-unrealized-casts" to %f6
45    : (!transform.any_op) -> !transform.any_op
46  transform.yield %module : !transform.any_op
47}
48
49} // transform module
50