add gpu flag build for triton client (#2815)

deepjavalibrary · Oct 23, 2023 · fe86680 · fe86680
1 parent f0b4334
commit fe86680
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: