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

feat(gpu): implement fhe rand on gpu #1958

Open
wants to merge 1 commit into
base: main
Choose a base branch
from

Conversation

guillermo-oyarzun
Copy link
Member

closes: please link all relevant issues

PR content/description

Check-list:

  • Tests for the changes have been added (for bug fixes / features)
  • Docs have been added / updated (for bug fixes / features)
  • Relevant issues are marked as resolved/closed, related issues are linked in the description
  • Check for breaking changes (including serialization changes) and add them to commit message following the conventional commit specification

@guillermo-oyarzun guillermo-oyarzun self-assigned this Jan 13, 2025
@cla-bot cla-bot bot added the cla-signed label Jan 13, 2025
@guillermo-oyarzun guillermo-oyarzun force-pushed the go/feature/implement-fherand-on-gpu branch from 349cfaa to 9088fb0 Compare January 13, 2025 18:04
Copy link
Contributor

@agnesLeroy agnesLeroy left a comment

Choose a reason for hiding this comment

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

Hey @guillermo-oyarzun! Thanks a lot for this PR, here comes my review. My main question is about the par_generate... entry points: at the moment the cuda calls are made on the same streams, which makes execution sequential. What we could do is use a different set of streams for each of the par_iter iteration maybe? Wdyt?

backends/tfhe-cuda-backend/cuda/include/linear_algebra.h Outdated Show resolved Hide resolved
tfhe/src/integer/gpu/server_key/radix/oprf.rs Outdated Show resolved Hide resolved
tfhe/src/integer/gpu/server_key/radix/oprf.rs Outdated Show resolved Hide resolved
tfhe/src/integer/gpu/server_key/radix/oprf.rs Outdated Show resolved Hide resolved
tfhe/src/integer/gpu/server_key/radix/oprf.rs Outdated Show resolved Hide resolved
tfhe/src/integer/gpu/server_key/radix/oprf.rs Outdated Show resolved Hide resolved
@guillermo-oyarzun guillermo-oyarzun force-pushed the go/feature/implement-fherand-on-gpu branch 2 times, most recently from d870628 to 5d54d65 Compare January 14, 2025 10:54
@guillermo-oyarzun guillermo-oyarzun force-pushed the go/feature/implement-fherand-on-gpu branch from 5d54d65 to 63eb9b8 Compare January 14, 2025 12:04
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