[mlir] Add `complex.atan2` operation.

Differential Revision: https://reviews.llvm.org/D126357

mlir/include/mlir/Dialect/Complex/IR/ComplexOps.td

@@ -75,6 +75,24 @@ def AddOp : ComplexArithmeticOp<"add"> {
   }];
 }
 
+//===----------------------------------------------------------------------===//
+// Atan2
+//===----------------------------------------------------------------------===//
+
+def Atan2Op : ComplexArithmeticOp<"atan2"> {
+  let summary = "complex 2-argument arctangent";
+  let description = [{
+    For complex numbers it is expressed using complex logarithm
+    atan2(y, x) = -i * log((x + i * y) / sqrt(x**2 + y**2))
+
+    Example:
+
+    ```mlir
+    %a = complex.atan2 %b, %c : complex<f32>
+    ```
+  }];
+}
+
 //===----------------------------------------------------------------------===//
 // ConstantOp
 //===----------------------------------------------------------------------===//

mlir/test/Dialect/Complex/ops.mlir

@@ -77,5 +77,8 @@ func.func @ops(%f: f32) {
   // CHECK: complex.rsqrt %[[C]] : complex<f32>
   %rsqrt = complex.rsqrt %complex : complex<f32>
 
+  // CHECK: complex.atan2 %[[C]], %[[C]] : complex<f32>
+  %atan2 = complex.atan2 %complex, %complex : complex<f32>
+
   return
 }