Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .github/workflows/cmake.yml
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ name: CMake

on:
push:
branches: [ main ]
branches: [ main, fix-mkl ]
pull_request:
branches: [ main ]

Expand Down
4 changes: 4 additions & 0 deletions .gitmodules
Original file line number Diff line number Diff line change
Expand Up @@ -14,3 +14,7 @@
path = IP-DiskANN
url = https://github.com/intellistream/IP-DiskANN.git
branch = main
[submodule "PLSH"]
path = PLSH
url = https://github.com/Kwan-Yiu/PLSH.git
branch = master
10 changes: 8 additions & 2 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -370,8 +370,14 @@ target_include_directories(CANDYBENCH PUBLIC "include")
#endif ()

# 设置 MKL 库的路径
set(MKL_INCLUDE_DIR "/opt/intel/oneapi/mkl/2025.2/include")
set(MKL_LIB_DIR "/opt/intel/oneapi/mkl/2025.2/lib/intel64")
# 优先使用环境变量 MKLROOT,如果未设置则使用默认路径
if(DEFINED ENV{MKLROOT})
set(MKL_INCLUDE_DIR "$ENV{MKLROOT}/include")
set(MKL_LIB_DIR "$ENV{MKLROOT}/lib/intel64")
else()
set(MKL_INCLUDE_DIR "/opt/intel/oneapi/mkl/latest/include")
set(MKL_LIB_DIR "/opt/intel/oneapi/mkl/latest/lib/intel64")
endif()

#set(MPI_INCLUDE_PATH "/usr/include/openmpi-x86_64")
#set(MPI_LIBRARIES "/usr/lib/x86_64-linux-gnu/openmpi/lib/libmpi.so")
Expand Down
7 changes: 7 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -51,6 +51,8 @@ RUN wget -qO - https://apt.repos.intel.com/intel-gpg-keys/GPG-PUB-KEY-INTEL-SW-P

ENV MKLROOT=/opt/intel/oneapi/mkl/latest
ENV LD_LIBRARY_PATH=${MKLROOT}/lib:${MKLROOT}/lib/intel64:${LD_LIBRARY_PATH}
ENV CPATH=${MKLROOT}/include:${CPATH}
ENV LIBRARY_PATH=${MKLROOT}/lib:${MKLROOT}/lib/intel64:${LIBRARY_PATH}


RUN pip install --no-cache-dir \
Expand Down Expand Up @@ -85,6 +87,11 @@ RUN mkdir -p build && cd build && \
cmake .. && \
make -j && make install

WORKDIR /app/PLSH
RUN mkdir -p build && cd build && \
cmake .. && \
make -j && make install

WORKDIR /app/big-ann-benchmarks
RUN pip install -r requirements_py3.10.txt && \
git clone https://github.com/Microsoft/DiskANN && \
Expand Down
1 change: 1 addition & 0 deletions PLSH
Submodule PLSH added at 0d033a
2 changes: 1 addition & 1 deletion big-ann-benchmarks