Add tiled VAE parallelism & extended model support - #14

Merged

nsakkine merged 102 commits into

xdit-project:mainfrom

pds-amd:feat/vae-bench-harness

Aug 19, 2026

Commits on Aug 4, 2026

test: make the distributed tests runnable without an accelerator

pds-amd
and
cursoragent
committed
8675416
fix(groupnorm): reduce the variance about the group mean, not the rank's

pds-amd
and
cursoragent
committed
c8e49ed
test: cover the two existing decoder adapters without a GPU

pds-amd
and
cursoragent
committed
039d655
feat(qwenimage): shard the Qwen-Image VAE decoder

pds-amd
and
cursoragent
committed
67871e3
fix(patchconv): stop cropping a halo that was already spent on padding

pds-amd
and
cursoragent
committed
a66bd6d
feat(hunyuanvideo): shard the HunyuanVideo and HunyuanVideo 1.5 VAE decoders

pds-amd
and
cursoragent
committed
eed8603
feat(ltx2): shard the LTX-2 VAE decoder

pds-amd
and
cursoragent
committed
103ec61
docs(setup): say why the diffusers floor stays at 0.35

pds-amd
and
cursoragent
committed
49310aa
fix(patchify): split rows unevenly instead of padding them to divide

pds-amd
and
cursoragent
committed
4bda9b3
feat(qwenimage): shard the Qwen-Image VAE encoder

pds-amd
and
cursoragent
committed
9ed7caa
feat(hunyuanvideo): shard the HunyuanVideo and HunyuanVideo 1.5 VAE encoders

pds-amd
and
cursoragent
committed
9b87a9d
feat(ltx2): shard the LTX-2 VAE encoder

pds-amd
and
cursoragent
committed
b0dc649
test: keep the gloo rendezvous port below the ephemeral range

pds-amd
and
cursoragent
committed
46fadd0
feat(vae): shard the 2D VAE encoder

pds-amd
and
cursoragent
committed
69b0eae

Commits on Aug 5, 2026

Commits on Aug 6, 2026

Commits on Aug 7, 2026

Commits on Aug 10, 2026

Fix omitted cache cursors and describe-only runs

pds-amd
and
cursoragent
committed
9b2588c
Keep VAE parallel state adapter-local

pds-amd
and
cursoragent
committed
5210428
Own VAE sharding and tiling policy

pds-amd
and
cursoragent
committed
4f7e872
Make VAE benchmark runner-independent

pds-amd
and
cursoragent
committed
20e154e
Separate VAE adaptation from instrumentation

pds-amd
and
cursoragent
committed
9486985
Version the public VAE orchestration API

pds-amd
and
cursoragent
committed
c545f3e
Tighten VAE tests and invariants

pds-amd
and
cursoragent
committed
9235fd9
Enforce structural benchmark agreement

pds-amd
and
cursoragent
committed
1da6688
Cover uneven Wan zero-pad bands

pds-amd
and
cursoragent
committed
1509aa0
Support rectangular VAE tile windows

pds-amd
and
cursoragent
committed
e2c3107

Commits on Aug 11, 2026

Support strip-shaped VAE tiles

pds-amd
and
cursoragent
committed
c29549f
Merge remote-tracking branch 'origin/feat/vae-bench-harness' into integrate/vae-bench-push
pds-amd
committed
616577e
Merge branch 'feat/vae-bench-harness' into integrate/vae-bench-push
pds-amd
committed
eeac96a
Escape tile scheduler local optima

pds-amd
and
cursoragent
committed
c6567b2
Preserve adapted decoder runtime state

pds-amd
and
cursoragent
committed
bd3df86
Simplify distributed VAE adapters

pds-amd
and
cursoragent
committed
7adaa5e
Simplify VAE tiling controls

pds-amd
and
cursoragent
committed
bb286e8
Simplify benchmark execution modes

pds-amd
and
cursoragent
committed
6297c80
Redesign benchmark cases around Pareto tiles

pds-amd
and
cursoragent
committed
e2200b3
Generalize asymmetric zero-pad convolution

pds-amd
and
cursoragent
committed
62f5d61
Search tile overlap instead of pinning it to the VAE native
pds-amd
committed
68d57a5
Bound the narrow tile axis away from banding
pds-amd
committed
9ebe500
Keep every blend above a quarter of its window
pds-amd
committed
a7665f7
Update the overlap ladder test to the quarter-of-window bound
pds-amd
committed
6a21457
Stop the memory profile returning throughput's transpose
pds-amd
committed
25f40de
Default the suite to the compositions a caller can select
pds-amd
committed
aa55691
Select the throughput plan on window area, not total work
pds-amd
committed
f4ac69f
Name the tile profiles for geometry, not for predicted outcome
pds-amd
committed
d10d59d
Carry the shape matrix and the measured device in the harness
pds-amd
committed
fe4fd74

Commits on Aug 12, 2026

Harden reproducible benchmark sweeps

pds-amd
and
cursoragent
committed
cd8186c
Warn when tile ranks repeat work

pds-amd
and
cursoragent
committed
12043df
Harden dependency compatibility

pds-amd
and
cursoragent
committed
89822f4
Refine VAE benchmark plan selection

pds-amd
and
cursoragent
committed
cdc8e80
Merge remote benchmark updates

pds-amd
and
cursoragent
committed
0a640dc
Clarify distributed VAE decode strategies

pds-amd
and
cursoragent
committed
f96e41d
Remove legacy adapter model copies

pds-amd
and
cursoragent
committed
da0c17e
Remove implementation-detail tests

pds-amd
and
cursoragent
committed
90fac5f

Commits on Aug 13, 2026

Fix rectangular latent row lookup

pds-amd
and
cursoragent
committed
9165ee0
Preserve rectangular latent geometry

pds-amd
and
cursoragent
committed
fa6cd5c
Expose xDiT VAE orchestration helpers

pds-amd
and
cursoragent
committed
500ef13
Validate rectangular tile call counts

pds-amd
and
cursoragent
committed
764e23d
Type VAE row split failures

pds-amd
and
cursoragent
committed
17fcce7

Commits on Aug 17, 2026

fix(workflows): update renamed adapter compatibility test
pds-amd
committed
c3cfcfb
fix(setup): scope find_packages to distvae namespace
pds-amd
committed
3da9f4f
fix: cache tile share assignments
pds-amd
committed
316e9fd

Commits on Aug 18, 2026

fix: share safe convolution chunking

pds-amd
and
cursoragent
committed
1804be2
docs: describe distributed group normalization

pds-amd
and
cursoragent
committed
1ce25b5
docs: explain patch shard materialization

pds-amd
and
cursoragent
committed
01161ab
fix: bound tile assignment refinement

pds-amd
and
cursoragent
committed
15cd143

Commits on Aug 19, 2026

fix: bound distributed test hangs

pds-amd
and
cursoragent
committed
5000830