Add BFLOAT16 and HFLOAT16 tests

Author: martin-frbg

test/CMakeLists.txt

@@ -36,6 +36,28 @@ foreach(test_bin ${OpenBLAS_Tests})
 target_link_libraries(${test_bin} ${OpenBLAS_LIBNAME})
 endforeach()
 
+if (BUILD_BFLOAT16)
+	add_executable(test_bgemm compare_sgemm_bgemm.c)
+	target_compile_definitions(test_bgemm PUBLIC -DIBFLOAT16 -DOBFLOAT16)
+	target_link_libraries(test_bgemm ${OpenBLAS_LIBNAME})
+	add_executable(test_bgemv compare_sgemv_bgemv.c)
+	target_compile_definitions(test_bgemv PUBLIC -DIBFLOAT16 -DOBFLOAT16)
+	target_link_libraries(test_bgemv ${OpenBLAS_LIBNAME})
+	add_executable(test_sbgemm compare_sgemm_sbgemm.c)
+	target_compile_definitions(test_sbgemm PUBLIC -DIBFLOAT16)
+	target_link_libraries(test_sbgemm ${OpenBLAS_LIBNAME})
+	add_executable(test_sbgemv compare_sgemv_sbgemv.c)
+	target_compile_definitions(test_sbgemv PUBLIC -DIBFLOAT16)
+	target_link_libraries(test_sbgemv ${OpenBLAS_LIBNAME})
+endif()
+
+if (BUILD_HFLOAT16)
+	add_executable(test_shgemm compare_sgemm_shgemm.c)
+	target_link_libraries(test_shgemm ${OpenBLAS_LIBNAME})
+	add_executable(test_shgemv compare_sgemv_shgemv.c)
+	target_link_libraries(test_shgemv ${OpenBLAS_LIBNAME})
+endif()
+
 # $1 exec, $2 input, $3 output_result
 if(WIN32)
 FILE(WRITE ${CMAKE_CURRENT_BINARY_DIR}/test_helper.ps1
@@ -94,3 +116,21 @@ add_test(NAME "${float_type}blas3_3m"
 endif()
 endif()
 endforeach()
+
+if (BUILD_BFLOAT16)
+	add_test(NAME "bgemm" 
+	COMMAND $<TARGET_FILE:test_bgemm>)
+	add_test(NAME "bgemv" 
+	COMMAND $<TARGET_FILE:test_bgemv>)
+	add_test(NAME "sbgemm" 
+	COMMAND $<TARGET_FILE:test_sbgemm>)
+	add_test(NAME "sbgemv" 
+	COMMAND $<TARGET_FILE:test_sbgemv>)
+endif()
+
+if (BUILD_HFLOAT16)
+	add_test(NAME "shgemm" 
+	COMMAND $<TARGET_FILE:test_shgemm>)
+	add_test(NAME "shgemv" 
+	COMMAND $<TARGET_FILE:test_shgemv>)
+endif()