[torchlib] Improve index_put #1749

Open

Open

[torchlib] Improve index_put#1749

justinchuby
openedon Jul 22, 2024

Handle None as inputs
Turn the function into trace_only
Fix failing CI tests

Ref:

[ONNX] opset 11 aten::index_put generating invalid add ops with None args pytorch/pytorch#131349
[ONNX] None as input to aten::index_put unsupported pytorch/pytorch#119363

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

Assignees

Labels

contribution welcomegood first issuetopic: torch_lib

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests