add gpu flag build for triton client

deepjavalibrary · Oct 20, 2023 · 790ab3d · 790ab3d
1 parent 0c3b65a
commit 790ab3d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/build_triton_and_ft.yml b/.github/workflows/build_triton_and_ft.yml
@@ -38,7 +38,7 @@ jobs:
       - name: Build Triton Binary
         shell: 'script -q -e -c "bash --noprofile --norc -eo pipefail {0}"'
         run: |
-          python3 build.py --enable-logging --enable-metrics --enable-stats --enable-cpu-metrics --endpoint http
+          python3 build.py --enable-logging --enable-metrics --enable-stats --enable-cpu-metrics --enable-gpu --endpoint http
       - name: Configure AWS Credentials
         uses: aws-actions/configure-aws-credentials@v2
         with: