make bfloat16 functions conditional on BUILD_HALF

This commit is contained in:
Martin Kroeker 2020-04-29 18:31:24 +02:00 committed by GitHub
parent 1377810961
commit 33059ad1de
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -53,6 +53,7 @@ gotoblas_t TABLE_NAME = {
GEMM_DEFAULT_OFFSET_A, GEMM_DEFAULT_OFFSET_B, GEMM_DEFAULT_ALIGN,
#ifdef BUILD_HALF
0, 0, 0,
SHGEMM_DEFAULT_UNROLL_M, SHGEMM_DEFAULT_UNROLL_N,
#ifdef SHGEMM_DEFAULT_UNROLL_MN
@ -109,7 +110,7 @@ gotoblas_t TABLE_NAME = {
#else
NULL,NULL,
#endif
#endif
0, 0, 0,
SGEMM_DEFAULT_UNROLL_M, SGEMM_DEFAULT_UNROLL_N,