[Bugfix] Fix lost async diffusion outputs in request-level batching (#6023)

Signed-off-by: samithuang <285365963@qq.com>
This commit is contained in:
Samit authored and GitHub committed 2026-08-15 18:27:51 +08:00
1 parent ce6ab0ca46
commit 136cb27b05
9 files changed
+475 -55

No files matched your search

+4
View File
@@ -19,6 +19,10 @@ ALLOWED_FILES = {
"tests/helpers/process.py",
"vllm_omni/diffusion/distributed/group_coordinator.py",
"tests/diffusion/attention/test_attention_sp.py",
# Measures the wire size of diffusion outputs. The transport pickles them,
# and only pickle reproduces the per-view storage duplication under test
# (torch.save deduplicates shared storages). No untrusted data is loaded.
"tests/diffusion/test_diffusion_ipc.py",
}
PICKLE_RE = re.compile(