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 some more core aten ops #6462

Merged
merged 4 commits into from
Feb 6, 2024
Merged

Conversation

wonjoolee95
Copy link
Collaborator

Fixes #5848

  • aten::as_strided is an op that should never be dispatched to torch_xla, as it is a view op and functionalization pass in PyTorch should decompose into aten::as_strided_copy (as noted in https://github.com/pytorch/xla/blob/master/codegen/xla_native_functions.yaml#L387, note that functionalization is, by default, enabled in torch_xla and we want that to be the default/norm).Therefore, remove this from the test.

Fixes #5861

  • Introduce a wrapper to pass in kwarg

Fixes #5859

  • Introduce a wrapper to pass in kwarg

Fixes #5897

  • Do manual dtype conversion in aten_sin

Copy link
Collaborator

@ManfeiBai ManfeiBai left a comment

Choose a reason for hiding this comment

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

LGTM

@wonjoolee95 wonjoolee95 merged commit e8dfc86 into master Feb 6, 2024
18 checks passed
amithrm pushed a commit to amithrm/xla that referenced this pull request Mar 1, 2024
bhavya01 pushed a commit that referenced this pull request Apr 22, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
2 participants