[in-place][functionalization] Correct `.from_bsym_swap_proxies` on `new_bsym`, not `bsym`

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

[in-place][functionalization] Correct `.from_bsym_swap_proxies` on `new_bsym`, not `bsym` #653

Closed crcrpar closed 3 days ago

crcrpar commented 3 days ago

What does this PR do?

Fixes #652.

The wrong .from_bsym_swap_proxies call on bsym had discarded the signature change made in https://github.com/Lightning-AI/lightning-thunder/blob/1aaa463323dfe875ee75da9a3454779b69cb665c/thunder/core/transform_common.py#L490