diff --git a/mlir/include/mlir/Dialect/StandardOps/IR/Ops.td b/mlir/include/mlir/Dialect/StandardOps/IR/Ops.td --- a/mlir/include/mlir/Dialect/StandardOps/IR/Ops.td +++ b/mlir/include/mlir/Dialect/StandardOps/IR/Ops.td @@ -2405,6 +2405,39 @@ let hasFolder = 1; } +//===----------------------------------------------------------------------===// +// PowFOp +//===----------------------------------------------------------------------===// + +def PowFOp : FloatArithmeticOp<"powf"> { + let summary = "floating point raised to the power of operation"; + let description = [{ + Syntax: + + ``` + operation ::= ssa-id `=` `std.powf` ssa-use `,` ssa-use `:` type + ``` + + The `powf` operation takes two operands and returns one result, each of + these is required to be the same type. This type may be a floating point + scalar type, a vector whose element type is a floating point type, or a + floating point tensor. + + Example: + + ```mlir + // Scalar exponentiation. + %a = powf %b, %c : f64 + + // SIMD pointwise vector exponentiation, e.g. for Intel SSE. + %f = powf %g, %h : vector<4xf32> + + // Tensor pointwise exponentiation. + %x = powf %y, %z : tensor<4x?xbf16> + ``` + }]; +} + //===----------------------------------------------------------------------===// // PrefetchOp //===----------------------------------------------------------------------===// diff --git a/mlir/test/IR/core-ops.mlir b/mlir/test/IR/core-ops.mlir --- a/mlir/test/IR/core-ops.mlir +++ b/mlir/test/IR/core-ops.mlir @@ -86,6 +86,9 @@ // CHECK: %[[I6:.*]] = muli %[[I2]], %[[I2]] : i32 %i6 = muli %i2, %i2 : i32 + // CHECK: %[[F7:.*]] = powf %[[F2]], %[[F2]] : f32 + %f7 = powf %f2, %f2 : f32 + // CHECK: %[[C0:.*]] = create_complex %[[F2]], %[[F2]] : complex %c0 = "std.create_complex"(%f2, %f2) : (f32, f32) -> complex