forked from jiuyuan/InfiniTensor
update CMakeList.txt Makefile for micro benchmark
This commit is contained in:
parent
48322dbf27
commit
454b7651a8
|
@ -262,3 +262,19 @@ if(BUILD_TEST)
|
|||
target_link_libraries(nnet_reader InfiniTensor)
|
||||
endif()
|
||||
endif()
|
||||
|
||||
function(build_bench files)
|
||||
file(GLOB BENCH_SOURCES ${files})
|
||||
foreach(benchsourcefile ${BENCH_SOURCES})
|
||||
get_filename_component(benchname ${benchsourcefile} NAME_WE)
|
||||
add_executable("benchmark_${benchname}" ${benchsourcefile})
|
||||
target_link_libraries("benchmark_${benchname}" InfiniTensor)
|
||||
# add_custom_target(NAME ${benchname} COMMAND ${benchname})
|
||||
endforeach(benchsourcefile ${BENCH_SOURCES})
|
||||
endfunction()
|
||||
|
||||
if (BENCH)
|
||||
if (USE_CUDA)
|
||||
build_bench(benchmark/kernels/cuda/*.cc)
|
||||
endif()
|
||||
endif()
|
||||
|
|
4
Makefile
4
Makefile
|
@ -5,13 +5,15 @@ CUDA ?= OFF
|
|||
BANG ?= OFF
|
||||
INTELCPU ?= off
|
||||
BACKTRACE ?= ON
|
||||
TEST ?= ON
|
||||
TEST ?= OFF
|
||||
BENCH ?= ON
|
||||
|
||||
CMAKE_OPT = -DCMAKE_BUILD_TYPE=$(TYPE)
|
||||
CMAKE_OPT += -DUSE_CUDA=$(CUDA)
|
||||
CMAKE_OPT += -DUSE_BANG=$(BANG)
|
||||
CMAKE_OPT += -DUSE_BACKTRACE=$(BACKTRACE)
|
||||
CMAKE_OPT += -DBUILD_TEST=$(TEST)
|
||||
CMAKE_OPT += -BENCH=$(BENCH)
|
||||
|
||||
ifeq ($(INTELCPU), ON)
|
||||
CMAKE_OPT += -DUSE_INTELCPU=ON -DCMAKE_CXX_COMPILER=dpcpp
|
||||
|
|
Loading…
Reference in New Issue