文本文件  |  45行  |  2.1 KB



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(NOT BTL_NOVEC)

endif (TENSOR_FOUND)