Skip to content

Arm backend: Add tests for Stable Diffusion 3.5 Large model - #22606

Merged
usamahz merged 2 commits into
pytorch:mainfrom
FabulousSuperDude:sd35_test
Sep 9, 2026
Merged

Arm backend: Add tests for Stable Diffusion 3.5 Large model#22606
usamahz merged 2 commits into
pytorch:mainfrom
FabulousSuperDude:sd35_test

Conversation

@FabulousSuperDude

@FabulousSuperDude FabulousSuperDude commented Sep 8, 2026

Copy link
Copy Markdown
Contributor
  • Add SD3.5 Large test configs for CLIP text encoders, T5 text encoder, MMDiT transformer, and VAE decoder, with optional upstream-sync validation.
  • Add Arm backend lowering tests for CLIP-L, CLIP-bigG, T5, transformer, and VAE decoder components across TOSA and VGF pipelines.
  • Add wrapper and model-loader unit tests covering wrapper output signatures, component subfolder loading, selective component loading, and dummy input shape generation.

cc @digantdesai @freddan80 @per @zingo @oscarandersson8218 @mansnils @Sebastian-Larsson @robell @rascani

@pytorch-bot

pytorch-bot Bot commented Sep 8, 2026

Copy link
Copy Markdown

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/executorch/22606

Note: Links to docs will display an error until the docs builds have been completed.

❌ 1 New Failure

As of commit fbb9ccc with merge base 78c85d7 (image):

NEW FAILURE - The following job has failed:

This comment was automatically generated by Dr. CI and updates every 15 minutes.

@meta-cla meta-cla Bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Sep 8, 2026
@github-actions github-actions Bot added ciflow/trunk module: arm Issues related to arm backend labels Sep 8, 2026
@pytorch-bot

pytorch-bot Bot commented Sep 8, 2026

Copy link
Copy Markdown

Workflows were awaiting approval. CI has now been triggered for the ciflow labels on this PR.

@FabulousSuperDude

Copy link
Copy Markdown
Contributor Author

@pytorchbot label "partner: arm"

@pytorch-bot pytorch-bot Bot added the partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm label Sep 8, 2026
@FabulousSuperDude

Copy link
Copy Markdown
Contributor Author

@pytorchbot label "release notes: arm"

@pytorch-bot pytorch-bot Bot added the release notes: arm Changes to the ARM backend delegate label Sep 8, 2026
@FabulousSuperDude
FabulousSuperDude marked this pull request as ready for review September 8, 2026 15:57
- Add SD3.5 Large test configs for CLIP text encoders, T5 text encoder,
  MMDiT transformer, and VAE decoder, with optional upstream-sync
  validation.
- Add Arm backend lowering tests for CLIP-L, CLIP-bigG, T5, transformer,
  and VAE decoder components across TOSA and VGF pipelines.
- Add wrapper and model-loader unit tests covering wrapper output
  signatures, component subfolder loading, selective component loading,
  and dummy input shape generation.

Signed-off-by: Sangwon Ha <sangwon.ha@arm.com>
Change-Id: Id7a58f7046c17b33875ee9ef1ffa9050fa41e207
Enable TOSA reference-model execution for the T5 floating-point test.
Share an int64-to-int32 pass factory between the CLIP and T5 tests,
creating fresh pass instances for each pipeline.

Set qtol=2 for the quantized VAE VGF test to accommodate
cross-platform VKML rounding differences.

Signed-off-by: Sangwon Ha <sangwon.ha@arm.com>
Change-Id: I73d0ce40410766ab010e6e7410521aaadfa35a78
@usamahz
usamahz merged commit 3dad019 into pytorch:main Sep 9, 2026
421 of 422 checks passed
@FabulousSuperDude
FabulousSuperDude deleted the sd35_test branch September 9, 2026 17:42
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/trunk CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. module: arm Issues related to arm backend partner: arm For backend delegation, kernels, demo, etc. from the 3rd-party partner, Arm release notes: arm Changes to the ARM backend delegate

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants