-
Notifications
You must be signed in to change notification settings - Fork 5.6k
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
[PRIM][IR]Support prim in new ir #56342
[PRIM][IR]Support prim in new ir #56342
Conversation
你的PR提交成功,感谢你对开源项目的贡献! |
… support_prim_in_new_ir
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
* support ir api form prim * convert vector of int to intarray * support ir api for prim * support vjp prim mode in new ir * remove useless code * add test for prim * modify utils * remove useless code --------- Co-authored-by: cyber-pioneer <chenzhuo@tju.edu.cn>
PR types
New features
PR changes
Others
Description
Pcard-66975
背景:
本PR在PR55647基础上完成后续组合算子在新IR下适配的工作。
本PR完成的工作:
mean
算子前向拆解子图完成divide
算子和sum
算子反向组合模式在新IR下打通DescTensor
重命名为LazyTensor
接下来要完成的工作