Skip to content
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

stack, split_with_num infer spmd #58854

Merged
merged 26 commits into from
Nov 15, 2023

Conversation

liuzhenhai93
Copy link
Contributor

@liuzhenhai93 liuzhenhai93 commented Nov 9, 2023

PR types

Others

PR changes

Others

Description

  • stack 前反向
  • concat 反向
  • code gen 支持 vector of tensor 输出
  • card-77379 拼接类算子切分推导规则

wanghuancoder
wanghuancoder previously approved these changes Nov 13, 2023
Copy link
Contributor

@wanghuancoder wanghuancoder left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

paddle/phi/api/lib/api_gen_utils.cc Outdated Show resolved Hide resolved
paddle/phi/infermeta/spmd_rules/concat.cc Outdated Show resolved Hide resolved
paddle/phi/infermeta/spmd_rules/stack.cc Outdated Show resolved Hide resolved
chenwhql
chenwhql previously approved these changes Nov 14, 2023
Copy link
Contributor

@chenwhql chenwhql left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

GhostScreaming
GhostScreaming previously approved these changes Nov 14, 2023
Copy link
Contributor

@GhostScreaming GhostScreaming left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@liuzhenhai93 liuzhenhai93 dismissed stale reviews from GhostScreaming and chenwhql via 2f62816 November 14, 2023 12:13
Copy link
Contributor

@JZ-LIANG JZ-LIANG left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM

@chenwhql chenwhql merged commit 2d39be1 into PaddlePaddle:develop Nov 15, 2023
SecretXV pushed a commit to SecretXV/Paddle that referenced this pull request Nov 28, 2023
* format

* polish

* polish

* polish

* polish

* yaml

* polish

* format

* polish_test

* polish

* code gen

* polish

* polish

* follow comment

* fix ci

* fix ci

* polish

* polish
@liuzhenhai93 liuzhenhai93 deleted the infer_split_with_num branch December 4, 2023 13:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

5 participants