add: madhyasthal+jit/x64: GTE and GTEI impl

Author: xTrayambak

src/bali/runtime/compiler/amd64/midtier.nim

@@ -371,6 +371,66 @@ proc compileLowered(cgen: MidtierJIT, pipeline: pipeline.Pipeline): Option[JITSe
         cgen.s.mov(regRsi.reg, regRax)
         cgen.s.mov(regRdx, int64(inst.args[0].vreg))
         cgen.s.call(cgen.callbacks.addAtom)
+    of InstKind.GreaterThanInt:
+      let
+        a = inst.args[0].vreg
+        b = inst.args[1].vreg
+
+      alignStack 8:
+        cgen.s.mov(regRdi, cast[int64](cgen.vm))
+        cgen.s.mov(regRsi, int64(a))
+        cgen.s.call(cgen.callbacks.getAtom)
+
+        cgen.s.mov(regRdi.reg, regRax)
+        cgen.s.call(getRawFloat)
+
+        cgen.s.movsd(regXmm1, regXmm0.reg)
+
+        cgen.s.mov(regRdi, cast[int64](cgen.vm))
+        cgen.s.mov(regRsi, int64(b))
+        cgen.s.call(cgen.callbacks.getAtom)
+
+      cgen.s.comisd(regXmm0, regXmm1.reg)
+
+      # A <= B: pc += 1
+      cgen.patchConditionalJmps &=
+        ConditionalJump(label: cgen.s.label(), op: i + 1, condition: condNotBelow)
+      cgen.s.jcc(condNotBelow, BackwardsLabel(EnsureNoInt8Align))
+
+      # A > B: pc += 2
+      cgen.patchConditionalJmps &=
+        ConditionalJump(label: cgen.s.label(), op: i + 2, condition: condBelow)
+      cgen.s.jcc(condBelow, BackwardsLabel(EnsureNoInt8Align))
+    of InstKind.GreaterThanEqualInt:
+      let
+        a = inst.args[0].vreg
+        b = inst.args[1].vreg
+
+      alignStack 8:
+        cgen.s.mov(regRdi, cast[int64](cgen.vm))
+        cgen.s.mov(regRsi, int64(a))
+        cgen.s.call(cgen.callbacks.getAtom)
+
+        cgen.s.mov(regRdi.reg, regRax)
+        cgen.s.call(getRawFloat)
+
+        cgen.s.movsd(regXmm1, regXmm0.reg)
+
+        cgen.s.mov(regRdi, cast[int64](cgen.vm))
+        cgen.s.mov(regRsi, int64(b))
+        cgen.s.call(cgen.callbacks.getAtom)
+
+      cgen.s.comisd(regXmm0, regXmm1.reg)
+
+      # A >= B: pc += 1
+      cgen.patchConditionalJmps &=
+        ConditionalJump(label: cgen.s.label(), op: i + 1, condition: condNbequal)
+      cgen.s.jcc(condNbequal, BackwardsLabel(EnsureNoInt8Align))
+
+      # A > B: pc += 2
+      cgen.patchConditionalJmps &=
+        ConditionalJump(label: cgen.s.label(), op: i + 2, condition: condBequal)
+      cgen.s.jcc(condBequal, BackwardsLabel(EnsureNoInt8Align))
     else:
       debug "jit/amd64: midtier cannot lower op into x64 code: " & $inst.kind
       return

src/bali/runtime/compiler/madhyasthal/ir.nim

@@ -27,6 +27,8 @@ type
     Equate
     LesserThanInt
     Increment
+    GreaterThanInt
+    GreaterThanEqualInt
 
   Register* {.size: sizeof(uint8), pure.} = enum
     ReturnValue = 0
@@ -183,4 +185,16 @@ func increment*(a: uint32): Inst =
   Inst(
     kind: InstKind.Increment, args: [ArgVariant(kind: avkPos, vreg: a), ArgVariant()]
   )
+
+func greaterThanI*(a, b: uint32): Inst =
+  Inst(
+    kind: InstKind.GreaterThanInt,
+    args: [ArgVariant(kind: avkPos, vreg: a), ArgVariant(kind: avkPos, vreg: b)],
+  )
+
+func greaterThanEqI*(a, b: uint32): Inst =
+  Inst(
+    kind: InstKind.GreaterThanEqualInt,
+    args: [ArgVariant(kind: avkPos, vreg: a), ArgVariant(kind: avkPos, vreg: b)],
+  )
 {.pop.}

src/bali/runtime/compiler/madhyasthal/lowering.nim

@@ -258,6 +258,20 @@ proc lowerStream*(fn: var Function, stream: var OpStream): bool =
         lesserThanI(
           uint32(&op.arguments[0].getInt()), uint32(&op.arguments[1].getInt())
         )
+    of GreaterThanInt:
+      let op = stream.consume()
+
+      fn.insts &=
+        greaterThanI(
+          uint32(&op.arguments[0].getInt()), uint32(&op.arguments[1].getInt())
+        )
+    of GreaterThanEqualInt:
+      let op = stream.consume()
+
+      fn.insts &=
+        greaterThanEqI(
+          uint32(&op.arguments[0].getInt()), uint32(&op.arguments[1].getInt())
+        )
     of Increment:
       let op = stream.consume()
 

src/bali/runtime/compiler/madhyasthal/naive_dce.nim

@@ -112,7 +112,10 @@ func scanForUsedRegs*(
       used.incl(inst.args[0].vreg)
     of InstKind.ReadProperty:
       used.incl(inst.args[0].vreg)
-    of {InstKind.Equate, InstKind.LesserThanInt}:
+    of {
+      InstKind.Equate, InstKind.LesserThanInt, InstKind.GreaterThanInt,
+      InstKind.GreaterThanEqualInt,
+    }:
       used.incl(inst.args[0].vreg)
       used.incl(inst.args[1].vreg)
     else: