oneflow
oneflow copied to clipboard
Support the general constant folding pass in OneFlow MLIR
Folding all operations involving only constants which are represented as nn.Parameter
s.
- [x] Make a constant-like
VariableOp
for folding constants. (VariableWithoutCtrlOp
) - [x] Add pattern rewrites for converting
VariableOp
toVariableWithoutCtrlOp
. - [ ] Solidating
VariableWithoutCtrlOp
by reading corresponding data from local files. - [ ] Folding specific operations on
VariableWithoutCtrlOp
. - [ ] Generalize the folding pass for all operations involved.
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
Tianyu Zhao seems not to be a GitHub user. You need a GitHub account to be able to sign the CLA. If you have already a GitHub account, please add the email address used for this commit to your account.
You have signed the CLA already but the status is still pending? Let us recheck it.