xla
Add autocast support for einsum
#8420
Merged
Overview
Commits
8
Changes
View On GitHub

Add autocast support for einsum #8420

jeffhataws merged 8 commits into pytorch:master from aws-nm9:einsum_autocast
aws-nm9
aws-nm9 Add autocast support for einsum
983b66dd
avizon-aws
avizon-aws commented on 2024-11-27
aws-nm9 Added and addressed comments
ccdf4a1b
rpsilva-aws
rpsilva-aws commented on 2024-11-27
jeffhataws jeffhataws requested a review from miladm miladm 293 days ago
jeffhataws jeffhataws requested a review from cowanmeg cowanmeg 293 days ago
jeffhataws jeffhataws requested a review from ManfeiBai ManfeiBai 293 days ago
jeffhataws
aws-nm9 YAPF formatting, consolidated tests, addressed comments
bd532baa
aws-nm9
aws-nm9 Re-lint to match indents
94fc62cb
bhavya01
bhavya01 commented on 2024-12-06
aws-nm9 Added testcase for fp32 outputs, addressing comments
f03fe6a1
aws-nm9 Linting fixes
b9e292b9
bhavya01
bhavya01 approved these changes on 2024-12-09
bhavya01
aws-nm9 Lint fix - add space after classname
d82f3a01
aws-nm9 Lint fix - blank space
9183c9c9
jeffhataws jeffhataws requested a review from tengyifei tengyifei 289 days ago
jeffhataws
jeffhataws approved these changes on 2024-12-09
jeffhataws jeffhataws merged 7b5aca68 into master 289 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum