Add Int comparison primitives in GHC.Classes

clash-lang · Feb 5, 2021 · 70718d0 · 70718d0
1 parent 833ac34
commit 70718d0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/changelog/2021-02-05T14_43_53+01_00_added_ltprim_rule.md b/changelog/2021-02-05T14_43_53+01_00_added_ltprim_rule.md
@@ -0,0 +1 @@
+CHANGED: Added a rule for missing Int comparisons in GHC.Classes in the compile time evaluator. [#1648](https://github.com/clash-lang/clash-compiler/issues/1648)
diff --git a/clash-ghc/src-ghc/Clash/GHC/Evaluator/Primitive.hs b/clash-ghc/src-ghc/Clash/GHC/Evaluator/Primitive.hs
@@ -918,10 +918,24 @@ ghcPrimStep tcm isSubj pInfo tys args mach = case primName pInfo of
                    }
        in (\e -> setTerm (Data e) mach) <$> dc
 
+  "GHC.Classes.eqInt" | Just (i,j) <- intCLiterals args
+    -> reduce (boolToBoolLiteral tcm ty (i == j))
+
+  "GHC.Classes.neInt" | Just (i,j) <- intCLiterals args
+    -> reduce (boolToBoolLiteral tcm ty (i /= j))
+
+  "GHC.Classes.leInt" | Just (i,j) <- intCLiterals args
+    -> reduce (boolToBoolLiteral tcm ty (i <= j))
+
+  "GHC.Classes.ltInt" | Just (i,j) <- intCLiterals args
+    -> reduce (boolToBoolLiteral tcm ty (i < j))
 
   "GHC.Classes.geInt" | Just (i,j) <- intCLiterals args
     -> reduce (boolToBoolLiteral tcm ty (i >= j))
 
+  "GHC.Classes.gtInt" | Just (i,j) <- intCLiterals args
+    -> reduce (boolToBoolLiteral tcm ty (i > j))
+
   "GHC.Classes.&&"
     | [ lArg , rArg ] <- args
     , eval <- Evaluator ghcStep ghcUnwind ghcPrimStep ghcPrimUnwind