[FEATURE] Run Wan2.2 pipelines on Ascend NPU - #41

Closed

yx0716 wants to merge 7 commits into

Tele-AI:mainfrom

yx0716:feat/npu-decode

Commits on Aug 31, 2026

fix(models): gate channels_last_3d activations to CUDA devices
yx0716
committed
6710dce
fix(distributed): default device mesh to the detected platform
yx0716
committed
ba34f29
fix(worker): marshal parallel-worker queue tensors through CPU off CUDA
yx0716
committed
f7d693d
refactor(ops): move fused FP8 QKV Triton kernels into kernel.triton
yx0716
committed
79ad176
fix(distributed): allocate pipeline P2P buffers on the platform device
yx0716
committed
162b9e3

Commits on Sep 1, 2026