Handle f32.neg and f64.neg

Author: samestep

crates/floretta/src/reverse.rs

@@ -893,6 +893,12 @@ impl Func {
                 self.push_i64();
                 self.fwd.instructions().i64_rotr();
             }
+            Operator::F32Neg => {
+                self.pop();
+                self.push_f32();
+                self.fwd.instructions().f32_neg();
+                self.bwd.instructions(|insn| insn.f32_neg());
+            }
             Operator::F32Add => {
                 self.pop2();
                 self.push_f32();
@@ -922,6 +928,12 @@ impl Func {
                 self.fwd.instructions().call(FUNC_F32_DIV_FWD);
                 self.bwd.instructions(|insn| insn.call(FUNC_F32_DIV_BWD));
             }
+            Operator::F64Neg => {
+                self.pop();
+                self.push_f64();
+                self.fwd.instructions().f64_neg();
+                self.bwd.instructions(|insn| insn.f64_neg());
+            }
             Operator::F64Add => {
                 self.pop2();
                 self.push_f64();

crates/floretta/src/reverse/tests.rs

@@ -771,6 +771,19 @@ fn test_i64_rotr() {
     .test()
 }
 
+#[test]
+fn test_f32_neg() {
+    Backprop {
+        wat: include_str!("../wat/f32_neg.wat"),
+        name: "neg",
+        input: 3.0f32,
+        output: -3.0f32,
+        cotangent: 1.0f32,
+        gradient: -1.0f32,
+    }
+    .test()
+}
+
 #[test]
 fn test_f32_add() {
     Backprop {
@@ -823,6 +836,19 @@ fn test_f32_div() {
     .test()
 }
 
+#[test]
+fn test_f64_neg() {
+    Backprop {
+        wat: include_str!("../wat/f64_neg.wat"),
+        name: "neg",
+        input: 3.,
+        output: -3.,
+        cotangent: 1.,
+        gradient: -1.,
+    }
+    .test()
+}
+
 #[test]
 fn test_f64_add() {
     Backprop {

crates/floretta/src/wat/f32_neg.wat

@@ -0,0 +1,4 @@
+(module
+  (func (export "neg") (param f32) (result f32)
+    (f32.neg
+      (local.get 0))))

crates/floretta/src/wat/f64_neg.wat

@@ -0,0 +1,4 @@
+(module
+  (func (export "neg") (param f64) (result f64)
+    (f64.neg
+      (local.get 0))))