Changed GenerateObjects to append combination codes (e.g. dtrmm_TU).

diff --git a/cmake/utils.cmake b/cmake/utils.cmake
index f839245..b0c108b 100644 (file)
--- a/cmake/utils.cmake
+++ b/cmake/utils.cmake
@@ -15,6 +15,8 @@ endfunction ()
 # Returns all combinations of the input list, as a list with colon-separated combinations
 # E.g. input of A B C returns A B C A:B A:C B:C
 # N.B. The input is meant to be a list, and to past a list to a function in CMake you must quote it (e.g. AllCombinations("${LIST_VAR}")).
+# @returns LIST_OUT a list of combinations
+#          CODES_OUT a list of codes corresponding to each combination, with N meaning the item is not present, and the first letter of the list item meaning it is presen
 function(AllCombinations list_in)
   list(LENGTH list_in list_count)
   set(num_combos 1)
@@ -22,7 +24,10 @@ function(AllCombinations list_in)
   math(EXPR num_combos "(${num_combos} << ${list_count}) - 1")
   set(LIST_OUT "")
   foreach (c RANGE 0 ${num_combos})
+
     set(current_combo "")
+    set(current_code "")
+
     # this is a little ridiculous just to iterate through a list w/ indices
     math(EXPR last_list_index "${list_count} - 1")
     foreach (list_index RANGE 0 ${last_list_index})
@@ -35,57 +40,24 @@ function(AllCombinations list_in)
         else ()
           set(current_combo ${list_elem})
         endif ()
+        string(SUBSTRING ${list_elem} 0 1 code_char)
+      else ()
+        set(code_char "N")
       endif ()
+      set(current_code "${current_code}${code_char}")
     endforeach ()
-    list(APPEND LIST_OUT ${current_combo})
-  endforeach ()
-  list(APPEND LIST_OUT " ") # Empty set is a valic combination, but CMake isn't appending the empty string for some reason, use a space
-  set(LIST_OUT ${LIST_OUT} PARENT_SCOPE)
-endfunction ()
-
-# generates object files for each of the sources for each of the combinations of the preprocessor definitions passed in
-# @param sources_in the source files to build from
-# @param defines_in the preprocessor definitions that will be combined to create the object files
-# @param all_defines_in (optional) preprocessor definitions that will be applied to all objects
-function(GenerateObjects sources_in defines_in all_defines_in)
-  AllCombinations("${defines_in}")
-  set(define_combos ${LIST_OUT})
-  set(OBJ_LIST_OUT "")
-  foreach (source_file ${sources_in})
-    foreach (def_combo ${define_combos})
-
-      # replace colon separated list with semicolons, this turns it into a CMake list that we can use foreach with
-      string(REPLACE ":" ";" def_combo ${def_combo})
 
-      # build a unique variable name for this obj file by picking two letters from the defines (can't use one in this case)
-      set(obj_name "")
-      foreach (combo_elem ${def_combo})
-        string(REGEX MATCH "^[A-Z][A-Z]" letter ${combo_elem})
-        set(obj_name "${obj_name}${letter}")
-      endforeach ()
-
-      # parse file name
-      string(REGEX MATCH "^[a-zA-Z_0-9]+" source_name ${source_file})
-      string(TOUPPER ${source_name} source_name)
-
-      # prepend the uppercased file name to the obj name
-      set(obj_name "${source_name}_${obj_name}_OBJS")
+    if (current_combo STREQUAL "")
+      list(APPEND LIST_OUT " ") # Empty set is a valid combination, but CMake isn't appending the empty string for some reason, use a space
+    else ()
+      list(APPEND LIST_OUT ${current_combo})
+    endif ()
+    list(APPEND CODES_OUT ${current_code})
 
-      # now add the object and set the defines
-      add_library(${obj_name} OBJECT ${source_file})
-      set(cur_defines ${def_combo})
-      if ("${cur_defines}" STREQUAL " ")
-        set(cur_defines ${all_defines_in})
-      else ()
-        list(APPEND cur_defines ${all_defines_in})
-      endif ()
-      if (cur_defines AND NOT "${cur_defines}" STREQUAL " ") # using space as the empty set
-        set_target_properties(${obj_name} PROPERTIES COMPILE_DEFINITIONS "${cur_defines}")
-      endif ()
-      list(APPEND OBJ_LIST_OUT ${obj_name})
-    endforeach ()
   endforeach ()
-  set(OBJ_LIST_OUT ${OBJ_LIST_OUT} PARENT_SCOPE)
+
+  set(LIST_OUT ${LIST_OUT} PARENT_SCOPE)
+  set(CODES_OUT ${CODES_OUT} PARENT_SCOPE)
 endfunction ()
 
 # generates object files for each of the sources, using the BLAS naming scheme to pass the funciton name as a preprocessor definition
@@ -94,7 +66,10 @@ endfunction ()
 # @param defines_in (optional) preprocessor definitions that will be applied to all objects
 # @param name_in (optional) if this is set this name will be used instead of the filename. Use a * to indicate where the float character should go, if no star the character will be prepended.
 #                           e.g. with DOUBLE set, "i*max" will generate the name "idmax", and "max" will be "dmax"
-function(GenerateNamedObjects sources_in float_type_in defines_in name_in use_cblas)
+# @param replace_k_with replaces the "k" in the filename with this string (e.g. symm_k should be symm_TU)
+# @param append_with appends the filename with this string (e.g. trmm_R should be trmm_RTUU or some other combination of characters)
+function(GenerateNamedObjects sources_in float_type_in defines_in name_in use_cblas replace_k_with append_with)
+
   set(OBJ_LIST_OUT "")
   foreach (source_file ${sources_in})
 
@@ -113,6 +88,12 @@ function(GenerateNamedObjects sources_in float_type_in defines_in name_in use_cb
       endif ()
     endif ()
 
+    if (replace_k_with)
+      string(REGEX REPLACE "k$" ${replace_k_with} obj_name ${obj_name})
+    else ()
+      set(obj_name "${obj_name}${append_with}")
+    endif ()
+
     # now add the object and set the defines
     set(obj_defines ${defines_in})
 
@@ -133,3 +114,53 @@ function(GenerateNamedObjects sources_in float_type_in defines_in name_in use_cb
   endforeach ()
   set(OBJ_LIST_OUT ${OBJ_LIST_OUT} PARENT_SCOPE)
 endfunction ()
+
+# generates object files for each of the sources for each of the combinations of the preprocessor definitions passed in
+# @param sources_in the source files to build from
+# @param defines_in the preprocessor definitions that will be combined to create the object files
+# @param float_type_in the float type to define for this build (e.g. SINGLE/DOUBLE/etc)
+# @param replace_k Replace the "k" in the filename with the define combo letters (else it appends). E.g. symm_k with TRANS and UNIT defined will be symm_TU.
+# @param all_defines_in (optional) preprocessor definitions that will be applied to all objects
+function(GenerateCombinationObjects sources_in defines_in float_type_in all_defines_in replace_k)
+
+  AllCombinations("${defines_in}")
+  set(define_combos ${LIST_OUT})
+  set(define_codes ${CODES_OUT})
+
+  set(COMBO_OBJ_LIST_OUT "")
+  list(LENGTH define_combos num_combos)
+  math(EXPR num_combos "${num_combos} - 1")
+
+  foreach (c RANGE 0 ${num_combos})
+
+    list(GET define_combos ${c} define_combo)
+    list(GET define_codes ${c} define_code)
+
+    foreach (source_file ${sources_in})
+
+      # replace colon separated list with semicolons, this turns it into a CMake list that we can use foreach with
+      string(REPLACE ":" ";" define_combo ${define_combo})
+
+      # now add the object and set the defines
+      set(cur_defines ${define_combo})
+      if ("${cur_defines}" STREQUAL " ")
+        set(cur_defines ${all_defines_in})
+      else ()
+        list(APPEND cur_defines ${all_defines_in})
+      endif ()
+
+      set(replace_k_name "")
+      set(append_name "")
+      if (replace_k)
+        set(replace_k_name ${define_code})
+      else ()
+        set(append_name ${define_code})
+      endif ()
+
+      GenerateNamedObjects("${source_file}" "${float_type_in}" "${cur_defines}" "" 0 "${replace_k_name}" "${append_name}") 
+      list(APPEND COMBO_OBJ_LIST_OUT ${obj_name})
+    endforeach ()
+  endforeach ()
+
+  set(COMBO_OBJ_LIST_OUT ${COMBO_OBJ_LIST_OUT} PARENT_SCOPE)
+endfunction ()

diff --git a/driver/level2/CMakeLists.txt b/driver/level2/CMakeLists.txt
index 1fbf7c7..bf0a585 100644 (file)
--- a/driver/level2/CMakeLists.txt
+++ b/driver/level2/CMakeLists.txt
@@ -1,6 +1,8 @@
 
 include_directories(${CMAKE_SOURCE_DIR})
 
+# TODO: These all need NAME/CNAME set (see GenerateNamedObjects)
+
 # sources that need to be compiled twice, once with no flags and once with LOWER
 set(UL_SOURCES
   sbmv_k.c

diff --git a/driver/level3/CMakeLists.txt b/driver/level3/CMakeLists.txt
index 57865d1..4427c8e 100644 (file)
--- a/driver/level3/CMakeLists.txt
+++ b/driver/level3/CMakeLists.txt
@@ -25,19 +25,32 @@ endif ()
 # loop through gemm.c defines
 set(GEMM_DEFINES NN NT TN TT)
 foreach (GEMM_DEFINE ${GEMM_DEFINES})
-  add_library(GEMM_${GEMM_DEFINE}_OBJS OBJECT gemm.c)
-  set_target_properties(GEMM_${GEMM_DEFINE}_OBJS PROPERTIES COMPILE_DEFINITIONS ${GEMM_DEFINE})
-  list(APPEND DBLAS_OBJS GEMM_${GEMM_DEFINE}_OBJS)
+  string(TOLOWER ${GEMM_DEFINE} GEMM_DEFINE_LC)
+  GenerateNamedObjects("gemm.c" "DOUBLE" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "")
+  list(APPEND DBLAS_OBJS ${OBJ_LIST_OUT})
 endforeach ()
 
+AllCombinations("TRANS;UPPER;UNIT")
+set(define_combos ${LIST_OUT})
+foreach (def_combo ${define_combos})
+  # replace colon separated list with semicolons, this turns it into a CMake list that we can use foreach with
+  string(REPLACE ":" ";" def_combo ${def_combo})
 
-GenerateObjects("trmm_L.c;trmm_R.c;trsm_L.c;trsm_R.c" "TRANS;UPPER;UNIT" "")
+  # build a unique variable name for this obj file by picking two letters from the defines (can't use one in this case)
+  set(obj_name "")
+  foreach (combo_elem ${def_combo})
+    string(REGEX MATCH "^[A-Z][A-Z]" letter ${combo_elem})
+    set(obj_name "${obj_name}${letter}")
+  endforeach ()
+endforeach ()
+
+GenerateCombinationObjects("trmm_L.c;trmm_R.c;trsm_L.c;trsm_R.c" "TRANS;UPPER;UNIT" "DOUBLE" "" 0)
 list(APPEND DBLAS_OBJS ${OBJ_LIST_OUT})
-GenerateObjects("symm_k.c" "LOWER;RSIDE" "NN")
+GenerateCombinationObjects("symm_k.c" "LOWER;RSIDE" "DOUBLE" "NN" 1)
 list(APPEND DBLAS_OBJS ${OBJ_LIST_OUT})
-GenerateObjects("syrk_k.c;syr2k_k.c" "LOWER;TRANS" "")
+GenerateCombinationObjects("syrk_k.c;syr2k_k.c" "LOWER;TRANS" "DOUBLE" "" 1)
 list(APPEND DBLAS_OBJS ${OBJ_LIST_OUT})
-GenerateObjects("syrk_kernel.c;syr2k_kernel.c" "LOWER" "")
+GenerateCombinationObjects("syrk_kernel.c;syr2k_kernel.c" "LOWER" "DOUBLE" "" 1)
 list(APPEND DBLAS_OBJS ${OBJ_LIST_OUT})
 
 #if (SMP)