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

Add TensorExt dialect with rotate op #478

Merged
merged 1 commit into from Mar 6, 2024

Conversation

j2kun
Copy link
Collaborator

@j2kun j2kun commented Mar 5, 2024

This is intended to be a generic rotate op that operates on any tensor type, mirroring the fhe.rotate from HECO, but operating in conjunction with secret.generic.

Later, a lowering from secret to BGV would have to lower an op like

secret.generic ins(%0: secret.secret<tensor<64xi16>>) {
^bb0(%arg1: tensor<16xi16>):
  %1 = tensor_ext.rotate %arg1, %c7
  secret.yield %1
}

where the rotation amount is a plaintext value defined in the ambient scope.

@j2kun j2kun added the pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing label Mar 5, 2024
@copybara-service copybara-service bot merged commit edc6d45 into google:main Mar 6, 2024
9 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
pull_ready Indicates whether a PR is ready to pull. The copybara worker will import for internal testing
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants