chen 63a03ee152 [feature]2.2 custom_allreduce support cudagraph recapture (#4307) ...

* custom_allreduce support cudagraph recapture

* delete code

* add shut_down/restart default group

2025-09-29 18:14:21 +08:00

..

__init__.py

[LLM] First commit the llm deployment code

2025-06-09 19:20:15 +08:00

cudagraph_piecewise_backend.py

[feature]2.2 custom_allreduce support cudagraph recapture (#4307)

2025-09-29 18:14:21 +08:00

decorator.py

[Executor] CUDAGraph support RL training (#3265)

2025-08-25 20:59:30 +08:00

dynamic_dims_marker.py

[SOT] Mark dynamic dims by type annotations (#2771)

2025-07-22 00:23:52 -07:00

graph_optimization_backend.py

[Executor] CUDAGraph support RL training (#3265)

2025-08-25 20:59:30 +08:00

utils.py

[SOT] Add sot warmup (NVIDIA GPU Only) (#2929)

2025-07-22 21:36:14 +08:00