69 lines · plain
1// Test affine demotion pass2 3// RUN: fir-opt --split-input-file --demote-affine %s | FileCheck %s4 5#map0 = affine_map<()[s0, s1] -> (s1 - s0 + 1)>6#map1 = affine_map<()[s0] -> (s0 + 1)>7#map2 = affine_map<(d0)[s0, s1, s2] -> (d0 * s2 - s0)>8module {9 func.func @calc(%arg0: !fir.ref<!fir.array<?xf32>>, %arg1: !fir.ref<!fir.array<?xf32>>, %arg2: !fir.ref<!fir.array<?xf32>>) {10 %c1 = arith.constant 1 : index11 %c100 = arith.constant 100 : index12 %0 = fir.shape %c100 : (index) -> !fir.shape<1>13 %1 = affine.apply #map0()[%c1, %c100]14 %2 = fir.alloca !fir.array<?xf32>, %115 %3 = fir.convert %arg0 : (!fir.ref<!fir.array<?xf32>>) -> memref<?xf32>16 %4 = fir.convert %arg1 : (!fir.ref<!fir.array<?xf32>>) -> memref<?xf32>17 %5 = fir.convert %2 : (!fir.ref<!fir.array<?xf32>>) -> memref<?xf32>18 affine.for %arg3 = %c1 to #map1()[%c100] {19 %7 = affine.apply #map2(%arg3)[%c1, %c100, %c1]20 %8 = affine.load %3[%7] : memref<?xf32>21 %9 = affine.load %4[%7] : memref<?xf32>22 %10 = arith.addf %8, %9 : f3223 affine.store %10, %5[%7] : memref<?xf32>24 }25 %6 = fir.convert %arg2 : (!fir.ref<!fir.array<?xf32>>) -> memref<?xf32>26 affine.for %arg3 = %c1 to #map1()[%c100] {27 %7 = affine.apply #map2(%arg3)[%c1, %c100, %c1]28 %8 = affine.load %5[%7] : memref<?xf32>29 %9 = affine.load %4[%7] : memref<?xf32>30 %10 = arith.mulf %8, %9 : f3231 affine.store %10, %6[%7] : memref<?xf32>32 }33 return34 }35}36 37// CHECK: func @calc(%[[VAL_0:.*]]: !fir.ref<!fir.array<?xf32>>, %[[VAL_1:.*]]: !fir.ref<!fir.array<?xf32>>, %[[VAL_2:.*]]: !fir.ref<!fir.array<?xf32>>) {38// CHECK: %[[VAL_3:.*]] = arith.constant 1 : index39// CHECK: %[[VAL_4:.*]] = arith.constant 100 : index40// CHECK: %[[VAL_5:.*]] = fir.shape %[[VAL_4]] : (index) -> !fir.shape<1>41// CHECK: %[[VAL_6:.*]] = arith.constant 100 : index42// CHECK: %[[VAL_7:.*]] = fir.alloca !fir.array<?xf32>, %[[VAL_6]]43// CHECK: %[[VAL_8:.*]] = fir.convert %[[VAL_0]] : (!fir.ref<!fir.array<?xf32>>) -> !fir.ref<!fir.array<?xf32>>44// CHECK: %[[VAL_9:.*]] = fir.convert %[[VAL_1]] : (!fir.ref<!fir.array<?xf32>>) -> !fir.ref<!fir.array<?xf32>>45// CHECK: %[[VAL_10:.*]] = fir.convert %[[VAL_7]] : (!fir.ref<!fir.array<?xf32>>) -> !fir.ref<!fir.array<?xf32>>46// CHECK: affine.for %[[VAL_11:.*]] = 1 to 101 {47// CHECK: %[[VAL_12:.*]] = affine.apply #map(%[[VAL_11]]){{\[}}%[[VAL_3]], %[[VAL_4]], %[[VAL_3]]]48// CHECK: %[[VAL_13:.*]] = fir.coordinate_of %[[VAL_8]], %[[VAL_12]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>49// CHECK: %[[VAL_14:.*]] = fir.load %[[VAL_13]] : !fir.ref<f32>50// CHECK: %[[VAL_15:.*]] = fir.coordinate_of %[[VAL_9]], %[[VAL_12]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>51// CHECK: %[[VAL_16:.*]] = fir.load %[[VAL_15]] : !fir.ref<f32>52// CHECK: %[[VAL_17:.*]] = arith.addf %[[VAL_14]], %[[VAL_16]] : f3253// CHECK: %[[VAL_18:.*]] = fir.coordinate_of %[[VAL_10]], %[[VAL_12]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>54// CHECK: fir.store %[[VAL_17]] to %[[VAL_18]] : !fir.ref<f32>55// CHECK: }56// CHECK: %[[VAL_19:.*]] = fir.convert %[[VAL_2]] : (!fir.ref<!fir.array<?xf32>>) -> !fir.ref<!fir.array<?xf32>>57// CHECK: affine.for %[[VAL_20:.*]] = 1 to 101 {58// CHECK: %[[VAL_21:.*]] = affine.apply #map(%[[VAL_20]]){{\[}}%[[VAL_3]], %[[VAL_4]], %[[VAL_3]]]59// CHECK: %[[VAL_22:.*]] = fir.coordinate_of %[[VAL_10]], %[[VAL_21]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>60// CHECK: %[[VAL_23:.*]] = fir.load %[[VAL_22]] : !fir.ref<f32>61// CHECK: %[[VAL_24:.*]] = fir.coordinate_of %[[VAL_9]], %[[VAL_21]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>62// CHECK: %[[VAL_25:.*]] = fir.load %[[VAL_24]] : !fir.ref<f32>63// CHECK: %[[VAL_26:.*]] = arith.mulf %[[VAL_23]], %[[VAL_25]] : f3264// CHECK: %[[VAL_27:.*]] = fir.coordinate_of %[[VAL_19]], %[[VAL_21]] : (!fir.ref<!fir.array<?xf32>>, index) -> !fir.ref<f32>65// CHECK: fir.store %[[VAL_26]] to %[[VAL_27]] : !fir.ref<f32>66// CHECK: }67// CHECK: return68// CHECK: }69