|
|
|
@@ -134,6 +134,20 @@ function (build_core TARGET_CORE KDIR TSUFFIX KERNEL_DEFINITIONS) |
|
|
|
set(USE_TRMM true) |
|
|
|
endif () |
|
|
|
|
|
|
|
set(USE_DIRECT_SGEMM false) |
|
|
|
if (X86_64) |
|
|
|
set(USE_DIRECT_SGEMM true) |
|
|
|
endif() |
|
|
|
|
|
|
|
if (USE_DIRECT_SGEMM) |
|
|
|
# if (NOT DEFINED SGEMMDIRECTKERNEL) |
|
|
|
set (SGEMMDIRECTKERNEL sgemm_direct_skylakex.c) |
|
|
|
set (SGEMMDIRECTPERFORMANT sgemm_direct_performant.c) |
|
|
|
# endif() |
|
|
|
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTKERNEL}" "" "gemm_direct" false "" "" false SINGLE) |
|
|
|
GenerateNamedObjects("${KERNELDIR}/${SGEMMDIRECTPERFORMANT}" "" "gemm_direct_performant" false "" "" false SINGLE) |
|
|
|
endif() |
|
|
|
|
|
|
|
foreach (float_type SINGLE DOUBLE HALF) |
|
|
|
string(SUBSTRING ${float_type} 0 1 float_char) |
|
|
|
if (${float_type} STREQUAL "HALF") |
|
|
|
|