Skip to content

Commit

Permalink
FIX fixes from pr review and other small things
Browse files Browse the repository at this point in the history
  • Loading branch information
dantegd committed Dec 17, 2024
1 parent 836a299 commit 45eebf0
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 3 deletions.
2 changes: 1 addition & 1 deletion context/cuvs-bench/get_datasets.sh
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
set -eo pipefail

# find cuVS in the environment
PACKAGE_FILE_PATH=$(python -c "import cuvs-bench; print(package_name.__file__)")
PACKAGE_FILE_PATH=$(python -c "import cuvs_bench; print(package_name.__file__)")
PACKAGE_DIR=$(dirname "$PACKAGE_FILE_PATH")

# Apply the patch
Expand Down
5 changes: 4 additions & 1 deletion cuvs-bench/cpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,8 @@ mamba install -y -n base "python=${PYTHON_VER}"
mamba update --all -y -n base
mamba install -y -n base \
"cuvs-bench-cpu=${RAPIDS_VER}.*" \
"python=${PYTHON_VER}"
"python=${PYTHON_VER}" \
"requests"
conda clean -afy
chmod -R 777 /opt/conda
EOF
Expand All @@ -52,6 +53,8 @@ FROM bench-base AS cuvs-bench-cpu-datasets

SHELL ["/bin/bash", "-euo", "pipefail", "-c"]

COPY cuvs-bench/cuvs_bench_get_dataset.patch /home/rapids/cuvs-bench/cuvs_bench_get_dataset.patch

COPY cuvs-bench/get_datasets.sh /home/rapids/cuvs-bench/get_datasets.sh

COPY cuvs-bench/run_benchmark.sh /data/scripts/run_benchmark_preloaded_datasets.sh
Expand Down
5 changes: 4 additions & 1 deletion cuvs-bench/gpu/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,8 @@ RUN <<EOF
mamba update --all -y -n base
mamba install -y -n base \
"cuvs-bench=${RAPIDS_VER}.*" \
"cuda-version=${CUDA_VER%.*}.*"
"cuda-version=${CUDA_VER%.*}.*" \
"requests"
conda clean -afy
chmod -R 777 /opt/conda
EOF
Expand All @@ -52,6 +53,8 @@ FROM cuvs-bench AS cuvs-bench-datasets

SHELL ["/bin/bash", "-euo", "pipefail", "-c"]

COPY cuvs-bench/cuvs_bench_get_dataset.patch /home/rapids/cuvs-bench/cuvs_bench_get_dataset.patch

COPY cuvs-bench/get_datasets.sh /home/rapids/cuvs-bench/get_datasets.sh

COPY cuvs-bench/run_benchmarks_preloaded_datasets.sh /data/scripts/run_benchmarks_preloaded_datasets.sh
Expand Down

0 comments on commit 45eebf0

Please sign in to comment.