@@ -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} )
4848endmacro ()
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
5173add_library (benchmark benchmark/benchmark.c)
5274
@@ -107,6 +129,10 @@ add_spatz_test_oneParam(dp-faxpy dp-faxpy/main.c 1024)
107129add_spatz_test_oneParam(dp-fdotp dp-fdotp/main.c 128)
108130add_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+
110136add_spatz_test_threeParam(dp-fconv2d dp-fconv2d/main.c 32 32 7)
111137add_spatz_test_threeParam(dp-fconv2d dp-fconv2d/main.c 64 64 7)
112138
0 commit comments