New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[PyTorch] Non-reentrant mode for activation recompute #670

Merged

ksivaman merged 12 commits into NVIDIA:main from denera:databricks/non-reentrant-checkpoint

Feb 24, 2024

Merged

[PyTorch] Non-reentrant mode for activation recompute #670

bias GELU fusion consistency between checkpoint test and reference co…

DCO / DCO succeeded Feb 23, 2024 in 1s

DCO

All commits are signed off!

View more details on DCO