Added pow intrinsic to LLVMIR dialect

[georgemitenkov]

Summary

Added pow intrinsic to LLVMIR dialect. Added a roundrip test for it.

[ftynse]

Could we also test the conversion to actual LLVM IR?

mlir/test/Dialect/LLVMIR/roundtrip.mlir

104

Please don't pattern-match SSA names, https://mlir.llvm.org/getting_started/TestingGuide/#filecheck-best-practices

106

Could you just use %arg1 as both arguments? Constants can be moved around or simplified, so let's not rely on the constant operation being immediately before the one ew test.

[georgemitenkov]

Added a test for conversion to LLVM proper

Added a pow intrinsic test to Target/llvmir-intrinsic.mlir. Changed pow call in roundtrip test to take %arg1 as arguments.

[antiagainst]

Nice, thanks George! I'll leave to @ftynse to finally approve and land this. :)

[georgemitenkov]

Added conversion test for pow LLVM intrinsic

Added missing commits.

mlir/include/mlir/Dialect/LLVMIR/LLVMOps.td

 def LLVM_Log10Op : LLVM_UnaryIntrinsicOp<"log10">;
 def LLVM_Log2Op : LLVM_UnaryIntrinsicOp<"log2">;
 def LLVM_LogOp : LLVM_UnaryIntrinsicOp<"log">;
 def LLVM_Prefetch : LLVM_ZeroResultIntrOp<"prefetch", [0]>,
                     Arguments<(ins LLVM_Type:$addr, LLVM_Type:$rw,
                                LLVM_Type:$hint, LLVM_Type:$cache)>;
 def LLVM_SinOp : LLVM_UnaryIntrinsicOp<"sin">;
 def LLVM_SqrtOp : LLVM_UnaryIntrinsicOp<"sqrt">;
+def LLVM_PowOp : LLVM_BinarySameArgsIntrinsicOp<"pow">;
 
 //
 // Vector Reductions.
 //
 
 def LLVM_experimental_vector_reduce_add : LLVM_VectorReduction<"add">;
 def LLVM_experimental_vector_reduce_and : LLVM_VectorReduction<"and">;
 def LLVM_experimental_vector_reduce_mul : LLVM_VectorReduction<"mul">;

mlir/test/Dialect/LLVMIR/roundtrip.mlir

// CHECK-NEXT:  %28 = llvm.fpext %arg1 : !llvm.float to !llvm.fp128
   %28 = llvm.fpext %arg1 : !llvm.float to !llvm.fp128
 
 // CHECK:  %29 = llvm.fneg %arg1 : !llvm.float
   %29 = llvm.fneg %arg1 : !llvm.float
 
 // CHECK: "llvm.intr.sin"(%arg1) : (!llvm.float) -> !llvm.float
   %30 = "llvm.intr.sin"(%arg1) : (!llvm.float) -> !llvm.float
 
+// CHECK:       %31 = llvm.mlir.constant(4.200000e+00 : f32) : !llvm.float
+// CHECK-NEXT:  %32 = "llvm.intr.pow"(%arg1, %31) : (!llvm.float, !llvm.float) -> !llvm.float
+  %31 = llvm.mlir.constant(4.200000e+00 : f32) : !llvm.float
+  %32 = "llvm.intr.pow"(%arg1, %31) : (!llvm.float, !llvm.float) -> !llvm.float
+
 // CHECK:  llvm.return
   llvm.return
 }
 
 // An larger self-contained function.
 // CHECK-LABEL:func @foo(%arg0: !llvm.i32) -> !llvm<"{ i32, double, i32 }"> {
 func @foo(%arg0: !llvm.i32) -> !llvm<"{ i32, double, i32 }"> {
 // CHECK-NEXT:  %0 = llvm.mlir.constant(3 : i64) : !llvm.i32
func @useFenceInst() {
   // CHECK:  syncscope("agent") seq_cst 
   llvm.fence syncscope("agent") seq_cst
   // CHECK:  seq_cst 
   llvm.fence syncscope("") seq_cst
   // CHECK:  release
   llvm.fence release
   return
 }
 No newline at end of file