Skip to content

Commit f54497b

Browse files
authored
add missing DJL images (#5628)
Signed-off-by: sirutBuasai <sirutbuasai27@outlook.com>
1 parent fdf9c75 commit f54497b

11 files changed

+121
-20
lines changed

docs/DEVELOPMENT.md

Lines changed: 33 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -15,30 +15,30 @@ cd docs/src && python main.py --verbose
1515
```
1616
docs/
1717
├── src/
18-
│ ├── data/ # Per-image configuration files
18+
│ ├── data/ # Per-image configuration files
1919
│ │ ├── template/
20-
│ │ │ └── image-template.yml # Template with all fields documented
20+
│ │ │ └── image-template.yml # Template with all fields documented
2121
│ │ ├── pytorch-training/
22-
│ │ │ └── <version>-<accelerator>-<platform>.yml
22+
│ │ │ └── <version>-<accelerator>-<platform>.yml # Naming is for organization only
2323
│ │ └── ...
24-
│ ├── legacy/ # Historical support data
24+
│ ├── legacy/ # Historical support data
2525
│ │ └── legacy_support.yml
26-
│ ├── tables/ # Table column configurations
26+
│ ├── tables/ # Table column configurations
2727
│ │ └── <repository>.yml
2828
│ ├── templates/
29-
│ │ ├── reference/ # Reference page templates
30-
│ │ └── releasenotes/ # Release notes templates
31-
│ ├── constants.py # Path constants and GLOBAL_CONFIG
32-
│ ├── generate.py # Generation logic
33-
│ ├── global.yml # Shared terminology and configuration
34-
│ ├── hooks.py # MkDocs hooks
35-
│ ├── image_config.py # ImageConfig class
36-
│ ├── macros.py # MkDocs macros plugin
37-
│ ├── main.py # CLI entry point
38-
│ ├── sorter.py # Sorting tiebreaker functions
39-
│ └── utils.py # Utility functions
40-
├── reference/ # Generated reference pages
41-
├── releasenotes/ # Generated release notes
29+
│ │ ├── reference/ # Reference page templates
30+
│ │ └── releasenotes/ # Release notes templates
31+
│ ├── constants.py # Path constants and GLOBAL_CONFIG
32+
│ ├── generate.py # Generation logic
33+
│ ├── global.yml # Shared terminology and configuration
34+
│ ├── hooks.py # MkDocs hooks
35+
│ ├── image_config.py # ImageConfig class
36+
│ ├── macros.py # MkDocs macros plugin
37+
│ ├── main.py # CLI entry point
38+
│ ├── sorter.py # Sorting tiebreaker functions
39+
│ └── utils.py # Utility functions
40+
├── reference/ # Generated reference pages
41+
├── releasenotes/ # Generated release notes
4242
└── mkdocs.yml
4343
```
4444

@@ -55,17 +55,19 @@ Create `docs/src/data/<repository>/<version>-<accelerator>-<platform>.yml`:
5555
framework: PyTorch
5656
version: "2.9"
5757
accelerator: gpu # gpu, cpu, or neuronx
58-
python: py312
5958
platform: ec2 # ec2 or sagemaker
6059
tags:
6160
- "2.9.0-gpu-py312-cu130-ubuntu22.04-ec2"
6261

6362
# Optional metadata
63+
python: py312
6464
cuda: cu130
6565
os: ubuntu22.04
6666
public_registry: true
6767
```
6868
69+
The YAML file name is for organizational purposes only. However, make sure that the image configuration file lives in the correct repository directory.
70+
6971
See `docs/src/data/template/image-template.yml` for all available fields.
7072

7173
### Step 2: Regenerate
@@ -137,6 +139,10 @@ Release notes are generated automatically for images with `announcements` and `p
137139

138140
Sections render in YAML order as bullet lists.
139141

142+
Section headers in optional sections are rendered via the section key.
143+
To format your optional section headers, add a new field in `docs/src/global.yml` under `display_names` section.
144+
Eg: deprecation_notice section will render its header as `## deprecation_notice` unless a formatted string is provided in `docs/src/global.yml`.
145+
140146
______________________________________________________________________
141147

142148
## Adding a New Repository
@@ -181,6 +187,8 @@ columns:
181187
```
182188

183189
**Available fields:** `framework_version`, `python`, `cuda`, `sdk`, `accelerator`, `platform`, `os`, `example_url`, `version`, `ga`, `eop`, `framework_group`, `repository`, `release_note_link`
190+
To add additional fields, ensure that the image configuration YAML file contains said field of the same name.
191+
Additionally, if you require the field to be formatted, add an additional attribute in `ImageConfig` class of `display_<field_name>` to grab the formatted field.
184192

185193
______________________________________________________________________
186194

@@ -195,6 +203,8 @@ pytorch:
195203
eop: "2025-10-29"
196204
```
197205

206+
Generally, this is only required if an image configuration file does not already exist and the image is already past its support.
207+
198208
______________________________________________________________________
199209

200210
## Global Configuration
@@ -204,14 +214,17 @@ ______________________________________________________________________
204214
- **Terminology:** `aws`, `dlc_long`, `sagemaker`, etc.
205215
- **display_names:** Repository and package display names
206216
- **framework_groups:** Support policy consolidation groups
207-
- **table_order:** Order of tables in available_images.md
217+
- **table_order:** Order of tables displayed within the documentations website (eg: available_images.md and support_policy.md)
208218
- **platforms/accelerators:** Display mappings
209219

210220
______________________________________________________________________
211221

212222
## Running Generation
213223

214224
```bash
225+
# Help
226+
python main.py --help
227+
215228
# Full generation
216229
python main.py --verbose
217230
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
framework: DJLServing
2+
version: "0.27"
3+
accelerator: cpu
4+
engine: "CPU Full"
5+
platform: sagemaker
6+
7+
tags:
8+
- "0.27.0-cpu-full"
Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
framework: DJLServing
2+
version: "0.28"
3+
accelerator: cpu
4+
engine: "CPU Full"
5+
platform: sagemaker
6+
7+
tags:
8+
- "0.28.0-cpu-full"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.28"
3+
accelerator: gpu
4+
cuda: cu124
5+
engine: "LMI 10.0.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.28.0-lmi10.0.0-cu124"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.28"
3+
accelerator: gpu
4+
cuda: cu122
5+
engine: "TensorRT-LLM 0.9.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.28.0-tensorrtllm0.9.0-cu122"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.29"
3+
accelerator: gpu
4+
cuda: cu124
5+
engine: "LMI 11.0.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.29.0-lmi11.0.0-cu124"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.29"
3+
accelerator: gpu
4+
cuda: cu124
5+
engine: "TensorRT-LLM 0.11.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.29.0-tensorrtllm0.11.0-cu124"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.30"
3+
accelerator: gpu
4+
cuda: cu124
5+
engine: "LMI 12.0.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.30.0-lmi12.0.0-cu124"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.30"
3+
accelerator: gpu
4+
cuda: cu125
5+
engine: "TensorRT-LLM 0.12.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.30.0-tensorrtllm0.12.0-cu125"
Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,9 @@
1+
framework: DJLServing
2+
version: "0.31"
3+
accelerator: gpu
4+
cuda: cu124
5+
engine: "LMI 13.0.0"
6+
platform: sagemaker
7+
8+
tags:
9+
- "0.31.0-lmi13.0.0-cu124"

0 commit comments

Comments
 (0)