[Perf][DreamZero]dreamzero supports CUDA graph, torch.compile, and DiT Caching. (#4213)

Signed-off-by: amy-why-3459 <wuhaiyan17@huawei.com>
This commit is contained in:
amy-why-3459 authored and GitHub committed 2026-06-19 14:24:37 +08:00
1 parent 63ba11b8b1
commit f05e03aea8
22 files changed
+1204 -177

No files matched your search

+2
View File
@@ -13,6 +13,8 @@ Main cache backend implementations:
the cache-dit library. Inherits from CacheBackend. Used via cache_backend="cache_dit".
2. TeaCacheBackend: Hook-based backend for TeaCache acceleration. Inherits from
CacheBackend. Used via cache_backend="tea_cache".
3. StepCacheBackend: Velocity cosine step-skipping for DreamZero. Inherits from
CacheBackend. Used via cache_backend="step_cache".
All backends implement the same interface:
- enable(pipeline): Enable cache on the pipeline