Change "HALF" and "sh" to "BFLOAT16" and "sb"

2020-10-12 00:07:37 +02:00
parent 006c7f6671
commit e3a29f6b58
4 changed files with 41 additions and 36 deletions
@@ -16,8 +16,8 @@
 # HAVE_SSE2
 # HAVE_SSE3
 # MAKE
-# SHGEMM_UNROLL_M
-# SHGEMM_UNROLL_N
+# SBGEMM_UNROLL_M
+# SBGEMM_UNROLL_N
 # SGEMM_UNROLL_M
 # SGEMM_UNROLL_N
 # DGEMM_UNROLL_M
@@ -471,8 +471,8 @@ endif ()
    set(ZGEMM_UNROLL_N 2)
    set(SYMV_P 8)
  endif()
-  set(SHGEMM_UNROLL_M 8)
-  set(SHGEMM_UNROLL_N 4)
+  set(SBGEMM_UNROLL_M 8)
+  set(SBGEMM_UNROLL_N 4)

  # Or should this actually be NUM_CORES?
  if (${NUM_THREADS} GREATER 0)