Optimize function that loads pointers on GPU#3001

Open

timmoon10 wants to merge 23 commits into

NVIDIA:mainfrom

timmoon10:tmoon/optimize-get_device_pointer_for_data_and_scales

Commits on May 16, 2026

Generalize API for CUDA-Graph-safe copy to GPU.
timmoon10
committed
Dedup swizzle logic in get_device_pointer_for_data_and_scales

timmoon10
and
claude
committed
Make separate functions for load data_ptrs and swizzle + load data_ptrs.
timmoon10
committed
Change function name to nvte_load_value_on_device
timmoon10
committed
Fix code review issues before opening PR

timmoon10
and
claude
committed
Merge branch 'main' into tmoon/optimize-get_device_pointer_for_data_and_scales
timmoon10
committed
Formatter and review suggestions from @greptile-apps
timmoon10
committed

Commits on May 21, 2026