Added CUDADIR to scaning paths

boostorg · Jun 24, 2015 · 83273bc · 83273bc
1 parent e369b8d
commit 83273bc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/cmake/FindOpenCL.cmake b/cmake/FindOpenCL.cmake
@@ -59,11 +59,11 @@ ELSE (APPLE)
 	# The AMD SDK currently does not place its headers
 	# in /usr/include, therefore also search relative
 	# to the library
-	FIND_PATH(OPENCL_INCLUDE_DIRS CL/cl.h PATHS ${_OPENCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
-	FIND_PATH(_OPENCL_CPP_INCLUDE_DIRS CL/cl.hpp PATHS ${_OPENCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
+	FIND_PATH(OPENCL_INCLUDE_DIRS CL/cl.h PATHS "$ENV{CUDADIR}/include" ${_OPENCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
+	FIND_PATH(_OPENCL_CPP_INCLUDE_DIRS CL/cl.hpp PATHS "$ENV{CUDADIR}/include" ${_OPENCL_INC_CAND} "/usr/local/cuda/include" "/opt/cuda/include" "/opt/AMDAPP/include" ENV OpenCL_INCPATH)
 	# for Mali GPUs, libmali must be linked 
 	FIND_LIBRARY(OPENCL_LIBRARIES_MALI mali
-		PATHS ENV LD_LIBRARY_PATH ENV OpenCL_LIBPATH "/usr/lib64"
+		PATHS ENV LD_LIBRARY_PATH ENV OpenCL_LIBPATH "/usr/lib64" "$ENV{CUDADIR}/lib64"
 	)
 	IF (OPENCL_LIBRARIES_MALI)
 		SET(OPENCL_LIBRARIES ${OPENCL_LIBRARIES} ${OPENCL_LIBRARIES_MALI})