[Mlir-commits] [mlir] 47caa5d - [mlir][SPIR-V] Convert math.atan2 to GL atan2 (#201928)

[llvmlistbot at llvm.org]

Author: Arseniy Obolenskiy
Date: 2026-06-25T13:55:27+02:00
New Revision: 47caa5d8c66b84e39d1c15abf96cbbe3e3641ca1

URL: https://github.com/llvm/llvm-project/commit/47caa5d8c66b84e39d1c15abf96cbbe3e3641ca1
DIFF: https://github.com/llvm/llvm-project/commit/47caa5d8c66b84e39d1c15abf96cbbe3e3641ca1.diff

LOG: [mlir][SPIR-V] Convert math.atan2 to GL atan2 (#201928)

Added: 
    

Modified: 
    mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
    mlir/test/Conversion/MathToSPIRV/math-to-gl-spirv.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp b/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
index 82d87ede1e4be..61dddaf5155d3 100644
--- a/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
+++ b/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
@@ -647,6 +647,7 @@ void populateMathToSPIRVPatterns(const SPIRVTypeConverter &typeConverter,
            CheckedElementwiseOpPattern<math::AbsFOp, spirv::GLFAbsOp>,
            CheckedElementwiseOpPattern<math::AbsIOp, spirv::GLSAbsOp>,
            CheckedElementwiseOpPattern<math::AtanOp, spirv::GLAtanOp>,
+           CheckedElementwiseOpPattern<math::Atan2Op, spirv::GLAtan2Op>,
            CheckedElementwiseOpPattern<math::CeilOp, spirv::GLCeilOp>,
            CheckedElementwiseOpPattern<math::ClampFOp, spirv::GLFClampOp>,
            CheckedElementwiseOpPattern<math::CosOp, spirv::GLCosOp>,

diff  --git a/mlir/test/Conversion/MathToSPIRV/math-to-gl-spirv.mlir b/mlir/test/Conversion/MathToSPIRV/math-to-gl-spirv.mlir
index a8b2975ecb3c7..45e7bb01edc0b 100644
--- a/mlir/test/Conversion/MathToSPIRV/math-to-gl-spirv.mlir
+++ b/mlir/test/Conversion/MathToSPIRV/math-to-gl-spirv.mlir
@@ -139,6 +139,20 @@ func.func @float32_ternary_vector(%a: vector<4xf32>, %b: vector<4xf32>,
   return
 }
 
+// CHECK-LABEL: @float32_binary_scalar
+func.func @float32_binary_scalar(%lhs: f32, %rhs: f32) {
+  // CHECK: spirv.GL.Atan2 %{{.*}}, %{{.*}} : f32
+  %0 = math.atan2 %lhs, %rhs : f32
+  return
+}
+
+// CHECK-LABEL: @float32_binary_vector
+func.func @float32_binary_vector(%lhs: vector<3xf32>, %rhs: vector<3xf32>) {
+  // CHECK: spirv.GL.Atan2 %{{.*}}, %{{.*}} : vector<3xf32>
+  %0 = math.atan2 %lhs, %rhs : vector<3xf32>
+  return
+}
+
 // CHECK-LABEL: @float32_clamp_scalar
 func.func @float32_clamp_scalar(%value: f32, %min: f32, %max: f32) {
   // CHECK: spirv.GL.FClamp %{{.*}}, %{{.*}}, %{{.*}} : f32