From 0c3b65a2a3a5bdb3be72f091c2e3069a085960c3 Mon Sep 17 00:00:00 2001 From: Qing Lan Date: Mon, 2 Oct 2023 17:53:55 -0700 Subject: [PATCH] allow to just build for 1 flow --- .github/workflows/build_triton_and_ft.yml | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/.github/workflows/build_triton_and_ft.yml b/.github/workflows/build_triton_and_ft.yml index 2893e1567ff..61889327dd8 100644 --- a/.github/workflows/build_triton_and_ft.yml +++ b/.github/workflows/build_triton_and_ft.yml @@ -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 @@ -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 @@ -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