-
Notifications
You must be signed in to change notification settings - Fork 662
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
Migrate repeat python layer to functor #7256
Conversation
std::vector<int32_t> expand_shape_vec; | ||
std::vector<int32_t> output_reshape_vec; | ||
|
||
int32_t diff = repeat_shape.NumAxes() - input_shape.NumAxes(); |
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.
diff -> numaxes_diff
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.
已修正
…neflow-Inc/oneflow into migrate_repeat_python_layer_to_functor
Code got formatted by CI. Please request CI again if you still want to have this PR merged. If the PR is from a forked repo, please download the patch files from the GitHub Actions web page and apply them locally. |
下次可以先 ninja of_format |
tile.py 里面调用 repeat 函数的方式有问题,repeat跟pytorch对齐后,应该传入的是*size,而不是tuple类型,这个分支先不合并了,等开发完tile一起合并。 |
CI failed when running job: cuda-misc. PR label automerge has been removed |
作为tile算子迁移的前置任务,和tile一起pr了,此pr关闭 #7305