From 63ee666a22fe097657375c999cbe27c28396340d Mon Sep 17 00:00:00 2001
From: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com>
Date: Mon, 2 Dec 2024 09:25:23 +0100
Subject: [PATCH] add buildx command for vllm cpu docker image build

Signed-off-by: Maxime Fournioux <55544262+mfournioux@users.noreply.github.com>
---
 .github/workflows/lint-test.yaml | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/.github/workflows/lint-test.yaml b/.github/workflows/lint-test.yaml
index 118f2ae10eb35..fd091084767d1 100644
--- a/.github/workflows/lint-test.yaml
+++ b/.github/workflows/lint-test.yaml
@@ -49,12 +49,14 @@ jobs:
       - name: Create kind cluster
         uses: helm/kind-action@v1.10.0
 
+      - name: Build the Docker image vllm cpu
+        run: docker buildx build --file Dockerfile.cpu --tag vllm-cpu-env --shm-size=4g .
+
       - name: Configuration of docker images, network and namespace for the kind cluster
         run: |
-          docker pull adsai/vllm-cpu-env:latest
           docker pull amazon/aws-cli:2.6.4
           kind load docker-image  amazon/aws-cli:2.6.4 --name chart-testing
-          kind load docker-image adsai/vllm-cpu-env:latest --name chart-testing
+          kind load docker-image vllm-cpu-env:latest --name chart-testing
           docker network connect vllm-net "$(docker ps -aqf "name=chart-testing-control-plane")"
           kubectl create ns ns-vllm
 
@@ -62,7 +64,7 @@ jobs:
         run: |
           export AWS_ACCESS_KEY_ID=minioadmin
           export AWS_SECRET_ACCESS_KEY=minioadmin
-          helm install --wait --wait-for-jobs --timeout 10m0s --debug --create-namespace --namespace=ns-vllm test-vllm examples/chart-helm -f examples/chart-helm/values.yaml --set secrets.s3endpoint=http://minio:9000 --set secrets.s3bucketname=testbucket --set secrets.s3accesskeyid=$AWS_ACCESS_KEY_ID --set secrets.s3accesskey=$AWS_SECRET_ACCESS_KEY --set image.env[0].name=VLLM_CPU_KVCACHE_SPACE --set resources.requests.cpu=1 --set resources.requests.memory=4Gi --set resources.limits.cpu=2 --set resources.limits.memory=5Gi --set-string image.env[0].value="1" --set-string extraInit.s3modelpath="opt-125m/" --set-string 'resources.limits.nvidia\.com/gpu=0' --set-string 'resources.requests.nvidia\.com/gpu=0' --set-string image.repository="adsai/vllm-cpu-env"
+          helm install --wait --wait-for-jobs --timeout 10m0s --debug --create-namespace --namespace=ns-vllm test-vllm examples/chart-helm -f examples/chart-helm/values.yaml --set secrets.s3endpoint=http://minio:9000 --set secrets.s3bucketname=testbucket --set secrets.s3accesskeyid=$AWS_ACCESS_KEY_ID --set secrets.s3accesskey=$AWS_SECRET_ACCESS_KEY --set image.env[0].name=VLLM_CPU_KVCACHE_SPACE --set resources.requests.cpu=1 --set resources.requests.memory=4Gi --set resources.limits.cpu=2 --set resources.limits.memory=5Gi --set-string image.env[0].value="1" --set-string extraInit.s3modelpath="opt-125m/" --set-string 'resources.limits.nvidia\.com/gpu=0' --set-string 'resources.requests.nvidia\.com/gpu=0' --set-string image.repository="vllm-cpu-env"
     
       - name: curl test
         run: |