58 lines · plain
1// RUN: mlir-opt %s -pass-pipeline='builtin.module(func.func(scf-forall-to-for))' -split-input-file | FileCheck %s2 3func.func private @callee(%i: index, %j: index)4 5// CHECK-LABEL: @two_iters6// CHECK-SAME: %[[UB1:.+]]: index, %[[UB2:.+]]: index7func.func @two_iters(%ub1: index, %ub2: index) {8 scf.forall (%i, %j) in (%ub1, %ub2) {9 func.call @callee(%i, %j) : (index, index) -> ()10 }11 // CHECK: scf.for %[[IV1:.+]] = %{{.*}} to %[[UB1]]12 // CHECK: scf.for %[[IV2:.+]] = %{{.*}} to %[[UB2]]13 // CHECK: func.call @callee(%[[IV1]], %[[IV2]])14 return15}16 17// -----18 19func.func private @callee(%i: index, %j: index)20 21// CHECK-LABEL: @repeated22// CHECK-SAME: %[[UB1:.+]]: index, %[[UB2:.+]]: index23func.func @repeated(%ub1: index, %ub2: index) {24 scf.forall (%i, %j) in (%ub1, %ub2) {25 func.call @callee(%i, %j) : (index, index) -> ()26 }27 // CHECK: scf.for %[[IV1:.+]] = %{{.*}} to %[[UB1]]28 // CHECK: scf.for %[[IV2:.+]] = %{{.*}} to %[[UB2]]29 // CHECK: func.call @callee(%[[IV1]], %[[IV2]])30 scf.forall (%i, %j) in (%ub1, %ub2) {31 func.call @callee(%i, %j) : (index, index) -> ()32 }33 // CHECK: scf.for %[[IV1:.+]] = %{{.*}} to %[[UB1]]34 // CHECK: scf.for %[[IV2:.+]] = %{{.*}} to %[[UB2]]35 // CHECK: func.call @callee(%[[IV1]], %[[IV2]])36 return37}38 39// -----40 41func.func private @callee(%i: index, %j: index, %k: index, %l: index)42 43// CHECK-LABEL: @nested44// CHECK-SAME: %[[UB1:.+]]: index, %[[UB2:.+]]: index, %[[UB3:.+]]: index, %[[UB4:.+]]: index45func.func @nested(%ub1: index, %ub2: index, %ub3: index, %ub4: index) {46 // CHECK: scf.for %[[IV1:.+]] = %{{.*}} to %[[UB1]]47 // CHECK: scf.for %[[IV2:.+]] = %{{.*}} to %[[UB2]]48 // CHECK: scf.for %[[IV3:.+]] = %{{.*}} to %[[UB3]]49 // CHECK: scf.for %[[IV4:.+]] = %{{.*}} to %[[UB4]]50 // CHECK: func.call @callee(%[[IV1]], %[[IV2]], %[[IV3]], %[[IV4]])51 scf.forall (%i, %j) in (%ub1, %ub2) {52 scf.forall (%k, %l) in (%ub3, %ub4) {53 func.call @callee(%i, %j, %k, %l) : (index, index, index, index) -> ()54 }55 }56 return57}58