Migrate norms and softmax kernels to NVRTC - #3156

Merged

timmoon10 merged 12 commits into

NVIDIA:mainfrom

CarlosGomes98:cgomes/nvrtc-phase0

Jul 14, 2026

Commits on Jul 10, 2026

[Common] NVRTC for fused softmax and normalization (Phase 0)
CarlosGomes98
committed
28df6db
Add fully qualified name to softmax kernels
CarlosGomes98
committed
61dec6a
Add static fallback option, fix softmax acc_t dtype
CarlosGomes98
committed
7b79000
add missing license
CarlosGomes98
committed
019c0c9
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
CarlosGomes98
committed
fa42fd8
greptile changes
CarlosGomes98
committed
43c3297
fix formatting, .clang-format
CarlosGomes98
committed
4883fa8
[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot]
authored and
CarlosGomes98
committed
7f78167
import cleanup
CarlosGomes98
committed
fcbfeed
Test more columns for softmax, mr changes
CarlosGomes98
committed
518a68e
Fix tests
CarlosGomes98
committed
c7d875a

Commits on Jul 13, 2026

Merge branch 'main' into cgomes/nvrtc-phase0
timmoon10
authored
826881e