Skip to content

Commit e3e4715

Browse files
sw: add gemv kernel
1 parent b4d8094 commit e3e4715

File tree

10 files changed

+58793
-0
lines changed

10 files changed

+58793
-0
lines changed

sw/spatzBenchmarks/CMakeLists.txt

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -47,6 +47,28 @@ macro(add_spatz_test_threeParam name file param1 param2 param3)
4747
target_compile_definitions(test-${SNITCH_TEST_PREFIX}${target_name} PUBLIC DATAHEADER="data/data_${param1}_${param2}_${param3}.h" SNRT_NFPU_PER_CORE=${SNRT_NFPU_PER_CORE})
4848
endmacro()
4949

50+
# Use data type as another parameter
51+
macro(add_spatz_test_oneParam_type name file param1 type)
52+
set(target_name ${name}_M${param1})
53+
add_snitch_test(${target_name} ${file})
54+
target_link_libraries(test-${SNITCH_TEST_PREFIX}${target_name} benchmark ${SNITCH_RUNTIME})
55+
target_compile_definitions(test-${SNITCH_TEST_PREFIX}${target_name} PUBLIC DATAHEADER="data/data_${param1}_${type}.h" PREC=${type} SNRT_NFPU_PER_CORE=${SNRT_NFPU_PER_CORE})
56+
endmacro()
57+
58+
macro(add_spatz_test_twoParam_type name file param1 param2 type)
59+
set(target_name ${name}_M${param1}_N${param2})
60+
add_snitch_test(${target_name} ${file})
61+
target_link_libraries(test-${SNITCH_TEST_PREFIX}${target_name} benchmark ${SNITCH_RUNTIME})
62+
target_compile_definitions(test-${SNITCH_TEST_PREFIX}${target_name} PUBLIC DATAHEADER="data/data_${param1}_${param2}_${type}.h" PREC=${type} SNRT_NFPU_PER_CORE=${SNRT_NFPU_PER_CORE})
63+
endmacro()
64+
65+
macro(add_spatz_test_threeParam_type name file param1 param2 param3 type)
66+
set(target_name ${name}_M${param1}_N${param2}_K${param3})
67+
add_snitch_test(${target_name} ${file})
68+
target_link_libraries(test-${SNITCH_TEST_PREFIX}${target_name} benchmark ${SNITCH_RUNTIME})
69+
target_compile_definitions(test-${SNITCH_TEST_PREFIX}${target_name} PUBLIC DATAHEADER="data/data_${param1}_${param2}_${param3}_${type}.h" PREC=${type} SNRT_NFPU_PER_CORE=${SNRT_NFPU_PER_CORE})
70+
endmacro()
71+
5072
# Benchmark library
5173
add_library(benchmark benchmark/benchmark.c)
5274

@@ -107,6 +129,10 @@ add_spatz_test_oneParam(dp-faxpy dp-faxpy/main.c 1024)
107129
add_spatz_test_oneParam(dp-fdotp dp-fdotp/main.c 128)
108130
add_spatz_test_oneParam(dp-fdotp dp-fdotp/main.c 4096)
109131

132+
add_spatz_test_twoParam_type(dp-gemv gemv/main.c 64 128 64)
133+
add_spatz_test_twoParam_type(sp-gemv gemv/main.c 128 128 32)
134+
add_spatz_test_twoParam_type(hp-gemv gemv/main.c 256 128 16)
135+
110136
add_spatz_test_threeParam(dp-fconv2d dp-fconv2d/main.c 32 32 7)
111137
add_spatz_test_threeParam(dp-fconv2d dp-fconv2d/main.c 64 64 7)
112138

0 commit comments

Comments
 (0)