Wan-Series Model Ascend NPU Adaptation Skill

Purpose

Provide a systematic, step-by-step guide for adapting Wan-series (and similar DiT-based) video generation models from NVIDIA CUDA/GPU to Huawei Ascend NPU. The skill encodes 9 major adaptation domains covering every layer of the inference stack, from device initialization to distributed parallelism.

When to Use

• Porting a Wan-series (Wan2.1 / Wan2.2) model from CUDA to Ascend NPU
• Adapting any DiT-based video diffusion model for Ascend hardware
• Optimizing inference performance on Ascend NPU (attention, quantization, VAE parallel)
• Setting up multi-card distributed inference on Atlas 800 series hardware
• Integrating MindIE SD acceleration library into a PyTorch video generation pipeline

Adaptation Domains Overview

The adaptation work is organized into 9 domains. Each domain has a dedicated reference file under references/ with detailed instructions, code patterns, and pitfalls.