biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Neuron] Remove bypass on EAGLEConfig and add a test (#18514)

Browse Source 
Signed-off-by: Elaine Zhao <elaineyz@amazon.com>
This commit is contained in:
aws-elaineyz
2025-05-22 21:26:32 -07:00
committed by GitHub
parent 583507d130
commit ed5d408255
4 changed files with 95 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
.buildkite/scripts/hardware_ci/run-neuron-test.sh
View File

@@ -53,4 +53,11 @@ docker run --rm -it --device=/dev/neuron0 --network bridge \
-e "NEURON_COMPILE_CACHE_URL=${NEURON_COMPILE_CACHE_MOUNT}" \
--name "${container_name}" \
${image_name} \
/bin/bash -c "python3 /workspace/vllm/examples/offline_inference/neuron.py && python3 -m pytest /workspace/vllm/tests/neuron/1_core/ -v --capture=tee-sys && python3 -m pytest /workspace/vllm/tests/neuron/2_core/ -v --capture=tee-sys"
/bin/bash -c "
python3 /workspace/vllm/examples/offline_inference/neuron.py;
python3 -m pytest /workspace/vllm/tests/neuron/1_core/ -v --capture=tee-sys;
for f in /workspace/vllm/tests/neuron/2_core/*.py; do
echo 'Running test file: '$f;
python3 -m pytest \$f -v --capture=tee-sys;
done
"