[DEV] Make CUDA graph compatible with FP8 params (tensorwise & blockwise). - #2087

Merged

yaox12 merged 9 commits into

kunlunl:kunlunl/fp8_param_cuda_graph_support

Nov 12, 2025

Commits on Nov 11, 2025

Add support for FP8 param + CUDA Graph
kunlunl
committed
8634f53
Update unit test for fp8 params + cuda graph
kunlunl
committed
9915b1f
Fix doc string & cuda graph scope
kunlunl
committed
3b96f30
Update CUDA graph arg
kunlunl
committed
040e827
Really eanble cuda graph in UT
kunlunl
committed
d450bbe
Rename keep_columnwise to manual_post_all_gather_processing
kunlunl
committed
ffca6f0
Fix lint error
kunlunl
committed
becb3fb
Fix lint error
kunlunl
committed
0d79b3a

Commits on Nov 12, 2025

Remove invalid use_te_rng_tracker arg
kunlunl
committed
63fd275