Skip to content

Commit

Permalink
fix handling new llm test models folder (#2926)
Browse files Browse the repository at this point in the history
  • Loading branch information
dtrawins authored Jan 2, 2025
1 parent 69fed4d commit d877288
Showing 1 changed file with 6 additions and 2 deletions.
8 changes: 6 additions & 2 deletions Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -670,10 +670,14 @@ cpu_extension:
mkdir -p ./lib/${OS}
docker cp $$(docker create --rm sample_cpu_extension:latest):/workspace/libcustom_relu_cpu_extension.so ./lib/${OS}

run_unit_tests:
prepare_models:
./prepare_llm_models.sh ${TEST_LLM_PATH}
ifeq ($(RUN_GPU_TESTS),1)
./prepare_gpu_models.sh ${GPU_MODEL_PATH} && \
./prepare_gpu_models.sh ${GPU_MODEL_PATH}
endif

run_unit_tests: prepare_models
ifeq ($(RUN_GPU_TESTS),1)
docker run \
--device=/dev/dri \
--group-add=$(shell stat -c "%g" /dev/dri/render* | head -n 1) \
Expand Down

0 comments on commit d877288

Please sign in to comment.