Skip to content

Conversation

@zpcore
Copy link
Member

@zpcore zpcore commented Sep 18, 2023

resolves #5583.

@zpcore zpcore requested review from JackCaoG and lsy323 September 18, 2023 20:17
@zpcore zpcore added lowering ATen Operation lowering codegen labels Sep 18, 2023
@lsy323
Copy link
Collaborator

lsy323 commented Sep 18, 2023

Thanks for preparing the follow up PR for #5586. LGTM pending on CI completion.

@zpcore
Copy link
Member Author

zpcore commented Sep 18, 2023

This is the follow up with the discussion in #5586.

We have switched to the codegen lowering method for arg_max and arg_min. So we can clean up the arg_min and arg_max related files and dependencies used in manual lowering.

@zpcore
Copy link
Member Author

zpcore commented Sep 18, 2023

Hi @JackCaoG , looks like the GPU test is running forever. Is this due to the #5570 as you mentioned?

@JackCaoG
Copy link
Collaborator

no.. it is just slow..

@zpcore zpcore merged commit a4874e2 into master Sep 19, 2023
@zpcore zpcore deleted the piz/codegen-1 branch September 19, 2023 00:23
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

codegen lowering ATen Operation lowering

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Codegen argmin and argmax op

4 participants