Add OpUMod

hernanponcedeleon · Nov 16, 2024 · 6322d95 · 6322d95
1 parent 4fc81b1
commit 6322d95
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/...rc/main/java/com/dat3m/dartagnan/parsers/program/visitors/spirv/VisitorOpsArithmetic.java b/...rc/main/java/com/dat3m/dartagnan/parsers/program/visitors/spirv/VisitorOpsArithmetic.java
@@ -62,6 +62,11 @@ public Event visitOpSDiv(SpirvParser.OpSDivContext ctx) {
         return visitIntegerBinExpression(ctx.idResult(), ctx.idResultType(), ctx.operand1(), ctx.operand2(), DIV);
     }
 
+    @Override
+    public Event visitOpUMod(SpirvParser.OpUModContext ctx) {
+        return visitIntegerBinExpression(ctx.idResult(), ctx.idResultType(), ctx.operand1(), ctx.operand2(), UREM);
+    }
+
     private Event visitIntegerUnExpression(
             SpirvParser.IdResultContext idCtx,
             SpirvParser.IdResultTypeContext typeCtx,
@@ -122,7 +127,8 @@ public Set<String> getSupportedOps() {
                 "OpISub",
                 "OpIMul",
                 "OpUDiv",
-                "OpSDiv"
+                "OpSDiv",
+                "OpUMod"
         );
     }
 }
diff --git a/...n/src/main/java/com/dat3m/dartagnan/parsers/program/visitors/spirv/VisitorOpsLogical.java b/...n/src/main/java/com/dat3m/dartagnan/parsers/program/visitors/spirv/VisitorOpsLogical.java
@@ -59,7 +59,7 @@ public Event visitOpSelect(SpirvParser.OpSelectContext ctx) {
                     "expected two operands type '%s but received '%s' and '%s'",
                     id, type, op1.getType(), op2.getType());
         }
-        if (op1.getType() instanceof IntegerType) {
+        if (op1.getType() instanceof IntegerType || op1.getType() instanceof  BooleanType) {
             return builder.addEvent(new Local(register, expressions.makeITE(cond, op1, op2)));
         }
         throw new ParsingException("Illegal definition for '%s', " +