triton/bench/CMakeLists.txt

set(CMAKE_BUILD_TYPE Release)

set(BLAS_DEF)
set(BLAS_LIBS atidlas ${OPENCL_LIBRARIES})

#CUBLAS
find_package(CUDA)
if(CUDA_FOUND)
    set(BLAS_DEF ${BLAS_DEF} "-DBENCH_CUBLAS")
endif()

#CLAMDBLAS
find_package(CLAMDBLAS)
if(CLAMDBLAS_FOUND)
    set(BLAS_DEF ${BLAS_DEF} "-DBENCH_CLAMDBLAS")
    include_directories(${CLAMDBLAS_INCLUDE_DIR})
    set(BLAS_LIBS ${BLAS_LIBS}  ${CLAMDBLAS_LIBRARIES} )
endif()

#CBLAS
find_package(MKL)
if(MKL_FOUND)
    set(BLAS_DEF ${BLAS_DEF} "-DBENCH_CBLAS")
    include_directories(${MKL_INCLUDE_DIR})
    set(BLAS_LIBS ${BLAS_LIBS}  ${MKL_LIBRARIES} )
endif()

string(REPLACE ";" " " BLAS_DEF_STR "${BLAS_DEF}")

foreach(PROG blas overhead)
    include_directories(${CMAKE_CURRENT_SOURCE_DIR})
    if(CUDA_FOUND)
        set(CUDA_NVCC_FLAGS ${CUDA_NVCC_FLAGS} " ${BLAS_DEF_STR}")
        set(CUPROG ${CMAKE_CURRENT_BINARY_DIR}/${PROG}.cu)
        file(COPY ${PROG}.cpp DESTINATION ${CMAKE_CURRENT_BINARY_DIR})
	file(RENAME ${CMAKE_CURRENT_BINARY_DIR}/${PROG}.cpp ${CUPROG})
        cuda_add_executable(${PROG}-bench  ${CUPROG})
	cuda_add_cublas_to_target(${PROG}-bench)
    else()
        add_executable(${PROG}-bench  ${PROG}.cpp)
        set_target_properties(${PROG}-bench PROPERTIES COMPILE_FLAGS "-Wall -Wextra ${BLAS_DEF_STR} -std=c++11")
    endif()
     target_link_libraries(${PROG}-bench ${BLAS_LIBS})
endforeach(PROG)