From 2ba80bed2732edf42b1014ea4e34757849fc93d0 Mon Sep 17 00:00:00 2001
From: Letian Li <lotianmail@gmail.com>
Date: Thu, 23 May 2024 17:08:58 +0100
Subject: [PATCH] [Bugfix] Update Dockerfile.cpu to fix NameError: name
 'vllm_ops' is not defined (#5009)

---
 .buildkite/run-cpu-test.sh | 2 +-
 Dockerfile.cpu             | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.buildkite/run-cpu-test.sh b/.buildkite/run-cpu-test.sh
index f187d1f18172..414045fe163e 100644
--- a/.buildkite/run-cpu-test.sh
+++ b/.buildkite/run-cpu-test.sh
@@ -11,4 +11,4 @@ trap remove_docker_container EXIT
 remove_docker_container
 
 # Run the image and launch offline inference
-docker run --network host --env VLLM_CPU_KVCACHE_SPACE=1 --name cpu-test cpu-test python3 examples/offline_inference.py
+docker run --network host --env VLLM_CPU_KVCACHE_SPACE=1 --name cpu-test cpu-test python3 vllm/examples/offline_inference.py
diff --git a/Dockerfile.cpu b/Dockerfile.cpu
index 4251fddd6cc3..aec79824213f 100644
--- a/Dockerfile.cpu
+++ b/Dockerfile.cpu
@@ -17,4 +17,6 @@ RUN pip install -v -r requirements-cpu.txt --extra-index-url https://download.py
 
 RUN VLLM_TARGET_DEVICE=cpu python3 setup.py install
 
+WORKDIR /workspace/
+
 CMD ["/bin/bash"]