Skip to content

Minimize the memory usage of the fused cross entropy kernel - #3273

Open
ptrendx wants to merge 12 commits into
NVIDIA:mainfrom
ptrendx:pr_cross_entropy_memory_usage
Open

Minimize the memory usage of the fused cross entropy kernel#3273
ptrendx wants to merge 12 commits into
NVIDIA:mainfrom
ptrendx:pr_cross_entropy_memory_usage

Add the guard against torch.compile when overwriting input

e0adb7c
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 21, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO