Binyao Jiang 38c233fd04 [Piecewise] Support PCG weak_ref_tensor cuda kernel on AMD (#17291)

2026-01-20 14:05:32 -08:00

..

backend.py

EP Support for Piecewise Cuda Graph (#14164)

2025-12-20 01:59:27 +08:00

compilation_config.py

tiny refactor pcg split op registration (#16863)

2026-01-10 23:45:28 +08:00

compilation_counter.py

…

compile.py

[Piecewise Cuda Graph] rename, refactor and add more logging (#13675)

2025-11-21 13:28:38 +08:00

compiler_interface.py

Eager Compiler for Torch Compile (#11803)

2025-10-18 15:18:52 +08:00

cuda_piecewise_backend.py

Piecewise Cuda Graph Memory Usage (#15927)

2026-01-10 19:29:13 +08:00

fix_functionalization.py

…

fx_utils.py

…

inductor_pass.py

…

npu_piecewise_backend.py

[Ascend]Support of piecewise graph compilation for prefill on NPU (#12287)

2025-12-11 21:10:07 +08:00

pass_manager.py

…

piecewise_context_manager.py

Custom All Reduce for Piecewise Cuda Graph (#15356)

2025-12-25 23:55:15 +08:00

weak_ref_tensor.py

[Piecewise] Support PCG weak_ref_tensor cuda kernel on AMD (#17291)

2026-01-20 14:05:32 -08:00