interpreter: (riscv) add fmadd.d interpreter function (#3720)

xdslproject · Jan 8, 2025 · 3c3b445 · 3c3b445
1 parent 0d01d47
commit 3c3b445
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 0 deletions.
diff --git a/tests/interpreters/test_riscv_interpreter.py b/tests/interpreters/test_riscv_interpreter.py
@@ -157,6 +157,16 @@ def my_custom_instruction(
 
     # D extension arithmetic
 
+    assert interpreter.run_op(
+        riscv.FMAddDOp(
+            TestSSAValue(fregister),
+            TestSSAValue(fregister),
+            TestSSAValue(fregister),
+            rd=riscv.FloatRegisterType.unallocated(),
+        ),
+        (3.0, 4.0, 5.0),
+    ) == (17.0,)
+
     assert interpreter.run_op(
         riscv.FAddDOp(
             TestSSAValue(fregister),

diff --git a/xdsl/interpreters/riscv.py b/xdsl/interpreters/riscv.py
@@ -428,6 +428,17 @@ def run_fmv(
 
     # region D extension
 
+    @impl(riscv.FMAddDOp)
+    def run_fmadd_d(
+        self,
+        interpreter: Interpreter,
+        op: riscv.FMAddDOp,
+        args: tuple[Any, ...],
+    ):
+        args = RiscvFunctions.get_reg_values(interpreter, op.operands, args)
+        results = (args[0] * args[1] + args[2],)
+        return RiscvFunctions.set_reg_values(interpreter, op.results, results)
+
     @impl(riscv.FAddDOp)
     def run_fadd_d(
         self,