feat(div): Support fheDiv operation with/without scalar operand

zama-ai · Dec 30, 2024 · 0fd93e2 · 0fd93e2
1 parent 9e31ae6
commit 0fd93e2
Show file tree

Hide file tree

Showing 2 changed files with 47 additions and 0 deletions.
diff --git a/fhevm-engine/fhevm-go-native/fhevm/fhelib.go b/fhevm-engine/fhevm-go-native/fhevm/fhelib.go
@@ -46,6 +46,12 @@ func FheLibMethods() []*FheLibMethod {
 			runFunction:   fheMulRun,
 			ScalarSupport: true,
 		},
+		{
+			Name:          "fheDiv",
+			ArgTypes:      "(uint256,uint256,bytes1)",
+			runFunction:   fheDivRun,
+			ScalarSupport: true,
+		},
 		{
 			Name:              "fheRem",
 			ArgTypes:          "(uint256,uint256,bytes1)",

diff --git a/fhevm-engine/fhevm-go-native/fhevm/fhelib_ops.go b/fhevm-engine/fhevm-go-native/fhevm/fhelib_ops.go
@@ -150,6 +150,47 @@ func fheSubRun(sess ExecutorSession, unslicedInput []byte, _ ExtraData, outputHa
 	}
 }
 
+func fheDivRun(sess ExecutorSession, unslicedInput []byte, _ ExtraData, outputHandle []byte) error {
+	if len(unslicedInput) < 65 {
+		return fmt.Errorf("expected at least 65 bytes as input, got %d", len(unslicedInput))
+	}
+	input := unslicedInput[:65]
+
+	isScalar, err := isScalarOp(input)
+	if err != nil {
+		return err
+	}
+
+	var lhs, rhs []byte
+	if isScalar {
+		lhs, rhs, err = getScalarOperands(sess, input)
+	} else {
+		lhs, rhs, err = get2FheOperands(sess, input)
+	}
+	if err != nil {
+		return err
+	}
+
+	computation := ComputationToInsert{
+		Operation:    FheDiv,
+		OutputHandle: outputHandle,
+		Operands: []ComputationOperand{
+			{
+				Handle:      lhs,
+				FheUintType: handleType(lhs),
+				IsScalar:    false,
+			},
+			{
+				Handle:      rhs,
+				FheUintType: handleType(rhs),
+				IsScalar:    isScalar,
+			},
+		},
+	}
+
+	return sess.GetStore().InsertComputation(computation)
+}
+
 func fheMulRun(sess ExecutorSession, unslicedInput []byte, _ ExtraData, outputHandle []byte) error {
 	if len(unslicedInput) < 65 {
 		return fmt.Errorf("expected at least 65 bytes as input, got %d", len(unslicedInput))