From d50360bb8630ea309898a733a597195ce1736fbe Mon Sep 17 00:00:00 2001 From: christopherngutierrez Date: Wed, 19 Mar 2025 10:34:07 -0700 Subject: [PATCH 1/2] added current_rns into immediate naming to enable non-fresh multi-ops --- kerngen/pisa_generators/mod.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/kerngen/pisa_generators/mod.py b/kerngen/pisa_generators/mod.py index dabeecc6..4c01e069 100644 --- a/kerngen/pisa_generators/mod.py +++ b/kerngen/pisa_generators/mod.py @@ -40,10 +40,12 @@ def to_pisa(self) -> list[PIsaOp]: # Immediates last_q = self.input0.rns - 1 - it = Immediate(name="it" + self.var_suffix) + it = Immediate(name="it" + self.var_suffix + f"_{self.context.current_rns}") t = Immediate(name="t", rns=last_q) one, r2, iq = common_immediates( - r2_rns=last_q, iq_rns=last_q, iq_suffix=self.var_suffix + r2_rns=last_q, + iq_rns=last_q, + iq_suffix=self.var_suffix + f"_{self.context.current_rns}", ) # Drop down input rns From bdb500a6098e9b3cdc6b949d33f7cf5893135ff7 Mon Sep 17 00:00:00 2001 From: Kylan Race Date: Wed, 19 Mar 2025 17:23:49 -0500 Subject: [PATCH 2/2] Updated Rescale to include Current-RNS on Label --- kerngen/pisa_generators/rescale.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kerngen/pisa_generators/rescale.py b/kerngen/pisa_generators/rescale.py index 5b7441f3..d0683dbf 100644 --- a/kerngen/pisa_generators/rescale.py +++ b/kerngen/pisa_generators/rescale.py @@ -42,7 +42,7 @@ def to_pisa(self) -> list[PIsaOp]: one, r2, iq = common_immediates(r2_rns=last_q, iq_rns=last_q) one, r2, iq = common_immediates( - r2_rns=last_q, iq_rns=last_q, iq_suffix=self.var_suffix + r2_rns=last_q, iq_rns=last_q, iq_suffix=self.var_suffix + f"_{self.context.current_rns}" ) q_last_half = Polys("qLastHalf", 1, self.input0.rns)