[mlir][StandardToSPIRV] Add support for lowering unary ops
Differential Revision: https://reviews.llvm.org/D76661
diff --git a/mlir/test/Conversion/StandardToSPIRV/std-to-spirv.mlir b/mlir/test/Conversion/StandardToSPIRV/std-to-spirv.mlir
index cb5873a..91219ac 100644
--- a/mlir/test/Conversion/StandardToSPIRV/std-to-spirv.mlir
+++ b/mlir/test/Conversion/StandardToSPIRV/std-to-spirv.mlir
@@ -31,9 +31,33 @@
return
}
-// Check float operation conversions.
-// CHECK-LABEL: @float32_scalar
-func @float32_scalar(%lhs: f32, %rhs: f32) {
+// Check float unary operation conversions.
+// CHECK-LABEL: @float32_unary_scalar
+func @float32_unary_scalar(%arg0: f32) {
+ // CHECK: spv.GLSL.FAbs %{{.*}}: f32
+ %0 = absf %arg0 : f32
+ // CHECK: spv.GLSL.Ceil %{{.*}}: f32
+ %1 = ceilf %arg0 : f32
+ // CHECK: spv.GLSL.Cos %{{.*}}: f32
+ %2 = cos %arg0 : f32
+ // CHECK: spv.GLSL.Exp %{{.*}}: f32
+ %3 = exp %arg0 : f32
+ // CHECK: spv.GLSL.Log %{{.*}}: f32
+ %4 = log %arg0 : f32
+ // CHECK: spv.FNegate %{{.*}}: f32
+ %5 = negf %arg0 : f32
+ // CHECK: spv.GLSL.InverseSqrt %{{.*}}: f32
+ %6 = rsqrt %arg0 : f32
+ // CHECK: spv.GLSL.Sqrt %{{.*}}: f32
+ %7 = sqrt %arg0 : f32
+ // CHECK: spv.GLSL.Tanh %{{.*}}: f32
+ %8 = tanh %arg0 : f32
+ return
+}
+
+// Check float binary operation conversions.
+// CHECK-LABEL: @float32_binary_scalar
+func @float32_binary_scalar(%lhs: f32, %rhs: f32) {
// CHECK: spv.FAdd %{{.*}}, %{{.*}}: f32
%0 = addf %lhs, %rhs: f32
// CHECK: spv.FSub %{{.*}}, %{{.*}}: f32
