always return epilogue inputs from prologue

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

always return epilogue inputs from prologue #636

Closed t-vi closed 6 days ago

t-vi commented 6 days ago

working towards #628