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

Fix div lowering and core aten test script enhancement #6873

Merged
merged 2 commits into from
Apr 3, 2024

Conversation

lsy323
Copy link
Collaborator

@lsy323 lsy323 commented Apr 2, 2024

  • Fix a shape inf bug in div lowering
  • Enhance aten test script. torch.allclose would broadcast the operands (e.g. torch.allclose([3, 1], [4]) would broadcast both operands to [3,4] and then compare). Add shape check before torch.allclose

@lsy323 lsy323 requested review from bhavya01 and wonjoolee95 and removed request for bhavya01 April 2, 2024 20:59
Copy link
Collaborator

@JackCaoG JackCaoG left a comment

Choose a reason for hiding this comment

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

@vanbasten23 as you are thinking about doing the same thing. We should also cherry-pick this into 2.3 branch as this is a regression.

@JackCaoG JackCaoG requested a review from bhavya01 April 2, 2024 21:01
@lsy323 lsy323 changed the title Fix div lowering and core aten enhance test Fix div lowering and core aten test script enhancement Apr 2, 2024
@lsy323 lsy323 merged commit 895b0c2 into master Apr 3, 2024
18 checks passed
@lsy323 lsy323 deleted the lsiyuan/fix-div branch April 3, 2024 00:00
lsy323 added a commit that referenced this pull request Apr 3, 2024
Co-authored-by: Siyuan Liu <lsiyuan@google.coim>
lsy323 added a commit that referenced this pull request Apr 3, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants