diff options
| author | katsu560 <118887472+katsu560@users.noreply.github.com> | 2023-04-15 14:51:11 +0900 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-04-15 08:51:11 +0300 |
| commit | 106faaf2971d6c89d6010279a9a95737772470ef (patch) | |
| tree | d157500821819e0221dfddf4c9114f17cf6f2e78 | |
| parent | c85e03d12e4b8af22cb13aa9c618dcd5935862fd (diff) | |
cmake : add finding the OpenBLAS header file (#992)
| -rw-r--r-- | CMakeLists.txt | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/CMakeLists.txt b/CMakeLists.txt index d5715d9..5a20de3 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -120,6 +120,21 @@ if (LLAMA_OPENBLAS) add_compile_definitions(GGML_USE_OPENBLAS) add_link_options(${BLAS_LIBRARIES}) set(LLAMA_EXTRA_LIBS ${LLAMA_EXTRA_LIBS} openblas) + + # find header file + set(OPENBLAS_INCLUDE_SEARCH_PATHS + /usr/include + /usr/include/openblas + /usr/include/openblas-base + /usr/local/include + /usr/local/include/openblas + /usr/local/include/openblas-base + /opt/OpenBLAS/include + $ENV{OpenBLAS_HOME} + $ENV{OpenBLAS_HOME}/include + ) + find_path(OPENBLAS_INC NAMES cblas.h PATHS ${OPENBLAS_INCLUDE_SEARCH_PATHS}) + add_compile_options(-I${OPENBLAS_INC}) else() message(WARNING "OpenBLAS not found") endif() |