[RISCV][GlobalISel] Legalize floating-point remainder

5 files changed, 21 insertions, 6 deletions

diff --git a/backend/include/rcc/CodeGen/CommandFlags.h b/backend/include/rcc/CodeGen/CommandFlags.h
index 0ee2f0d..aa7ed0d 100644
--- a/backend/include/rcc/CodeGen/CommandFlags.h
+++ b/backend/include/rcc/CodeGen/CommandFlags.h
@@ -41,6 +41,12 @@ enum class CodeGenFileType : std::uint8_t {
   Null,
 };
 
+enum class GlobalISelAbortMode : std::uint8_t {
+  Disable,
+  Enable,
+  DisableWithDiagnostic,
+};
+
 struct CodeGenCommandFlags final {
   std::optional<std::string> Triple;
   std::optional<std::string> CPU;
@@ -63,6 +69,7 @@ struct CodeGenCommandFlags final {
 [[nodiscard]] Expected<CodeModel> parseCodeModel(std::string_view Value);
 [[nodiscard]] Expected<OptimizationLevel> parseOptimizationLevel(std::string_view Value);
 [[nodiscard]] Expected<CodeGenFileType> parseCodeGenFileType(std::string_view Value);
+[[nodiscard]] Expected<GlobalISelAbortMode> parseGlobalISelAbortMode(std::string_view Value);
 [[nodiscard]] Expected<bool> parseCodeGenCommandFlag(std::span<const std::string_view> Arguments,
                                                      std::size_t &Index,
                                                      CodeGenCommandFlags &Flags);
diff --git a/backend/include/rcc/CodeGen/GlobalISel/Legalizer.h b/backend/include/rcc/CodeGen/GlobalISel/Legalizer.h
index 2ba0012..e90d09c 100644
--- a/backend/include/rcc/CodeGen/GlobalISel/Legalizer.h
+++ b/backend/include/rcc/CodeGen/GlobalISel/Legalizer.h
@@ -15,15 +15,18 @@ namespace rcc {
 class Legalizer final : public MachineFunctionPass {
 public:
   Legalizer(const LegalizerInfo *Info, OptimizationLevel Optimization,
-            LibcallLoweringInfo Libcalls = {})
-      : Info(Info), Optimization(Optimization), Libcalls(Libcalls) {}
+            LibcallLoweringInfo Libcalls = {},
+            GlobalISelAbortMode AbortMode = GlobalISelAbortMode::Enable)
+      : Info(Info), Optimization(Optimization), Libcalls(Libcalls), AbortMode(AbortMode) {}
   Legalizer(const LegalizerInfo &Info, OptimizationLevel Optimization,
-            LibcallLoweringInfo Libcalls = {})
-      : Info(&Info), Optimization(Optimization), Libcalls(Libcalls) {}
+            LibcallLoweringInfo Libcalls = {},
+            GlobalISelAbortMode AbortMode = GlobalISelAbortMode::Enable)
+      : Info(&Info), Optimization(Optimization), Libcalls(Libcalls), AbortMode(AbortMode) {}
   Legalizer(std::unique_ptr<LegalizerInfo> Info, OptimizationLevel Optimization,
-            LibcallLoweringInfo Libcalls = {})
+            LibcallLoweringInfo Libcalls = {},
+            GlobalISelAbortMode AbortMode = GlobalISelAbortMode::Enable)
       : OwnedInfo(std::move(Info)), Info(OwnedInfo.get()), Optimization(Optimization),
-        Libcalls(Libcalls) {}
+        Libcalls(Libcalls), AbortMode(AbortMode) {}
 
   [[nodiscard]] const MachinePassContract &contract() const noexcept override;
   [[nodiscard]] Expected<MachinePassResult> run(MachineFunction &Function,
@@ -39,6 +42,7 @@ private:
   const LegalizerInfo *Info = nullptr;
   OptimizationLevel Optimization;
   LibcallLoweringInfo Libcalls;
+  GlobalISelAbortMode AbortMode;
 };
 
 } // namespace rcc
diff --git a/backend/include/rcc/CodeGen/GlobalISel/LegalizerHelper.h b/backend/include/rcc/CodeGen/GlobalISel/LegalizerHelper.h
index 3e14311..aff6443 100644
--- a/backend/include/rcc/CodeGen/GlobalISel/LegalizerHelper.h
+++ b/backend/include/rcc/CodeGen/GlobalISel/LegalizerHelper.h
@@ -103,6 +103,8 @@ private:
                                                                    const APInt &Amount,
                                                                    LowLevelType TargetType,
                                                                    LowLevelType AmountType);
+  [[nodiscard]] Expected<void> fewerElements(MachineInstr &Instruction, std::size_t TypeIndex,
+                                             LowLevelType NarrowType);
 
   struct ShiftParameters final {
     Register WordShift;
diff --git a/backend/include/rcc/CodeGen/GlobalISel/LegalizerInfo.h b/backend/include/rcc/CodeGen/GlobalISel/LegalizerInfo.h
index 3433e3a..78e6a32 100644
--- a/backend/include/rcc/CodeGen/GlobalISel/LegalizerInfo.h
+++ b/backend/include/rcc/CodeGen/GlobalISel/LegalizerInfo.h
@@ -139,6 +139,7 @@ public:
 
   LegalizeRuleSet &custom();
   LegalizeRuleSet &widenScalarToNextPow2(std::size_t TypeIndex, std::uint32_t MinimumBitWidth = 0);
+  LegalizeRuleSet &scalarize(std::size_t TypeIndex);
   LegalizeRuleSet &minScalar(std::size_t TypeIndex, LowLevelType MinimumType);
   LegalizeRuleSet &minScalar(bool Enabled, std::size_t TypeIndex, LowLevelType MinimumType);
   LegalizeRuleSet &maxScalar(std::size_t TypeIndex, LowLevelType MaximumType);
diff --git a/backend/include/rcc/Target/RISCV/RISCVPipeline.h b/backend/include/rcc/Target/RISCV/RISCVPipeline.h
index 07f3ece..262f7c1 100644
--- a/backend/include/rcc/Target/RISCV/RISCVPipeline.h
+++ b/backend/include/rcc/Target/RISCV/RISCVPipeline.h
@@ -23,6 +23,7 @@ namespace riscv {
 struct RISCVPipelineOptions final {
   OptimizationLevel Optimization = OptimizationLevel::O2;
   bool UseTargetCopyInstructions = false;
+  GlobalISelAbortMode GlobalISelAbort = GlobalISelAbortMode::Enable;
 };
 
 [[nodiscard]] Expected<void> addCanonicalPipeline(MachineFunctionPassManager &Pipeline,