[Mlir-commits] [mlir] [mlir][SPIR-V] Add SPIRVToLLVM conversions for FMod and SMod (PR #206933)
Arseniy Obolenskiy
llvmlistbot at llvm.org
Mon Jul 6 02:53:32 PDT 2026
https://github.com/aobolensk updated https://github.com/llvm/llvm-project/pull/206933
>From 2244a3cbcb0294fc1e9adb865ef974b6ff802a26 Mon Sep 17 00:00:00 2001
From: Arseniy Obolenskiy <arseniy.obolenskiy at amd.com>
Date: Wed, 1 Jul 2026 12:03:16 +0200
Subject: [PATCH] [mlir][SPIR-V] Add SPIRVToLLVM conversions for FMod and SMod
---
.../Conversion/SPIRVToLLVM/SPIRVToLLVM.cpp | 80 ++++++++++++++++++-
.../SPIRVToLLVM/arithmetic-ops-to-llvm.mlir | 54 +++++++++++++
2 files changed, 133 insertions(+), 1 deletion(-)
diff --git a/mlir/lib/Conversion/SPIRVToLLVM/SPIRVToLLVM.cpp b/mlir/lib/Conversion/SPIRVToLLVM/SPIRVToLLVM.cpp
index c43415b27b1b3..5be4cc6bc4ebd 100644
--- a/mlir/lib/Conversion/SPIRVToLLVM/SPIRVToLLVM.cpp
+++ b/mlir/lib/Conversion/SPIRVToLLVM/SPIRVToLLVM.cpp
@@ -921,6 +921,83 @@ class InverseSqrtPattern
}
};
+/// Converts `spirv.FMod` to `x - y * floor(x / y)`. The SPIR-V op requires the
+/// result to take the sign of the divisor, whereas `llvm.frem` keeps the sign
+/// of the dividend, so `frem` cannot be used directly.
+class FModPattern : public SPIRVToLLVMConversion<spirv::FModOp> {
+public:
+ using SPIRVToLLVMConversion<spirv::FModOp>::SPIRVToLLVMConversion;
+
+ LogicalResult
+ matchAndRewrite(spirv::FModOp op, OpAdaptor adaptor,
+ ConversionPatternRewriter &rewriter) const override {
+ auto dstType = getTypeConverter()->convertType(op.getType());
+ if (!dstType)
+ return rewriter.notifyMatchFailure(op, "type conversion failed");
+
+ Location loc = op.getLoc();
+ Value lhs = adaptor.getOperand1();
+ Value rhs = adaptor.getOperand2();
+ Value div = LLVM::FDivOp::create(rewriter, loc, dstType, lhs, rhs);
+ Value floored = LLVM::FFloorOp::create(rewriter, loc, dstType, div);
+ Value scaled = LLVM::FMulOp::create(rewriter, loc, dstType, rhs, floored);
+ rewriter.replaceOpWithNewOp<LLVM::FSubOp>(op, dstType, lhs, scaled);
+ return success();
+ }
+};
+
+/// Converts `spirv.SMod` to a signed remainder corrected to take the sign of
+/// the divisor. `llvm.srem` keeps the sign of the dividend, so the result is
+/// adjusted by adding the divisor when the remainder is non-zero and its sign
+/// differs from the divisor's.
+class SModPattern : public SPIRVToLLVMConversion<spirv::SModOp> {
+public:
+ using SPIRVToLLVMConversion<spirv::SModOp>::SPIRVToLLVMConversion;
+
+ LogicalResult
+ matchAndRewrite(spirv::SModOp op, OpAdaptor adaptor,
+ ConversionPatternRewriter &rewriter) const override {
+ auto srcType = op.getType();
+ auto dstType = getTypeConverter()->convertType(srcType);
+ if (!dstType)
+ return rewriter.notifyMatchFailure(op, "type conversion failed");
+
+ Location loc = op.getLoc();
+ Value lhs = adaptor.getOperand1();
+ Value rhs = adaptor.getOperand2();
+ Type i1Type = rewriter.getI1Type();
+ auto vecSrcType = dyn_cast<VectorType>(srcType);
+ Type cmpType =
+ vecSrcType ? VectorType::get(vecSrcType.getShape(), i1Type) : i1Type;
+
+ Value rem = LLVM::SRemOp::create(rewriter, loc, dstType, lhs, rhs);
+ IntegerAttr zeroAttr = rewriter.getIntegerAttr(
+ cast<IntegerType>(getElementTypeOrSelf(srcType)), 0);
+ Value zero;
+ if (vecSrcType)
+ zero = LLVM::ConstantOp::create(
+ rewriter, loc, dstType, SplatElementsAttr::get(vecSrcType, zeroAttr));
+ else
+ zero = LLVM::ConstantOp::create(rewriter, loc, dstType, zeroAttr);
+
+ Value remNonZero = LLVM::ICmpOp::create(rewriter, loc, cmpType,
+ LLVM::ICmpPredicate::ne, rem, zero);
+ Value remNeg = LLVM::ICmpOp::create(rewriter, loc, cmpType,
+ LLVM::ICmpPredicate::slt, rem, zero);
+ Value rhsNeg = LLVM::ICmpOp::create(rewriter, loc, cmpType,
+ LLVM::ICmpPredicate::slt, rhs, zero);
+ Value signMismatch =
+ LLVM::XOrOp::create(rewriter, loc, cmpType, remNeg, rhsNeg);
+ Value needsAdjust =
+ LLVM::AndOp::create(rewriter, loc, cmpType, remNonZero, signMismatch);
+
+ Value adjusted = LLVM::AddOp::create(rewriter, loc, dstType, rem, rhs);
+ rewriter.replaceOpWithNewOp<LLVM::SelectOp>(op, dstType, needsAdjust,
+ adjusted, rem);
+ return success();
+ }
+};
+
/// Converts `spirv.Load` and `spirv.Store` to LLVM dialect.
template <typename SPIRVOp>
class LoadStorePattern : public SPIRVToLLVMConversion<SPIRVOp> {
@@ -1824,7 +1901,8 @@ void mlir::populateSPIRVToLLVMConversionPatterns(
DirectConversionPattern<spirv::SDivOp, LLVM::SDivOp>,
DirectConversionPattern<spirv::SRemOp, LLVM::SRemOp>,
DirectConversionPattern<spirv::UDivOp, LLVM::UDivOp>,
- DirectConversionPattern<spirv::UModOp, LLVM::URemOp>,
+ DirectConversionPattern<spirv::UModOp, LLVM::URemOp>, FModPattern,
+ SModPattern,
// Bitwise ops
BitFieldInsertPattern, BitFieldUExtractPattern, BitFieldSExtractPattern,
diff --git a/mlir/test/Conversion/SPIRVToLLVM/arithmetic-ops-to-llvm.mlir b/mlir/test/Conversion/SPIRVToLLVM/arithmetic-ops-to-llvm.mlir
index dbbf8610afb4d..36d4c5facb057 100644
--- a/mlir/test/Conversion/SPIRVToLLVM/arithmetic-ops-to-llvm.mlir
+++ b/mlir/test/Conversion/SPIRVToLLVM/arithmetic-ops-to-llvm.mlir
@@ -233,3 +233,57 @@ spirv.func @srem_vector(%arg0: vector<4xi32>, %arg1: vector<4xi32>) "None" {
%0 = spirv.SRem %arg0, %arg1 : vector<4xi32>
spirv.Return
}
+
+//===----------------------------------------------------------------------===//
+// spirv.FMod
+//===----------------------------------------------------------------------===//
+
+// CHECK-LABEL: @fmod_scalar
+spirv.func @fmod_scalar(%arg0: f32, %arg1: f32) "None" {
+ // CHECK: %[[DIV:.*]] = llvm.fdiv %{{.*}}, %{{.*}} : f32
+ // CHECK: %[[FLOOR:.*]] = llvm.intr.floor(%[[DIV]]) : (f32) -> f32
+ // CHECK: %[[MUL:.*]] = llvm.fmul %{{.*}}, %[[FLOOR]] : f32
+ // CHECK: llvm.fsub %{{.*}}, %[[MUL]] : f32
+ %0 = spirv.FMod %arg0, %arg1 : f32
+ spirv.Return
+}
+
+// CHECK-LABEL: @fmod_vector
+spirv.func @fmod_vector(%arg0: vector<4xf32>, %arg1: vector<4xf32>) "None" {
+ // CHECK: %[[DIV:.*]] = llvm.fdiv %{{.*}}, %{{.*}} : vector<4xf32>
+ // CHECK: %[[FLOOR:.*]] = llvm.intr.floor(%[[DIV]]) : (vector<4xf32>) -> vector<4xf32>
+ // CHECK: %[[MUL:.*]] = llvm.fmul %{{.*}}, %[[FLOOR]] : vector<4xf32>
+ // CHECK: llvm.fsub %{{.*}}, %[[MUL]] : vector<4xf32>
+ %0 = spirv.FMod %arg0, %arg1 : vector<4xf32>
+ spirv.Return
+}
+
+//===----------------------------------------------------------------------===//
+// spirv.SMod
+//===----------------------------------------------------------------------===//
+
+// CHECK-LABEL: @smod_scalar
+spirv.func @smod_scalar(%arg0: i32, %arg1: i32) "None" {
+ // CHECK: %[[REM:.*]] = llvm.srem %{{.*}}, %{{.*}} : i32
+ // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(0 : i32) : i32
+ // CHECK: %[[NZ:.*]] = llvm.icmp "ne" %[[REM]], %[[ZERO]] : i32
+ // CHECK: %[[RNEG:.*]] = llvm.icmp "slt" %[[REM]], %[[ZERO]] : i32
+ // CHECK: %[[DNEG:.*]] = llvm.icmp "slt" %{{.*}}, %[[ZERO]] : i32
+ // CHECK: %[[XOR:.*]] = llvm.xor %[[RNEG]], %[[DNEG]] : i1
+ // CHECK: %[[ADJ:.*]] = llvm.and %[[NZ]], %[[XOR]] : i1
+ // CHECK: %[[ADD:.*]] = llvm.add %[[REM]], %{{.*}} : i32
+ // CHECK: llvm.select %[[ADJ]], %[[ADD]], %[[REM]] : i1, i32
+ %0 = spirv.SMod %arg0, %arg1 : i32
+ spirv.Return
+}
+
+// CHECK-LABEL: @smod_vector
+spirv.func @smod_vector(%arg0: vector<4xi32>, %arg1: vector<4xi32>) "None" {
+ // CHECK: %[[REM:.*]] = llvm.srem %{{.*}}, %{{.*}} : vector<4xi32>
+ // CHECK: %[[ZERO:.*]] = llvm.mlir.constant(dense<0> : vector<4xi32>) : vector<4xi32>
+ // CHECK: %[[NZ:.*]] = llvm.icmp "ne" %[[REM]], %[[ZERO]] : vector<4xi32>
+ // CHECK: %[[ADD:.*]] = llvm.add %[[REM]], %{{.*}} : vector<4xi32>
+ // CHECK: llvm.select %{{.*}}, %[[ADD]], %[[REM]] : vector<4xi1>, vector<4xi32>
+ %0 = spirv.SMod %arg0, %arg1 : vector<4xi32>
+ spirv.Return
+}
More information about the Mlir-commits
mailing list