search

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

reenable cudnn sdpa #639

Closed t-vi closed 5 days ago

t-vi commented 5 days ago

Fixes: #567

Thak you @vedaanta for reporting that it's fixed upstream.