Lightning-AI / lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.
Apache License 2.0
1.07k stars 60 forks source link

Check number of proxy args before indexing #642

Closed crcrpar closed 4 days ago

crcrpar commented 4 days ago

What does this PR do?

when an in-place op is applied to a tensor from a factory method, the producer bound symbol does not take any tensors as their args. Therefore it leads to out of range error.