[Mlir-commits] [mlir] 331569d - [mlir][SPIR-V] Convert math.cbrt to OpenCL cbrt (#201915)

[llvmlistbot at llvm.org]

Author: Arseniy Obolenskiy
Date: 2026-06-08T10:40:53+02:00
New Revision: 331569d4179012955a0922516cebf314af677913

URL: https://github.com/llvm/llvm-project/commit/331569d4179012955a0922516cebf314af677913
DIFF: https://github.com/llvm/llvm-project/commit/331569d4179012955a0922516cebf314af677913.diff

LOG: [mlir][SPIR-V] Convert math.cbrt to OpenCL cbrt (#201915)

Added: 
    

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

Removed: 
    


################################################################################
diff  --git a/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp b/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
index 5353fb64fb443..da9ce2b747fb4 100644
--- a/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
+++ b/mlir/lib/Conversion/MathToSPIRV/MathToSPIRV.cpp
@@ -652,6 +652,7 @@ void populateMathToSPIRVPatterns(const SPIRVTypeConverter &typeConverter,
       CheckedElementwiseOpPattern<math::CountLeadingZerosOp, spirv::CLClzOp>,
       CheckedElementwiseOpPattern<math::AtanOp, spirv::CLAtanOp>,
       CheckedElementwiseOpPattern<math::Atan2Op, spirv::CLAtan2Op>,
+      CheckedElementwiseOpPattern<math::CbrtOp, spirv::CLCbrtOp>,
       CheckedElementwiseOpPattern<math::CeilOp, spirv::CLCeilOp>,
       CheckedElementwiseOpPattern<math::CosOp, spirv::CLCosOp>,
       CheckedElementwiseOpPattern<math::ErfOp, spirv::CLErfOp>,

diff  --git a/mlir/test/Conversion/MathToSPIRV/math-to-opencl-spirv.mlir b/mlir/test/Conversion/MathToSPIRV/math-to-opencl-spirv.mlir
index fd040f1646661..d972d4f6dba9e 100644
--- a/mlir/test/Conversion/MathToSPIRV/math-to-opencl-spirv.mlir
+++ b/mlir/test/Conversion/MathToSPIRV/math-to-opencl-spirv.mlir
@@ -64,6 +64,8 @@ func.func @float32_unary_scalar(%arg0: f32) {
   %25 = math.atanh %arg0 : f32
   // CHECK: spirv.CL.trunc %{{.*}}: f32
   %26 = math.trunc %arg0 : f32
+  // CHECK: spirv.CL.cbrt %{{.*}}: f32
+  %27 = math.cbrt %arg0 : f32
   return
 }