Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

chore: clean and speed up fhe training tests #724

Merged
merged 1 commit into from
Jun 17, 2024

Conversation

RomanBredehoft
Copy link
Collaborator

@RomanBredehoft RomanBredehoft commented Jun 11, 2024

we are compiling the fhe training circuit for fhe execution but only do simulation, which should speed up the test by a x2 factor. Usually compilation is quick, but in the composition case it can take around tens of seconds so this can save several minutes of running

for that, I need to remove the fixed values we've put in our compile method, which are not breaking changes since these are Concrete's default values + should allow anyone to be able to only compile for simulation or fhe execution (as explained abvoe)

@RomanBredehoft RomanBredehoft requested a review from a team as a code owner June 11, 2024 15:51
@cla-bot cla-bot bot added the cla-signed label Jun 11, 2024
@@ -470,6 +470,6 @@ def deserialize_decrypt_dequantize(

# In training mode, note that this step does not make much sense for now. Still, nothing
# breaks since QuantizedModule don't do anything in post-processing
result = self.model.post_processing(*result)
result_post_processed = self.model.post_processing(*result)
Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@RomanBredehoft RomanBredehoft force-pushed the chore/clean_speed_up_fhe_training_tests branch from 3b9384f to d31429a Compare June 11, 2024 15:55
jfrery
jfrery previously approved these changes Jun 13, 2024
Copy link
Collaborator

@jfrery jfrery left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Nice! Thanks for the cleaning

@RomanBredehoft RomanBredehoft force-pushed the chore/clean_speed_up_fhe_training_tests branch from d31429a to d1d0988 Compare June 13, 2024 14:45
Copy link

Coverage passed ✅

Coverage details

---------- coverage: platform linux, python 3.8.18-final-0 -----------
Name    Stmts   Miss  Cover   Missing
-------------------------------------
TOTAL    7885      0   100%

60 files skipped due to complete coverage.

@RomanBredehoft RomanBredehoft merged commit b379903 into main Jun 17, 2024
13 checks passed
@RomanBredehoft RomanBredehoft deleted the chore/clean_speed_up_fhe_training_tests branch June 17, 2024 09:16
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants