-
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
[phi]migrate increment addmm multinomial cholesky InferShapes to phi #39913
[phi]migrate increment addmm multinomial cholesky InferShapes to phi #39913
Conversation
Thanks for your contribution! |
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,细节建议后续再完善一下
output_dims.push_back(x_dims[0]); | ||
output_dims.push_back(y_dims[1]); | ||
|
||
out->set_dims(make_ddim(output_dims)); |
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.
建议infermeta里一并set一下dtype,在原来体系里这边是一个漏洞
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.
了解,这几个我都加下.
std::vector<T> cumulative_probs(num_categories); | ||
|
||
std::uniform_real_distribution<T> dist(0, 1); | ||
auto gen_ptr = paddle::framework::DefaultCPUGenerator(); |
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.
thanks, done.
"X's shape[-2] = %d and shape[-1] = %d.", | ||
dims[rank - 2], | ||
dims[rank - 1])); | ||
out->set_dims(x.dims()); |
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.
这里也可以再设置下dtype
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 types
Others
PR changes
Others
Describe
1.Migrate increment addmm multinomial cholesky InferShapes to phi
2.Move MultinomialFunctor to paddle/phi/kernels/funcs
3.Make changes for comments in #39858