127 lines · plain
1// RUN: mlir-opt -pass-pipeline="builtin.module(func.func(convert-arith-to-llvm),convert-func-to-llvm,convert-cf-to-llvm,reconcile-unrealized-casts)" -split-input-file %s | FileCheck %s2// RUN: mlir-opt -pass-pipeline="builtin.module(func.func(convert-arith-to-llvm),convert-func-to-llvm{use-bare-ptr-memref-call-conv=1},convert-cf-to-llvm,reconcile-unrealized-casts)" -split-input-file %s | FileCheck %s --check-prefix=BAREPTR3// RUN: mlir-opt -pass-pipeline="builtin.module(func.func(convert-arith-to-llvm{index-bitwidth=32}),convert-func-to-llvm{index-bitwidth=32},convert-cf-to-llvm{index-bitwidth=32},reconcile-unrealized-casts)" -split-input-file %s | FileCheck %s --check-prefix=CHECK324 5// BAREPTR-LABEL: func @check_noalias6// BAREPTR-SAME: %{{.*}}: !llvm.ptr {llvm.noalias}, %{{.*}}: !llvm.ptr {llvm.noalias}7func.func @check_noalias(%static : memref<2xf32> {llvm.noalias}, %other : memref<2xf32> {llvm.noalias}) {8 return9}10 11// -----12 13// CHECK-LABEL: func @check_strided_memref_arguments(14// CHECK-COUNT-2: !llvm.ptr15// CHECK-COUNT-5: i6416// CHECK-COUNT-2: !llvm.ptr17// CHECK-COUNT-5: i6418// CHECK-COUNT-2: !llvm.ptr19// CHECK-COUNT-5: i6420func.func @check_strided_memref_arguments(%static: memref<10x20xf32, affine_map<(i,j)->(20 * i + j + 1)>>,21 %dynamic : memref<?x?xf32, affine_map<(i,j)[M]->(M * i + j + 1)>>,22 %mixed : memref<10x?xf32, affine_map<(i,j)[M]->(M * i + j + 1)>>) {23 return24}25 26// -----27 28// CHECK-LABEL: func @memref_index29// CHECK-SAME: %arg0: !llvm.ptr, %arg1: !llvm.ptr,30// CHECK-SAME: %arg2: i64, %arg3: i64, %arg4: i64)31// CHECK-SAME: -> !llvm.struct<(ptr, ptr, i64, array<1 x i64>, array<1 x i64>)>32// CHECK32-LABEL: func @memref_index33// CHECK32-SAME: %arg0: !llvm.ptr, %arg1: !llvm.ptr,34// CHECK32-SAME: %arg2: i32, %arg3: i32, %arg4: i32)35// CHECK32-SAME: -> !llvm.struct<(ptr, ptr, i32, array<1 x i32>, array<1 x i32>)>36func.func @memref_index(%arg0: memref<32xindex>) -> memref<32xindex> {37 return %arg0 : memref<32xindex>38}39 40// -----41 42// CHECK-LABEL: func @check_arguments43// CHECK-COUNT-2: !llvm.ptr44// CHECK-COUNT-5: i6445// CHECK-COUNT-2: !llvm.ptr46// CHECK-COUNT-5: i6447// CHECK-COUNT-2: !llvm.ptr48// CHECK-COUNT-5: i6449func.func @check_arguments(%static: memref<10x20xf32>, %dynamic : memref<?x?xf32>, %mixed : memref<10x?xf32>) {50 return51}52 53// -----54 55// Unranked memrefs are currently not supported in the bare-ptr calling56// convention. Check that the conversion to the LLVM-IR dialect doesn't happen57// in the presence of unranked memrefs when using such a calling convention.58 59// BAREPTR: func private @hoo(memref<*xi8>) -> memref<*xi8>60func.func private @hoo(memref<*xi8>) -> memref<*xi8>61 62// BAREPTR-LABEL: func @check_unranked_memref_func_call(%{{.*}}: memref<*xi8>) -> memref<*xi8>63func.func @check_unranked_memref_func_call(%in: memref<*xi8>) -> memref<*xi8> {64 // BAREPTR-NEXT: call @hoo(%{{.*}}) : (memref<*xi8>) -> memref<*xi8>65 %res = call @hoo(%in) : (memref<*xi8>) -> memref<*xi8>66 // BAREPTR-NEXT: return %{{.*}} : memref<*xi8>67 return %res : memref<*xi8>68}69 70// -----71 72// Should not convert memrefs with unsupported types in any convention.73 74// CHECK: @unsupported_memref_element_type75// CHECK-SAME: memref<76// CHECK-NOT: !llvm.struct77// BAREPTR: @unsupported_memref_element_type78// BAREPTR-SAME: memref<79// BAREPTR-NOT: !llvm.ptr80func.func private @unsupported_memref_element_type() -> memref<42 x !test.memref_element>81 82// CHECK: @unsupported_unranked_memref_element_type83// CHECK-SAME: memref<84// CHECK-NOT: !llvm.struct85// BAREPTR: @unsupported_unranked_memref_element_type86// BAREPTR-SAME: memref<87// BAREPTR-NOT: !llvm.ptr88func.func private @unsupported_unranked_memref_element_type() -> memref<* x !test.memref_element>89 90// -----91 92// BAREPTR: llvm.func @goo(f32) -> f3293func.func private @goo(f32) -> f3294 95// BAREPTR-LABEL: func @check_scalar_func_call96// BAREPTR-SAME: %[[in:.*]]: f32)97func.func @check_scalar_func_call(%in : f32) {98 // BAREPTR-NEXT: %[[call:.*]] = llvm.call @goo(%[[in]]) : (f32) -> f3299 %res = call @goo(%in) : (f32) -> (f32)100 return101}102 103// -----104 105!base_type = memref<64xi32, 201>106 107// CHECK-LABEL: func @loop_carried108// BAREPTR-LABEL: func @loop_carried109func.func @loop_carried(%arg0 : index, %arg1 : index, %arg2 : index, %base0 : !base_type, %base1 : !base_type) -> (!base_type, !base_type) {110 // This test checks that in the BAREPTR case, the branch arguments only forward the descriptor.111 // This test was lowered from a simple scf.for that swaps 2 memref iter_args.112 // BAREPTR: llvm.br ^bb1(%{{.*}}, %{{.*}}, %{{.*}} : i64, !llvm.struct<(ptr<201>, ptr<201>, i64, array<1 x i64>, array<1 x i64>)>, !llvm.struct<(ptr<201>, ptr<201>, i64, array<1 x i64>, array<1 x i64>)>)113 cf.br ^bb1(%arg0, %base0, %base1 : index, memref<64xi32, 201>, memref<64xi32, 201>)114 115 // BAREPTR-NEXT: ^bb1116 // BAREPTR-NEXT: llvm.icmp117 // BAREPTR-NEXT: llvm.cond_br %{{.*}}, ^bb2, ^bb3118 ^bb1(%0: index, %1: memref<64xi32, 201>, %2: memref<64xi32, 201>): // 2 preds: ^bb0, ^bb2119 %3 = arith.cmpi slt, %0, %arg1 : index120 cf.cond_br %3, ^bb2, ^bb3121 ^bb2: // pred: ^bb1122 %4 = arith.addi %0, %arg2 : index123 cf.br ^bb1(%4, %2, %1 : index, memref<64xi32, 201>, memref<64xi32, 201>)124 ^bb3: // pred: ^bb1125 return %1, %2 : memref<64xi32, 201>, memref<64xi32, 201>126}127