// RUN: mlir-opt %s -test-math-algebraic-simplification | FileCheck %s

func.func @pow3(%arg0: complex<f32>) -> complex<f32> {
  %c3 = arith.constant 3 : i32
  %0 = complex.powi %arg0, %c3 : complex<f32>, i32
  return %0 : complex<f32>
}
// CHECK-LABEL: func.func @pow3(
// CHECK-NOT: complex.powi
// CHECK: %[[M0:.+]] = complex.mul %{{.*}}, %{{.*}} : complex<f32>
// CHECK: %[[M1:.+]] = complex.mul %[[M0]], %{{.*}} : complex<f32>
// CHECK: return %[[M1]] : complex<f32>

func.func @pow9(%arg0: complex<f32>) -> complex<f32> {
  %c9 = arith.constant 9 : i32
  %0 = complex.powi %arg0, %c9 : complex<f32>, i32
  return %0 : complex<f32>
}
// CHECK-LABEL: func.func @pow9(
// CHECK: complex.powi %{{.*}}, %{{.*}} : complex<f32>, i32