blob: a58eed7864f4de8021f69d31eeb46e616d3676d3 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
|
cmake_minimum_required(VERSION 3.5 FATAL_ERROR)
find_package(MKL)
if (NOT ${MKL_FOUND})
find_package(BLAS)
endif()
#benchmarks
macro(add_benchmark BENCHNAME)
add_executable(${BENCHNAME} ${ARGN}
BenchUtils.cc ../test/QuantizationHelpers.cc)
set_target_properties(${BENCHNAME} PROPERTIES
CXX_STANDARD 11
CXX_EXTENSIONS NO)
target_compile_options(${BENCHNAME} PRIVATE
"-m64" "-mavx2" "-mfma" "-masm=intel")
target_link_libraries(${BENCHNAME} fbgemm)
add_dependencies(${BENCHNAME} fbgemm)
if(${MKL_FOUND})
message(STATUS "MKL_LIBRARIES= ${MKL_LIBRARIES}")
target_include_directories(${BENCHNAME} PRIVATE "${MKL_INCLUDE_DIR}")
target_link_libraries(${BENCHNAME} "${MKL_LIBRARIES}")
target_compile_options(${BENCHNAME} PRIVATE
"-DUSE_MKL")
endif()
if (${BLAS_FOUND})
message(STATUS "BLAS_LIBRARIES= ${BLAS_LIBRARIES}")
target_compile_options(${BENCHNAME} PRIVATE "-DUSE_BLAS")
target_link_libraries(${BENCHNAME} "${BLAS_LIBRARIES}")
endif()
set_target_properties(${BENCHNAME} PROPERTIES FOLDER test)
endmacro()
if(FBGEMM_BUILD_BENCHMARKS)
set(BENCHMARKS "")
file(GLOB BENCH_LIST "*Benchmark.cc")
foreach(BENCH_FILE ${BENCH_LIST})
get_filename_component(BENCH_NAME "${BENCH_FILE}" NAME_WE)
get_filename_component(BENCH_FILE_ONLY "${BENCH_FILE}" NAME)
add_benchmark("${BENCH_NAME}"
"${BENCH_FILE_ONLY}")
list(APPEND BENCHMARKS "${BENCH_NAME}")
endforeach()
add_custom_target(run_benchmarks
COMMAND ${BENCHMARKS})
add_dependencies(run_benchmarks
${BENCHMARKS})
endif()
|
