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

FixOp (argmax, argmin) | feat(torchlib) #1594

Merged
merged 1 commit into from
Jun 9, 2024

Conversation

titaiwangms
Copy link
Contributor

aten_argmax and aten_argmin have "dim=None" as default.

https://github.com/pytorch/pytorch/blob/2369c719d485af0787d95668947125a5605bed88/aten/src/ATen/native/native_functions.yaml#L810

Previous to "trace all traceable functions" PR, scripted function manages to handle unamtched attributes if they are None, but in traced function, this becomes errors of unrecognized arguments to the function.

@titaiwangms titaiwangms added the topic: torch_lib Related to the torch/aten function lib in development label Jun 8, 2024
@titaiwangms titaiwangms added bug Something isn't working and removed bug Something isn't working labels Jun 8, 2024
Copy link

codecov bot commented Jun 8, 2024

Codecov Report

All modified and coverable lines are covered by tests ✅

Project coverage is 75.95%. Comparing base (87618e8) to head (48d1663).

Additional details and impacted files
@@            Coverage Diff             @@
##             main    #1594      +/-   ##
==========================================
+ Coverage   75.93%   75.95%   +0.02%     
==========================================
  Files         228      228              
  Lines       24234    24246      +12     
  Branches     4197     4201       +4     
==========================================
+ Hits        18402    18416      +14     
+ Misses       5037     5035       -2     
  Partials      795      795              

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

Copy link
Collaborator

@justinchuby justinchuby left a comment

Choose a reason for hiding this comment

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

Thanks!

@justinchuby justinchuby merged commit 4c3a6be into microsoft:main Jun 9, 2024
39 of 42 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
topic: torch_lib Related to the torch/aten function lib in development
Projects
Development

Successfully merging this pull request may close these issues.

2 participants