Skip to content

Commit 0c3b65a

Browse files
author
Qing Lan
committed
allow to just build for 1 flow
1 parent 27c6a57 commit 0c3b65a

File tree

1 file changed

+7
-2
lines changed

1 file changed

+7
-2
lines changed

.github/workflows/build_triton_and_ft.yml

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,11 @@ on:
1111
description: 'fastertransformer branch/tag version'
1212
required: true
1313
default: 'main'
14+
build_triton_only:
15+
description: 'whether to just build triton library'
16+
required: false
17+
type: boolean
18+
default: false
1419
is_llama_build:
1520
description: 'whether to build custom llama source'
1621
required: false
@@ -47,7 +52,7 @@ jobs:
4752
aws cloudfront create-invalidation --distribution-id E371VB8JQ6NRVY --paths "/tritonserver/${{ github.event.inputs.triton }}/*"
4853
4954
create-runner:
50-
if: github.repository == 'deepjavalibrary/djl'
55+
if: ${{ github.repository == 'deepjavalibrary/djl' && ! github.event.inputs.build_triton_only }}
5156
runs-on: [ self-hosted, scheduler ]
5257
steps:
5358
- name: Create new CPU instance
@@ -64,7 +69,7 @@ jobs:
6469

6570

6671
build-fastertransformer:
67-
if: github.repository == 'deepjavalibrary/djl'
72+
if: ${{ github.repository == 'deepjavalibrary/djl' }}
6873
runs-on: [ self-hosted, cpu ]
6974
container: deepjavalibrary/djl-serving:fastertransformer-nightly
7075
timeout-minutes: 60

0 commit comments

Comments
 (0)