Skip to content

Commit

Permalink
allow to just build for 1 flow (#2798)
Browse files Browse the repository at this point in the history
  • Loading branch information
lanking520 committed Oct 6, 2023
1 parent 27c6a57 commit d432a65
Showing 1 changed file with 7 additions and 2 deletions.
9 changes: 7 additions & 2 deletions .github/workflows/build_triton_and_ft.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,11 @@ on:
description: 'fastertransformer branch/tag version'
required: true
default: 'main'
build_triton_only:
description: 'whether to just build triton library'
required: false
type: boolean
default: false
is_llama_build:
description: 'whether to build custom llama source'
required: false
Expand Down Expand Up @@ -47,7 +52,7 @@ jobs:
aws cloudfront create-invalidation --distribution-id E371VB8JQ6NRVY --paths "/tritonserver/${{ github.event.inputs.triton }}/*"
create-runner:
if: github.repository == 'deepjavalibrary/djl'
if: ${{ github.repository == 'deepjavalibrary/djl' && ! github.event.inputs.build_triton_only }}
runs-on: [ self-hosted, scheduler ]
steps:
- name: Create new CPU instance
Expand All @@ -64,7 +69,7 @@ jobs:


build-fastertransformer:
if: github.repository == 'deepjavalibrary/djl'
if: ${{ github.repository == 'deepjavalibrary/djl' }}
runs-on: [ self-hosted, cpu ]
container: deepjavalibrary/djl-serving:fastertransformer-nightly
timeout-minutes: 60
Expand Down

0 comments on commit d432a65

Please sign in to comment.