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

Turn `tf32` off for resnet18 test cases #641

Closed crcrpar closed 4 days ago

crcrpar commented 4 days ago

What does this PR do?

Fixes #634 :crossed_fingers:.