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

Compilation Folder Reset (#11539)

2025-10-14 09:19:12 +08:00

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

Compilation Folder Reset (#11539)

2025-10-14 09:19:12 +08:00

fx_utils.py

Compilation Folder Reset (#11539)

2025-10-14 09:19:12 +08:00

inductor_pass.py

Compilation Folder Reset (#11539)

2025-10-14 09:19:12 +08:00

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

Compilation Folder Reset (#11539)

2025-10-14 09:19:12 +08:00

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