Dropout with 8-bit RNG - #2014

Merged

timmoon10 merged 14 commits into

NVIDIA:mainfrom

vasunvidia:vrengasamy/fp8_dropout

Aug 31, 2025

Commits on Aug 28, 2025

Add dropout kernel with 8-bit RNG

timmoon10
and
vasunvidia
committed
b0cf4f2
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
3458ce6
Fix license
timmoon10
committed
220c6d4

Commits on Aug 29, 2025

Commits on Aug 31, 2025

Fix small statistical bug from using less-equal instead of less-than
timmoon10
committed
c075d94
Fix linter warning
timmoon10
committed
d35059a
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
528013f
Remove unnecessary helper function in PyTorch extensions
timmoon10
committed
9969285
Merge branch 'main' into vrengasamy/fp8_dropout
timmoon10
committed
ed07ce5
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot]
committed
1f26bd4