You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

CMakeLists.txt 8.7 kB

4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171
  1. include_directories(${PROJECT_SOURCE_DIR})
  2. include_directories(${PROJECT_BINARY_DIR})
  3. # N.B. In the original makefile there was a BLOCKS define used in the compilation of these files but I don't see any evidence of it being set anywhere. -hpa
  4. # loop through gemm.c defines
  5. set(GEMM_DEFINES NN NT TN TT)
  6. set(GEMM_COMPLEX_DEFINES RN CN RT CT NR TR RR CR NC TC RC CC)
  7. foreach (GEMM_DEFINE ${GEMM_DEFINES})
  8. string(TOLOWER ${GEMM_DEFINE} GEMM_DEFINE_LC)
  9. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0)
  10. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  11. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0)
  12. endif ()
  13. if (BUILD_BFLOAT16)
  14. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
  15. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  16. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0 "" "" false "BFLOAT16")
  17. endif ()
  18. endif ()
  19. endforeach ()
  20. if ( BUILD_COMPLEX16 AND NOT BUILD_DOUBLE)
  21. foreach (GEMM_DEFINE ${GEMM_DEFINES})
  22. string(TOLOWER ${GEMM_DEFINE} GEMM_DEFINE_LC)
  23. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "DOUBLE")
  24. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  25. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0 "" "" false "DOUBLE")
  26. endif()
  27. endforeach()
  28. endif()
  29. if ( BUILD_COMPLEX AND NOT BUILD_SINGLE)
  30. foreach (GEMM_DEFINE ${GEMM_DEFINES})
  31. string(TOLOWER ${GEMM_DEFINE} GEMM_DEFINE_LC)
  32. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE}" "gemm_${GEMM_DEFINE_LC}" 0 "" "" false "SINGLE")
  33. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  34. GenerateNamedObjects("gemm.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm_thread_${GEMM_DEFINE_LC}" 0 "" "" false "SINGLE")
  35. endif()
  36. endforeach()
  37. endif()
  38. set(TRMM_TRSM_SOURCES
  39. trmm_L.c
  40. trmm_R.c
  41. trsm_L.c
  42. trsm_R.c)
  43. foreach(trmm_trsm_source ${TRMM_TRSM_SOURCES})
  44. string(REGEX MATCH "[a-z]+_[A-Z]+" op_name ${trmm_trsm_source})
  45. GenerateCombinationObjects("${trmm_trsm_source}" "UPPER;UNIT" "L;N" "" 0 "${op_name}N")
  46. GenerateCombinationObjects("${trmm_trsm_source}" "UPPER;UNIT" "L;N" "TRANSA" 0 "${op_name}T")
  47. endforeach()
  48. GenerateCombinationObjects("symm_k.c" "RSIDE;LOWER" "L;U" "NN" 1)
  49. GenerateCombinationObjects("syrk_k.c" "LOWER;TRANS" "U;N" "" 1)
  50. GenerateCombinationObjects("syr2k_k.c" "LOWER;TRANS" "U;N" "" 1)
  51. GenerateCombinationObjects("syrk_kernel.c" "LOWER" "U" "" 2)
  52. GenerateCombinationObjects("syr2k_kernel.c" "LOWER" "U" "" 2)
  53. if (USE_THREAD)
  54. # N.B. these do NOT have a float type (e.g. DOUBLE) defined!
  55. GenerateNamedObjects("gemm_thread_m.c;gemm_thread_n.c;gemm_thread_mn.c;gemm_thread_variable.c;syrk_thread.c" "" "" 0 "" "" 1)
  56. if (NOT USE_SIMPLE_THREADED_LEVEL3)
  57. GenerateCombinationObjects("syrk_k.c" "LOWER;TRANS" "U;N" "THREADED_LEVEL3" 2 "syrk_thread")
  58. GenerateCombinationObjects("symm_k.c" "RSIDE;LOWER" "L;U" "THREADED_LEVEL3;NN" 2 "symm_thread")
  59. endif ()
  60. endif ()
  61. foreach (float_type ${FLOAT_TYPES})
  62. if (${float_type} STREQUAL "COMPLEX" OR ${float_type} STREQUAL "ZCOMPLEX")
  63. GenerateCombinationObjects("zherk_kernel.c" "LOWER;CONJ" "U;N" "HERK" 2 "herk_kernel" false ${float_type})
  64. # TRANS needs to be set/unset when CONJ is set/unset, so can't use it as a combination
  65. GenerateCombinationObjects("zherk_k.c" "LOWER" "U" "HERK" 3 "herk_N" false ${float_type})
  66. GenerateCombinationObjects("zherk_k.c" "LOWER" "U" "HERK;TRANS;CONJ" 3 "herk_C" false ${float_type})
  67. # Need to set CONJ for trmm and trsm
  68. GenerateCombinationObjects("trmm_L.c" "UPPER;UNIT" "L;N" "CONJ" 0 "trmm_LR" false ${float_type})
  69. GenerateCombinationObjects("trmm_L.c" "UPPER;UNIT" "L;N" "TRANSA;CONJ" 0 "trmm_LC" false ${float_type})
  70. GenerateCombinationObjects("trmm_R.c" "UPPER;UNIT" "L;N" "CONJ" 0 "trmm_RR" false ${float_type})
  71. GenerateCombinationObjects("trmm_R.c" "UPPER;UNIT" "L;N" "TRANSA;CONJ" 0 "trmm_RC" false ${float_type})
  72. GenerateCombinationObjects("trsm_L.c" "UPPER;UNIT" "L;N" "CONJ" 0 "trsm_LR" false ${float_type})
  73. GenerateCombinationObjects("trsm_L.c" "UPPER;UNIT" "L;N" "TRANSA;CONJ" 0 "trsm_LC" false ${float_type})
  74. GenerateCombinationObjects("trsm_R.c" "UPPER;UNIT" "L;N" "CONJ" 0 "trsm_RR" false ${float_type})
  75. GenerateCombinationObjects("trsm_R.c" "UPPER;UNIT" "L;N" "TRANSA;CONJ" 0 "trsm_RC" false ${float_type})
  76. #hemm
  77. GenerateCombinationObjects("zhemm_k.c" "LOWER" "U" "NN" 0 "hemm_L" false ${float_type})
  78. GenerateCombinationObjects("zhemm_k.c" "LOWER" "U" "NC;RSIDE" 0 "hemm_R" false ${float_type})
  79. #her2k
  80. GenerateCombinationObjects("zher2k_kernel.c" "LOWER;CONJ" "U;N" "" 2 "her2k_kernel" false ${float_type})
  81. GenerateNamedObjects("zher2k_k.c" "HER2K" "her2k_UN" false "" "" false ${float_type})
  82. GenerateNamedObjects("zher2k_k.c" "HER2K;TRANS;CONJ" "her2k_UC" false "" "" false ${float_type})
  83. GenerateNamedObjects("zher2k_k.c" "HER2K;LOWER" "her2k_LN" false "" "" false ${float_type})
  84. GenerateNamedObjects("zher2k_k.c" "HER2K;LOWER;TRANS;CONJ" "her2k_LC" false "" "" false ${float_type})
  85. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  86. #herk
  87. GenerateCombinationObjects("zherk_k.c" "LOWER" "U" "HERK;THREADED_LEVEL3" 3 "herk_thread_N" false ${float_type})
  88. GenerateCombinationObjects("zherk_k.c" "LOWER" "U" "HERK;THREADED_LEVEL3;TRANS;CONJ" 3 "herk_thread_C" false ${float_type})
  89. #hemm
  90. GenerateCombinationObjects("zhemm_k.c" "LOWER" "U" "NN;THREADED_LEVEL3" 0 "hemm_thread_L" false ${float_type})
  91. GenerateCombinationObjects("zhemm_k.c" "LOWER" "U" "NC;RSIDE;THREADED_LEVEL3" 0 "hemm_thread_R" false ${float_type})
  92. #her2k
  93. GenerateNamedObjects("zher2k_k.c" "HER2K" "her2k_UN" false "" "" false ${float_type})
  94. GenerateNamedObjects("zher2k_k.c" "HER2K;TRANS;CONJ" "her2k_UC" false "" "" false ${float_type})
  95. GenerateNamedObjects("zher2k_k.c" "HER2K;LOWER" "her2k_LN" false "" "" false ${float_type})
  96. GenerateNamedObjects("zher2k_k.c" "HER2K;LOWER;TRANS;CONJ" "her2k_LC" false "" "" false ${float_type})
  97. endif()
  98. # special gemm defines for complex
  99. foreach (gemm_define ${GEMM_COMPLEX_DEFINES})
  100. string(TOLOWER ${gemm_define} gemm_define_LC)
  101. GenerateNamedObjects("gemm.c" "${gemm_define}" "gemm_${gemm_define_LC}" false "" "" false ${float_type})
  102. if(USE_GEMM3M)
  103. GenerateNamedObjects("gemm3m.c" "${gemm_define}" "gemm3m_${gemm_define_LC}" false "" "" false ${float_type})
  104. endif()
  105. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  106. GenerateNamedObjects("gemm.c" "${gemm_define};THREADED_LEVEL3" "gemm_thread_${gemm_define_LC}" false "" "" false ${float_type})
  107. if(USE_GEMM3M)
  108. GenerateNamedObjects("gemm3m.c" "${gemm_define};THREADED_LEVEL3" "gemm3m_thread_${gemm_define_LC}" false "" "" false ${float_type})
  109. endif()
  110. endif ()
  111. endforeach ()
  112. if ( BUILD_COMPLEX16 AND NOT BUILD_DOUBLE)
  113. foreach (gemm_define ${GEMM_COMPLEX_DEFINES})
  114. string(TOLOWER ${gemm_define} gemm_define_LC)
  115. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  116. GenerateNamedObjects("gemm.c" "${gemm_define};THREADED_LEVEL3" "gemm_thread_${gemm_define_LC}" false "" "" false "DOUBLE" )
  117. endif()
  118. endforeach()
  119. endif ()
  120. if ( BUILD_COMPLEX AND NOT BUILD_SINGLE)
  121. foreach (gemm_define ${GEMM_COMPLEX_DEFINES})
  122. string(TOLOWER ${gemm_define} gemm_define_LC)
  123. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  124. GenerateNamedObjects("gemm.c" "${gemm_define};THREADED_LEVEL3" "gemm_thread_${gemm_define_LC}" false "" "" false "SINGLE" )
  125. endif()
  126. endforeach()
  127. endif ()
  128. # for gemm3m
  129. if(USE_GEMM3M)
  130. foreach (GEMM_DEFINE ${GEMM_DEFINES})
  131. string(TOLOWER ${GEMM_DEFINE} GEMM_DEFINE_LC)
  132. GenerateNamedObjects("gemm3m.c" "${GEMM_DEFINE}" "gemm3m_${GEMM_DEFINE_LC}" false "" "" false ${float_type})
  133. if (USE_THREAD AND NOT USE_SIMPLE_THREADED_LEVEL3)
  134. GenerateNamedObjects("gemm3m.c" "${GEMM_DEFINE};THREADED_LEVEL3" "gemm3m_thread_${GEMM_DEFINE_LC}" false "" "" false ${float_type})
  135. endif ()
  136. endforeach ()
  137. endif()
  138. endif ()
  139. endforeach ()
  140. #HPLOBJS =
  141. # dgemm_nn.c dgemm_nt.c dgemm_tn.c dgemm_tt.c
  142. # dtrsm_LNUU.c dtrsm_LNUN.c dtrsm_LNLU.c dtrsm_LNLN.c
  143. # dtrsm_LTUU.c dtrsm_LTUN.c dtrsm_LTLU.c dtrsm_LTLN.c
  144. # dtrsm_RNUU.c dtrsm_RNUN.c dtrsm_RNLU.c dtrsm_RNLN.c
  145. # dtrsm_RTUU.c dtrsm_RTUN.c dtrsm_RTLU.c dtrsm_RTLN.c
  146. #
  147. #if (USE_SIMPLE_THREADED_LEVEL3)
  148. # HPLOBJS += dgemm_thread_nn.c dgemm_thread_nt.c
  149. # dgemm_thread_tn.c dgemm_thread_tt.c
  150. #endif
  151. #
  152. add_library(driver_level3 OBJECT ${OPENBLAS_SRC})