Skip to content

Commit

Permalink
Fix CLBlast builds in CMake release mode
Browse files Browse the repository at this point in the history
Somehow the path handling is different here which requires me to
be quite a bit more annoying about it.
  • Loading branch information
LoganDark committed Jul 3, 2023
1 parent 0b91fe1 commit db47512
Showing 1 changed file with 7 additions and 6 deletions.
13 changes: 7 additions & 6 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -129,8 +129,8 @@ endif()
if (RWKV_CLBLAST)
cmake_minimum_required(VERSION 3.17)

file(GLOB OPENCL_INCLUDE_SEARCH_PATHS OpenCL-*/)
file(GLOB CLBLAST_INCLUDE_SEARCH_PATHS CLBlast-*/)
file(GLOB OPENCL_INCLUDE_SEARCH_PATHS ${CMAKE_SOURCE_DIR}/OpenCL-*/)
file(GLOB CLBLAST_INCLUDE_SEARCH_PATHS ${CMAKE_SOURCE_DIR}/CLBlast-*/)

set(OPENCL_INCLUDE_SEARCH_PATHS
/usr/include
Expand All @@ -148,12 +148,13 @@ if (RWKV_CLBLAST)
${CLBLAST_INCLUDE_SEARCH_PATHS}
)

find_path(OPENCL_INC NAMES opencl.h PATHS ${OPENCL_INCLUDE_SEARCH_PATHS})
find_library(OPENCL_LIB NAMES OpenCL)
find_path(CLBLAST_INC NAMES clblast.h PATHS ${CLBLAST_INCLUDE_SEARCH_PATHS})
find_library(CLBLAST_LIB NAMES clblast)
find_path(OPENCL_INC NAMES opencl.h PATHS ${OPENCL_INCLUDE_SEARCH_PATHS} PATH_SUFFIXES include/CL)
find_library(OPENCL_LIB NAMES OpenCL PATHS ${OPENCL_INCLUDE_SEARCH_PATHS} PATH_SUFFIXES lib)
find_path(CLBLAST_INC NAMES clblast.h PATHS ${CLBLAST_INCLUDE_SEARCH_PATHS} PATH_SUFFIXES include)
find_library(CLBLAST_LIB NAMES clblast PATHS ${CLBLAST_INCLUDE_SEARCH_PATHS} PATH_SUFFIXES lib)

if (OPENCL_LIB)
set(OPENCL_INC ${OPENCL_INC}/..) # disgusting
message(STATUS "OpenCL SDK found: ${OPENCL_INC}")

if (CLBLAST_LIB)
Expand Down

0 comments on commit db47512

Please sign in to comment.