110 lines · plain
1// RUN: fir-opt --convert-complex-pow %s | FileCheck %s2 3module {4 func.func @pow_c4_i4(%arg0: complex<f32>, %arg1: i32) -> complex<f32> {5 %0 = complex.powi %arg0, %arg1 : complex<f32>, i326 return %0 : complex<f32>7 }8 9 func.func @pow_c4_i4_fast(%arg0: complex<f32>, %arg1: i32) -> complex<f32> {10 %0 = complex.powi %arg0, %arg1 fastmath<fast> : complex<f32>, i3211 return %0 : complex<f32>12 }13 14 func.func @pow_c4_i8(%arg0: complex<f32>, %arg1: i64) -> complex<f32> {15 %0 = complex.powi %arg0, %arg1 : complex<f32>, i6416 return %0 : complex<f32>17 }18 19 func.func @pow_c8_i4(%arg0: complex<f64>, %arg1: i32) -> complex<f64> {20 %0 = complex.powi %arg0, %arg1 : complex<f64>, i3221 return %0 : complex<f64>22 }23 24 func.func @pow_c8_i8(%arg0: complex<f64>, %arg1: i64) -> complex<f64> {25 %0 = complex.powi %arg0, %arg1 : complex<f64>, i6426 return %0 : complex<f64>27 }28 29 func.func @pow_c16_i4(%arg0: complex<f128>, %arg1: i32) -> complex<f128> {30 %0 = complex.powi %arg0, %arg1 : complex<f128>, i3231 return %0 : complex<f128>32 }33 34 func.func @pow_c16_i8(%arg0: complex<f128>, %arg1: i64) -> complex<f128> {35 %0 = complex.powi %arg0, %arg1 : complex<f128>, i6436 return %0 : complex<f128>37 }38 39 func.func @pow_c4_fast(%arg0: complex<f32>, %arg1: f32) -> complex<f32> {40 %c1 = arith.constant 1.0 : f3241 %0 = complex.create %arg1, %c1 : complex<f32>42 %1 = complex.pow %arg0, %0 fastmath<fast> : complex<f32>43 return %1 : complex<f32>44 }45 46 func.func @pow_c8_complex(%arg0: complex<f64>, %arg1: f64) -> complex<f64> {47 %c2 = arith.constant 2.0 : f6448 %0 = complex.create %arg1, %c2 : complex<f64>49 %1 = complex.pow %arg0, %0 : complex<f64>50 return %1 : complex<f64>51 }52 53 func.func @pow_c16_complex(%arg0: complex<f128>, %arg1: f128) -> complex<f128> {54 %c3 = arith.constant 3.0 : f12855 %0 = complex.create %arg1, %c3 : complex<f128>56 %1 = complex.pow %arg0, %0 : complex<f128>57 return %1 : complex<f128>58 }59}60 61// CHECK-LABEL: func.func @pow_c4_i4(62// CHECK: fir.call @_FortranAcpowi(%{{.*}}, %{{.*}}) : (complex<f32>, i32) -> complex<f32>63// CHECK-NOT: complex.pow64// CHECK-NOT: complex.powi65 66// CHECK-LABEL: func.func @pow_c4_i4_fast(67// CHECK: fir.call @_FortranAcpowi(%{{.*}}, %{{.*}}) fastmath<fast> : (complex<f32>, i32) -> complex<f32>68// CHECK-NOT: complex.pow69// CHECK-NOT: complex.powi70 71// CHECK-LABEL: func.func @pow_c4_i8(72// CHECK: fir.call @_FortranAcpowk(%{{.*}}, %{{.*}}) : (complex<f32>, i64) -> complex<f32>73// CHECK-NOT: complex.pow74// CHECK-NOT: complex.powi75 76// CHECK-LABEL: func.func @pow_c8_i4(77// CHECK: fir.call @_FortranAzpowi(%{{.*}}, %{{.*}}) : (complex<f64>, i32) -> complex<f64>78// CHECK-NOT: complex.pow79// CHECK-NOT: complex.powi80 81// CHECK-LABEL: func.func @pow_c8_i8(82// CHECK: fir.call @_FortranAzpowk(%{{.*}}, %{{.*}}) : (complex<f64>, i64) -> complex<f64>83// CHECK-NOT: complex.pow84// CHECK-NOT: complex.powi85 86// CHECK-LABEL: func.func @pow_c16_i4(87// CHECK: fir.call @_FortranAcqpowi(%{{.*}}, %{{.*}}) : (complex<f128>, i32) -> complex<f128>88// CHECK-NOT: complex.pow89// CHECK-NOT: complex.powi90 91// CHECK-LABEL: func.func @pow_c16_i8(92// CHECK: fir.call @_FortranAcqpowk(%{{.*}}, %{{.*}}) : (complex<f128>, i64) -> complex<f128>93// CHECK-NOT: complex.pow94// CHECK-NOT: complex.powi95 96// CHECK-LABEL: func.func @pow_c4_fast(97// CHECK: %[[EXP:.*]] = complex.create %{{.*}}, %{{.*}} : complex<f32>98// CHECK: fir.call @cpowf(%{{.*}}, %[[EXP]]) fastmath<fast> : (complex<f32>, complex<f32>) -> complex<f32>99// CHECK-NOT: complex.pow100 101// CHECK-LABEL: func.func @pow_c8_complex(102// CHECK: %[[EXP:.*]] = complex.create %{{.*}}, %{{.*}} : complex<f64>103// CHECK: fir.call @cpow(%{{.*}}, %[[EXP]]) : (complex<f64>, complex<f64>) -> complex<f64>104// CHECK-NOT: complex.pow105 106// CHECK-LABEL: func.func @pow_c16_complex(107// CHECK: %[[EXP:.*]] = complex.create %{{.*}}, %{{.*}} : complex<f128>108// CHECK: fir.call @_FortranACPowF128(%{{.*}}, %[[EXP]]) : (complex<f128>, complex<f128>) -> complex<f128>109// CHECK-NOT: complex.pow110