Commit efae2614 authored by Jakub Klinkovský's avatar Jakub Klinkovský
Browse files

Automatic detection of CUBLAS

parent f95969ac

CMakeLists.txt

+13 −6
Original line number Diff line number Diff line
@@ -118,11 +118,19 @@ if( WITH_CUDA STREQUAL "yes" ) 
        ####

        # Check for cuBLAS

        #

        if( WITH_CUBLAS STREQUAL "yes" ) 

            message( "Enabling CUBLAS." )

            set( HAVE_CUBLAS TRUE)

        if( NOT WITH_CUBLAS STREQUAL "no" )

            find_path( CUBLAS_INCLUDE_DIR cublas_v2.h

                       /usr/local/cuda/include

                       ${CUDA_INCLUDE_DIR}

                       DOC "CUBLAS headers." )

            if( ${CUBLAS_INCLUDE_DIR} STREQUAL "CUBLAS_INCLUDE_DIR-NOTFOUND" )

                message( "CUBLAS not found." )

                set( HAVE_CUBLAS "//#define HAVE_CUBLAS 1" )

            else()

                message( "CUBLAS found. -- ${CUBLAS_INCLUDE_DIR}" )

                set( HAVE_CUBLAS "#define HAVE_CUBLAS 1" )

        endif( WITH_CUBLAS STREQUAL "yes" )       

            endif()

        endif( NOT WITH_CUBLAS STREQUAL "no" )



        ####

        # Check for CUSP
@@ -142,7 +150,6 @@ if( WITH_CUDA STREQUAL "yes" ) 
        # Check for CUSPARSE

        #

        if( NOT WITH_CUSPARSE STREQUAL "no" )

        

           find_path( CUSPARSE_INCLUDE_DIR cusparse.h

                      /usr/local/cuda/include                   

                      ${CUDA_INCLUDE_DIR}

build

+0 −3
Original line number Diff line number Diff line
@@ -6,7 +6,6 @@ WITH_CUDA="yes" 
WITH_TESTS="yes"



WITH_CUDA_ARCH="auto"

WITH_CUBLAS="no"

WITH_TEMPLATE_INSTANTIATION="yes"

INSTANTIATE_LONG_INT="no"

INSTANTIATE_INT="yes"
@@ -28,7 +27,6 @@ do 
        --build=*                        ) BUILD="${option#*=}" ;;

        --with-tests=*                   ) WITH_TESTS="${option#*=}" ;;

        --with-cuda=*                    ) WITH_CUDA="${option#*=}" ;;

        --with-cublas=*                  ) WITH_CUBLAS="${option#*=}" ;;

        --with-cuda-arch=*               ) WITH_CUDA_ARCH="${option#*=}";;

        --with-templates-instantiation=* ) WITH_TEMPLATE_INSTANTIATION="${option#*=}" ;;

        --instantiate-long-int=*         ) INSTANTIATE_LONG_INT="${option#*=}" ;;
@@ -82,7 +80,6 @@ ${CMAKE} ${ROOT_DIR} \ 
         -DCMAKE_INSTALL_PREFIX=${PREFIX} \

         -DWITH_CUDA=${WITH_CUDA} \

         -DWITH_CUDA_ARCH=${WITH_CUDA_ARCH} \

         -DWITH_CUBLAS=${WITH_CUBLAS} \

         -DWITH_TESTS=${WITH_TESTS} \

         -DPETSC_DIR=${PETSC_DIR} \

         -DDCMTK_DIR=${DCMTK_DIR} \

tests/benchmarks/CMakeLists.txt

+1 −1
Original line number Diff line number Diff line
@@ -3,7 +3,7 @@ ADD_SUBDIRECTORY( heat-equation-benchmark ) 


IF( BUILD_CUDA )

    CUDA_ADD_EXECUTABLE( tnl-cuda-benchmarks${debugExt} tnl-cuda-benchmarks.cu )

    if( WITH_CUBLAS STREQUAL "yes" )

    if( HAVE_CUBLAS STREQUAL "yes" )

        CUDA_ADD_CUBLAS_TO_TARGET( tnl-cuda-benchmarks${debugExt} )

    endif()

    TARGET_LINK_LIBRARIES( tnl-cuda-benchmarks${debugExt} tnl${debugExt}-${tnlVersion} ${CUSPARSE_LIBRARY} )