build vllm cpu ci image locally

Author: louie-tsai

.github/workflows/vllm-benchmark.yml

@@ -144,7 +144,7 @@ jobs:
           elif [[ "${DEVICE_NAME}" == "rocm" ]]; then
             DOCKER_IMAGE_PREFIX=docker.io/rocm/vllm-ci
           elif [[ "${DEVICE_NAME}" == "cpu" ]]; then
-            DOCKER_IMAGE_PREFIX=public.ecr.aws/q9t5s3a7/vllm-cpu-release-repo
+            DOCKER_IMAGE_PREFIX=vllm-cpu-test
           fi
           echo "DOCKER_IMAGE_PREFIX=$DOCKER_IMAGE_PREFIX" >> $GITHUB_ENV
 
@@ -181,6 +181,10 @@ jobs:
                 break
               fi
             done
+            if [[ "${DEVICE_NAME}" == "cpu" ]]; then
+              DOCKER_IMAGE: ${{ DOCKER_IMAGE_PREFIX }}:${{ HEAD_SHA }}
+              docker build --tag $DOCKER_IMAGE --target vllm-test -f docker/Dockerfile.cpu
+            fi
             popd
           fi