You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
44 lines
2.1 KiB
44 lines
2.1 KiB
|
|
|
|
if((NOT TENSOR_INCLUDE_DIR) AND Eigen_SOURCE_DIR)
|
|
# unless TENSOR_INCLUDE_DIR is defined, let's use current Eigen version
|
|
set(TENSOR_INCLUDE_DIR ${Eigen_SOURCE_DIR})
|
|
set(TENSOR_FOUND TRUE)
|
|
else()
|
|
find_package(Tensor)
|
|
endif()
|
|
|
|
if (TENSOR_FOUND)
|
|
|
|
include_directories(${TENSOR_INCLUDE_DIR})
|
|
btl_add_bench(btl_tensor_linear main_linear.cpp)
|
|
btl_add_bench(btl_tensor_vecmat main_vecmat.cpp)
|
|
btl_add_bench(btl_tensor_matmat main_matmat.cpp)
|
|
|
|
btl_add_target_property(btl_tensor_linear COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor")
|
|
btl_add_target_property(btl_tensor_vecmat COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor")
|
|
btl_add_target_property(btl_tensor_matmat COMPILE_FLAGS "-fno-exceptions -DBTL_PREFIX=tensor")
|
|
|
|
option(BTL_BENCH_NOGCCVEC "also bench Eigen explicit vec without GCC's auto vec" OFF)
|
|
if(CMAKE_COMPILER_IS_GNUCXX AND BTL_BENCH_NOGCCVEC)
|
|
btl_add_bench(btl_tensor_nogccvec_linear main_linear.cpp)
|
|
btl_add_bench(btl_tensor_nogccvec_vecmat main_vecmat.cpp)
|
|
btl_add_bench(btl_tensor_nogccvec_matmat main_matmat.cpp)
|
|
|
|
btl_add_target_property(btl_tensor_nogccvec_linear COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec")
|
|
btl_add_target_property(btl_tensor_nogccvec_vecmat COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec")
|
|
btl_add_target_property(btl_tensor_nogccvec_matmat COMPILE_FLAGS "-fno-exceptions -fno-tree-vectorize -DBTL_PREFIX=tensor_nogccvec")
|
|
endif()
|
|
|
|
|
|
if(NOT BTL_NOVEC)
|
|
btl_add_bench(btl_tensor_novec_linear main_linear.cpp OFF)
|
|
btl_add_bench(btl_tensor_novec_vecmat main_vecmat.cpp OFF)
|
|
btl_add_bench(btl_tensor_novec_matmat main_matmat.cpp OFF)
|
|
btl_add_target_property(btl_tensor_novec_linear COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec")
|
|
btl_add_target_property(btl_tensor_novec_vecmat COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec")
|
|
btl_add_target_property(btl_tensor_novec_matmat COMPILE_FLAGS "-fno-exceptions -DEIGEN_DONT_VECTORIZE -DBTL_PREFIX=tensor_novec")
|
|
|
|
endif()
|
|
|
|
endif ()
|
|
|