MaciejBalaNV c482626450 [optimization] Added Wan 2.2 VAE decode optimizations (#7056) ...

Signed-off-by: Maciej Bala <mbala@nvidia.com>
Signed-off-by: MaciejBalaNV <mbala@nvidia.com>

2026-09-22 11:37:49 -07:00

..

autoencoders

[optimization] Added Wan 2.2 VAE decode optimizations (#7056)

2026-09-22 11:37:49 -07:00

csrc

[Bugfix][Kernel] Stage strided Ulysses QKV with copy engine (#6814)

2026-08-31 17:28:49 +08:00

__init__.py

[REFACT]Refactor diffusion parallel state (#5531)

2026-08-19 11:23:10 +08:00

a2a_permute.py

[Bugfix][Kernel] Stage strided Ulysses QKV with copy engine (#6814)

2026-08-31 17:28:49 +08:00

cfg_parallel.py

[Feature][Diffusion]: CFG parallelism support for Boogu-Image (#6786)

2026-09-03 10:37:21 +08:00

comm.py

[Hardware] Support platforms and plugin system (#774)

2026-01-28 22:12:52 +08:00

group_coordinator.py

[REFACT]Refactor diffusion parallel state (#5531)

2026-08-19 11:23:10 +08:00

hsdp_utils.py

[Feat] support HSDP for Qwen-image series, Z-Image, GLM-Image (#2029)

2026-04-01 20:01:09 +08:00

hsdp.py

[Model] Add native MAGI-2 Preview diffusion support (#5918)

2026-09-05 15:32:30 +08:00

parallel_state.py

[Bugfix] Fix Parallel State Initialization For MoE + Diffusion (#7676)

2026-09-17 17:15:37 -06:00

pipeline_parallel.py

[Feature] Add support for Pipeline Parallel and integrate it into Wan 2.2 (#2322)

2026-05-19 09:42:17 +08:00

sp_plan.py

[Feature] Add AllGather-KV sequence-parallel attention str… (#4968)

2026-07-24 14:45:26 +00:00

sp_sharding.py

[Bugfix] Fix wan2.2 ti2v (#1221)

2026-02-06 17:28:49 +08:00

utils.py

Wan2.2: Fix the bug of using cache-dit with ulysses in t2v and i2v (#3927)

2026-06-25 15:15:59 +00:00

vae_patch_parallel.py

[Model]Add new nextstep_1(Diffusion) model(only T2I) (#612)

2026-02-26 00:09:11 +08:00