openfhe: use default value for evalAddCount / keySwitchCount when not specified in params

asraa · copybara-github · commit cb9cd38998a7 · 2025-04-04T09:33:39.000-07:00
Fixes #1656 PiperOrigin-RevId: 743971798
diff --git a/lib/Dialect/Openfhe/Transforms/ConfigureCryptoContext.cpp b/lib/Dialect/Openfhe/Transforms/ConfigureCryptoContext.cpp
@@ -302,6 +302,8 @@ struct ConfigureCryptoContext
     }
 
     // get evalAddCount/KeySwitchCount from func attribute, if present
+    config.evalAddCount = 0;
+    config.keySwitchCount = 0;
     if (auto openfheParamsAttr = op->getAttrOfType<mgmt::OpenfheParamsAttr>(
             mgmt::MgmtDialect::kArgOpenfheParamsAttrName)) {
       config.evalAddCount = openfheParamsAttr.getEvalAddCount();

Original file line number	Diff line number	Diff line change
`@@ -302,6 +302,8 @@ struct ConfigureCryptoContext`
`302`	`302`	`}`
`303`	`303`
`304`	`304`	`// get evalAddCount/KeySwitchCount from func attribute, if present`
	`305`	`+ config.evalAddCount = 0;`
	`306`	`+ config.keySwitchCount = 0;`
`305`	`307`	`if (auto openfheParamsAttr = op->getAttrOfType<mgmt::OpenfheParamsAttr>(`
`306`	`308`	`mgmt::MgmtDialect::kArgOpenfheParamsAttrName)) {`
`307`	`309`	`config.evalAddCount = openfheParamsAttr.getEvalAddCount();`