google / google/heir

Support custom input scales

Open
#1,791 0 comments 0 reactions 1 assignee Claimed by @AlexanderViand View on GitHub
dialect: lattigo dialect: mgmt dialect: secret python frontend
Dominant language
MLIR
Stars
906
Forks
171
Avg merge
4d 12h
Merged PRs (30d)
32

Description

Placeholder issue arising from today's office hour.

1) make lattigo backend usable from Python frontend
2) Allow setting scale in frontend, probably something like `def foo(x: Secret[F32, Scale[50]], y: Secret[F32]):` which gets converted to a `func.func` argAttr, e.g., `{ckks.scale = 50}`
3) Make `--wrap-generic` pass forward function arg/result attributes to the `__argattrs`/`res_attrs` of the `secret.generic` wrapper
4) (Optional) add a pass to convert the `{ckks.scale = 50}` to the proper management attribute, e.g., `{mgmt.mgmt = #mgmt.mgmt}`
4) Make `--populate-scale-ckks`pass honor these attributes

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.