program_mapper: add support for muli

Author: AlexanderViand

p-isa_tools/program_mapper/p_isa/pisakernel.cpp

@@ -62,6 +62,24 @@ inline std::string genKernInput(const pisa::poly::PolyOperation &op)
               << " " << inputs[i].num_of_polynomials << "\n";
     }
 
+    // For muli operations, add the immediate value as an additional DATA input
+    bool has_immediate = false;
+    if (op.Name() == "muli")
+    {
+        // Check if we have an operand parameter (immediate value)
+        try
+        {
+            op.getParam("operand");
+            // Add immediate value as input1 with 1 polynomial (scalar)
+            input << "DATA input" << op.numInputOperands() << " 1\n";
+            has_immediate = true;
+        }
+        catch (...)
+        {
+            // No operand parameter, treat as regular operation
+        }
+    }
+
     // OP
     input << std::uppercase << op.Name() << std::nouppercase;
     for (int i = 0; i < op.numOutputOperands(); ++i)
@@ -74,6 +92,11 @@ inline std::string genKernInput(const pisa::poly::PolyOperation &op)
         input << " "
               << "input" << i /*inputs[i].register_name*/;
     }
+    // For muli, add the immediate as the second input
+    if (has_immediate)
+    {
+        input << " input" << op.numInputOperands();
+    }
 
     return input.str();
 }

p-isa_tools/program_mapper/poly_program/operations/core.h

@@ -40,6 +40,19 @@ static const PolyOperationDesc Sub("sub", { OP_NAME, FHE_SCHEME, POLYMOD_DEG_LOG
  * */
 static const PolyOperationDesc Mul("mul", { OP_NAME, FHE_SCHEME, POLYMOD_DEG_LOG2, KEY_RNS, OUTPUT_ARGUMENT, INPUT_ARGUMENT, INPUT_ARGUMENT });
 
+/** \brief PolyOperation multiply immediate (scalar multiplication) PolyOperationDesc
+ *   Op name: muli
+ *  | Param | description |
+ *  | ----- | ----------- |
+ *  | FHE_SCHEME   | specifies the FHE_SCHEME of the poly operation |
+ *  | POLYMOD_DEG_LOG2   | Specifies the modulus degree of the input polynomials |
+ *  | KEY_RNS   | Specifies number of RNS key values |
+ *  | OUTPUT_ARGUMENT   | Destination ciphertext |
+ *  | INPUT_ARGUMENT   | Input ciphertext label |
+ *  | PARAM   | Immediate/scalar value |
+ * */
+static const PolyOperationDesc Muli("muli", { OP_NAME, FHE_SCHEME, POLYMOD_DEG_LOG2, KEY_RNS, OUTPUT_ARGUMENT, INPUT_ARGUMENT, PARAM });
+
 /** \brief PolyOperation Square PolyOperationDesc
  *   Adds two polynomials with specified specified polymodulus degree, RNS terms, and polynomial parts and writes the result to the specified output.
  *  | Param | description |

p-isa_tools/program_mapper/poly_program/poly_operation_library.h

@@ -13,6 +13,7 @@ static std::map<std::string, PolyOperationDesc> core_operation_library = {
     { "sub", library::core::Sub },
     { "mul", library::core::Mul },
     { "mul_plain", library::core::Mul },
+    { "muli", library::core::Muli }, // Multiply immediate (scalar multiplication)
     { "square", library::core::Square },
     { "ntt", library::core::Ntt },
     { "intt", library::core::Intt },

p-isa_tools/program_mapper/poly_program/polyprogram.cpp

@@ -209,6 +209,10 @@ void PolyOperation::setComponents(const heracles::fhe_trace::Instruction &instr_
             setGaloisElt(stoi(v.value()));
         if (k == "factor")
             setFactor(stoi(v.value()));
+        if (k == "operand")
+            // For muli operations, the operand is the immediate scalar value
+            // Store it as a parameter that will be used when generating the kernel
+            setParam({ k, { v.value(), ValueType::DOUBLE } });
     }
 }
 heracles::fhe_trace::Instruction *PolyOperation::getProtobuffFHETraceInstruction()