mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
[Static Runtime] Schema checks for index_put (#84152)
Summary: `index_put` can take a list of tensors, but Static Runtime always tries to convert its argument to a list of optional tensors. This was causing crashes for some users. Add some schema checks to prevent this, and add a new overload for the new case. Also, I found a clear bug in the JIT interpreter (mutating the argument when its not supposed to), so I fixed that too. Test Plan: New unit test Differential Revision: D39072214 Pull Request resolved: https://github.com/pytorch/pytorch/pull/84152 Approved by: https://github.com/tenpercent
This commit is contained in:
committed by
PyTorch MergeBot
parent
7532d5b125
commit
db7784e722
@ -1090,7 +1090,7 @@ static const std::vector<OperatorGeneratorArgs> opGenArgs{
|
||||
}
|
||||
auto self = pop(stack).toTensor();
|
||||
auto result =
|
||||
at::index_put_(self, opt_list_indices, values, accumulate);
|
||||
at::index_put(self, opt_list_indices, values, accumulate);
|
||||
push(stack, std::move(result));
|
||||
},
|
||||
aliasAnalysisFromSchema()),
|
||||
|
||||
Reference in New Issue
Block a user