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

Am/refactor/factorize expansion code #1870

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

IceTDrinker
Copy link
Member

@IceTDrinker IceTDrinker commented Dec 13, 2024

closes: https://github.com/zama-ai/tfhe-rs-internal/issues/787

Adds sanitization for all expansion code paths

Factorize in a single function to avoid issues later, managed to make it work without it being too weird @nsarlin-zama

Maybe traits would be better, this works, I let you tweak if needed

@nsarlin-zama
Copy link
Contributor

Yes maybe this could use the Expandable trait ?
And maybe have another parameter or encode in the ExpansionMode if the verification should be done or not ?

@IceTDrinker
Copy link
Member Author

This was not the goal of the PR, I let you do that on top/after if you want to

@IceTDrinker IceTDrinker force-pushed the am/refactor/factorize-expansion-code branch from 7d9ea48 to 6eb8cb0 Compare January 2, 2025 12:14
Copy link
Contributor

@mayeul-zama mayeul-zama left a comment

Choose a reason for hiding this comment

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

Looks good to me!
Not sure if some parts needs a closer look

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.

4 participants