VJP rule for `PrimIDs.CLONE`, with some `torch.memory_format` handling logic

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.2k stars 80 forks source link

VJP rule for `PrimIDs.CLONE`, with some `torch.memory_format` handling logic #1098

Closed crcrpar closed 2 months ago

crcrpar commented 2 months ago

What does this PR do?

Fixes #1068.