Skip to content

Commit 47186bb

Browse files
committed
feat: support hggc
Signed-off-by: thxCode <thxcode0824@gmail.com>
1 parent d297d69 commit 47186bb

8 files changed

Lines changed: 422 additions & 15 deletions

File tree

.github/workflows/discard.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ on:
2525
- corex
2626
- cuda
2727
- dtk
28+
- hggc
2829
- maca
2930
- musa
3031
- rocm

.github/workflows/pack.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@ on:
3535
- corex
3636
- cuda
3737
- dtk
38+
- hggc
3839
- maca
3940
- musa
4041
- rocm

.github/workflows/prune.yml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@ on:
2525
- corex
2626
- cuda
2727
- dtk
28+
- hggc
2829
- maca
2930
- musa
3031
- rocm

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -231,7 +231,7 @@ To add support for a new accelerated backend:
231231

232232
1. Create a new directory under `pack/` named with the new backend.
233233
2. Add a `Dockerfile` in the new directory following the [Dockerfile Convention](#dockerfile-convention).
234-
3. Update [pack.yml](.github/workflows/pack.yml) to include the new backend in the build matrix.
234+
3. Update [pack.yml](.github/workflows/pack.yml), [discard.yml](.github/workflows/discard.yml) and [prune.yml](.github/workflows/prune.yml) to include the new backend in the build matrix.
235235
4. Update [matrix.yml](pack/matrix.yaml) to include the new backend and its variants.
236236
5. Update `_RE_DOCKER_IMAGE` in [runner.py](gpustack_runner/runner.py) to recognize the new backend.
237237
6. [Optional] Update [tests](tests/gpustack_runner) if necessary.

gpustack_runner/runner.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from . import envs
1414

1515
_RE_DOCKER_IMAGE = re.compile(
16-
r"(?:(?P<prefix>[\w\\.\-]+(?:/[\w\\.\-]+)*)/)?runner:(?P<backend>(Host|cann|corex|cuda|dtk|maca|musa|rocm))(?P<backend_version>[XY\d\\.]+)(?:-(?P<backend_variant>\w+))?-(?P<service>(vllm|voxbox|mindie|sglang))(?P<service_version>[\w\\.]+)(?:-(?P<suffix>\w+))?",
16+
r"(?:(?P<prefix>[\w\\.\-]+(?:/[\w\\.\-]+)*)/)?runner:(?P<backend>(Host|cann|corex|cuda|dtk|hggc|maca|musa|rocm))(?P<backend_version>[XY\d\\.]+)(?:-(?P<backend_variant>\w+))?-(?P<service>(vllm|voxbox|mindie|sglang))(?P<service_version>[\w\\.]+)(?:-(?P<suffix>\w+))?",
1717
)
1818
"""
1919
Regex for Docker image parsing,

0 commit comments

Comments
 (0)