mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-09-11 12:24:53 +00:00
misc(ci): make the wf callable for reuse (bis)
This commit is contained in:
parent
50de9eda85
commit
40d299f0c7
16
.github/workflows/trtllm_tests.yaml
vendored
16
.github/workflows/trtllm_tests.yaml
vendored
@ -30,13 +30,17 @@ jobs:
|
|||||||
|
|
||||||
run-tests:
|
run-tests:
|
||||||
needs: build-and-push
|
needs: build-and-push
|
||||||
|
concurrency:
|
||||||
|
group: ${{ github.workflow }}-${{ github.job }}-trtllm-${{ github.head_ref || github.run_id }}
|
||||||
|
cancel-in-progress: true
|
||||||
runs-on:
|
runs-on:
|
||||||
container:
|
group: aws-highmemory-64-plus-priv
|
||||||
image: ${{ needs.build-and-push.outputs.docker_image }}
|
container:
|
||||||
credentials:
|
image: ${{ needs.build-and-push.outputs.docker_image }}
|
||||||
username: ${{ secrets.REGISTRY_USERNAME }}
|
credentials:
|
||||||
password: ${{ secrets.REGISTRY_PASSWORD }}
|
username: ${{ secrets.REGISTRY_USERNAME }}
|
||||||
options: --gpus all
|
password: ${{ secrets.REGISTRY_PASSWORD }}
|
||||||
|
options: --gpus all --shm-size=8g
|
||||||
|
|
||||||
steps:
|
steps:
|
||||||
- name: Run C++/CUDA tests
|
- name: Run C++/CUDA tests
|
||||||
|
Loading…
Reference in New Issue
Block a user