-
Notifications
You must be signed in to change notification settings - Fork 5.7k
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] [NewIR]Automatic code generation for vjp rules #56512
[Prim] [NewIR]Automatic code generation for vjp rules #56512
Conversation
…e/Paddle into support_prim_code_gen
你的PR提交成功,感谢你对开源项目的贡献! |
6f3ea1c
to
7c87d93
Compare
8974188
to
deb37b5
Compare
… support_prim_vjp_code_gen
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
@@ -1,10 +1,17 @@ | |||
set(eager_backend_files | |||
${PADDLE_SOURCE_DIR}/paddle/fluid/primitive/backend/generated/generated_eager_backend.cc |
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.
这里其实用 CMAKE_CURRENT_SOURCE_DIR
更好一些
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.
好的 为了不影响后续工作,会在下一个pr统一修改
DEPS final_dygraph_function eager_utils phi) | ||
endif() | ||
set(static_backend_files | ||
${PADDLE_SOURCE_DIR}/paddle/fluid/primitive/backend/generated/generated_static_backend.cc | ||
${PADDLE_SOURCE_DIR}/paddle/fluid/primitive/backend/manual/manual_static_backend.cc |
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.
同上
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.
好的 为了不影响后续工作,会在下一个pr统一修改
@@ -0,0 +1,33 @@ | |||
set(fwd_path | |||
"${PADDLE_SOURCE_DIR}/paddle/fluid/operators/generator/parsed_ops/ops.parsed.yaml" |
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.
这里可以把 ${PADDLE_SOURCE_DIR}/paddle/fluid/operators/generator/parsed_ops
单独定义一个目录变量,这样后面的set 就简短很多,也比较好维护
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.
好的 下个pr会修改
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
…6512) * support ir api form prim * support ir api for prim * support vjp prim mode in new ir * remove useless code * remove useless code * auto code generator for primitive vjp methods * add vjp and backend manual and fix segment fault --------- Co-authored-by: cyber-pioneer <chenzhuo@tju.edu.cn> Co-authored-by: cxxly <chenxx_id@163.com>
PR types
New features
PR changes
Others
Description
Pcard-66975
背景:
本PR在PR55647基础上完成后续组合算子在新IR下适配的工作。
本PR完成的工作:
vjp
所需要的组件自动生成工作vjp
接口和backend
组件手写功能接下来要完成的工作
attribute