Skip to content

pytorch-xdit:v26.9 MAD integration #59

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
nsakkine merged 3 commits into from
Sep 18, 2026
Merged

pytorch-xdit:v26.9 MAD integration #59

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
330853c
pytorch-xdit:v26.9 update
nsakkine Sep 15, 2026
53308b9
Update integration instructions
nsakkine Sep 18, 2026
dac5f8a
Changed Minimax to MiniMax in README
nsakkine Sep 18, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion integrations/MAD/MAD/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,8 @@ latencies can be found from `results.csv` once the benchmark runs have finished.
| pyt_xdit_flux_2_klein | [Flux.2 Klein](https://huggingface.co/black-forest-labs/FLUX.2-klein-9B) |
| pyt_xdit_hunyuanvideo | [HunyuanVideo](https://huggingface.co/tencent/HunyuanVideo) |
| pyt_xdit_hunyuanvideo_1_5 | [HunyuanVideo 1.5](https://huggingface.co/tencent/HunyuanVideo-1.5) |
| pyt_xdit_ltx_2_3 | [LTX-2.3](https://huggingface.co/dg845/LTX-2.3-Diffusers) |
| pyt_xdit_ltx_2_5 | [LTX-2.5](https://huggingface.co/Lightricks/LTX-2.5) |
| pyt_xdit_minimax_h3 | [MiniMax-H3](https://huggingface.co/MiniMaxAI/MiniMax-H3) |
| pyt_xdit_qwen_image | [Qwen-Image](https://huggingface.co/Qwen/Qwen-Image-2512) |
| pyt_xdit_qwen_image_edit | [Qwen-Image-Edit](https://huggingface.co/Qwen/Qwen-Image-Edit) |
| pyt_xdit_sd_3_5 | [Stable diffusion 3.5](https://huggingface.co/stabilityai/stable-diffusion-3.5-large) |
Expand Down
79 changes: 49 additions & 30 deletions integrations/MAD/MAD/models.json
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,7 +1,8 @@
{
[
{
"name": "pyt_xdit_flux",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",

@lauri9 lauri9 Sep 18, 2026

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I see that the paths for dockerfile and scripts have changed throughout. Is the relative path for dockerfile relative to the location of this file, integrations/MAD/MAD/? If so, would ../.. not point one level too high, at ./integrations/ instead of the root of the repository?

@nsakkine nsakkine Sep 18, 2026 •
edited
Loading

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Some confusions exists with our structure, true. The point is that the new relative paths align with the new MAD repository directory structure. They have been tested during the QA process. integrations/MAD/MAD contents is nowadays purely copy to a MAD repository. I have now updated integrations/MAD/README.md accordingly

"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -17,8 +18,8 @@
},
{
"name": "pyt_xdit_flux_kontext",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -34,8 +35,8 @@
},
{
"name": "pyt_xdit_flux_2",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -53,8 +54,8 @@
},
{
"name": "pyt_xdit_flux_2_klein",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -70,8 +71,8 @@
},
{
"name": "pyt_xdit_hunyuanvideo",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -87,8 +88,8 @@
},
{
"name": "pyt_xdit_hunyuanvideo_1_5",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -103,9 +104,26 @@
"multiple_results": "results.csv"
},
{
"name": "pyt_xdit_ltx_2_3",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"name": "pyt_xdit_ltx_2_5",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
"tags": [
"inference",
"diffusion",
"t2av",
"pyt",
"xdit"
],
"args": "--workload ltx2_5",
"multiple_results": "results.csv"
},
{
"name": "pyt_xdit_minimax_h3",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -116,13 +134,13 @@
"pyt",
"xdit"
],
"args": "--workload ltx2_3",
"args": "--workload minimax_h3",
"multiple_results": "results.csv"
},
{
"name": "pyt_xdit_qwen_image",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -138,8 +156,8 @@
},
{
"name": "pyt_xdit_qwen_image_edit",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -155,8 +173,8 @@
},
{
"name": "pyt_xdit_sd_3_5",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -172,8 +190,8 @@
},
{
"name": "pyt_xdit_wan_2_1",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -189,8 +207,8 @@
},
{
"name": "pyt_xdit_wan_2_2",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -206,8 +224,8 @@
},
{
"name": "pyt_xdit_z_image",
"dockerfile": "docker/pyt_xdit",
"scripts": "scripts/pyt_xdit/run.sh",
"dockerfile": "../../docker/pyt_xdit",
"scripts": "run.sh",
"n_gpus": "8",
"owner": "nsakkine@amd.com",
"training_precision": "",
Expand All @@ -220,4 +238,5 @@
],
"args": "--workload z_image",
"multiple_results": "results.csv"
}
}
]
2 changes: 1 addition & 1 deletion integrations/MAD/MAD/pyt_xdit.ubuntu.amd.Dockerfile
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@
# SOFTWARE.
#
#################################################################################
ARG BASE_DOCKER=amdsiloai/pytorch-xdit:v26.8
ARG BASE_DOCKER=amdsiloai/pytorch-xdit:v26.9
FROM ${BASE_DOCKER} AS base

RUN apt-get update && apt install -y lshw && rm -rf /var/lib/apt/lists/*
8 changes: 4 additions & 4 deletions integrations/MAD/README.md
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,9 +2,9 @@

To make these changes effective in MAD

- copy `MAD/README.md`, `MAD/pyt_xdit.ubuntu.amd.Dockerfile` and `MAD/run.sh` to
MAD repository under `benchmark/xdit`, `docker` and `scripts/pyt_xdit` respectively
- concatenate MAD `models.json` with the contents of `models.json` in this
directory while maintaining json-lines format
- copy `MAD/README.md` and `MAD/pyt_xdit.ubuntu.amd.Dockerfile` to
MAD repository under `benchmark/xdit` and `docker` respectively
- copy `MAD/models.json` and `MAD/run.sh` to MAD repository under
`scripts/pyt_xdit`

and make sure that these changes get integrated into ROCm/MAD.
Loading