Skip to content

Support S32 index operations for copy and fill#8463

Open
rpsilva-aws wants to merge 1 commit intopytorch:masterfrom
rpsilva-aws:rpsilva_index_ops_int32
Open

Support S32 index operations for copy and fill#8463
rpsilva-aws wants to merge 1 commit intopytorch:masterfrom
rpsilva-aws:rpsilva_index_ops_int32

Conversation

@rpsilva-aws
Copy link
Copy Markdown
Collaborator

Follow-up to pytorch/pytorch#142160. This extends the INT32 flexibility for the remaining index ops, namely index_fill and index_copy. This is a requirement to natively accommodate users who require using INT32 for large number of indices.

@rpsilva-aws
Copy link
Copy Markdown
Collaborator Author

e.g. C++ exception with description "index_fill_(): Expected dtype int64 for index.

@rpsilva-aws
Copy link
Copy Markdown
Collaborator Author

rpsilva-aws commented Dec 6, 2024

cc: @miladm @ManfeiBai @tengyifei, dependency on pytorch/pytorch#142160.

@rpsilva-aws
Copy link
Copy Markdown
Collaborator Author

The build currently fails due to pytorch/pytorch#142160. This is blocked until that is merged.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant