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 9.3 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234
  1. include_directories(${PROJECT_SOURCE_DIR})
  2. include_directories(${PROJECT_BINARY_DIR})
  3. set(BLAS1_SOURCES
  4. copy.c
  5. nrm2.c
  6. )
  7. set(BLAS1_REAL_ONLY_SOURCES
  8. rotm.c rotmg.c # N.B. these do not have complex counterparts
  9. rot.c
  10. asum.c
  11. sum.c
  12. )
  13. # these will have 'z' prepended for the complex version
  14. set(BLAS1_MANGLED_SOURCES
  15. axpy.c swap.c
  16. scal.c
  17. dot.c
  18. rotg.c
  19. axpby.c
  20. )
  21. # TODO: USE_NETLIB_GEMV should switch gemv.c to netlib/*gemv.f
  22. # these all have 'z' sources for complex versions
  23. set(BLAS2_SOURCES
  24. gemv.c ger.c
  25. trsv.c trmv.c
  26. syr2.c gbmv.c
  27. sbmv.c
  28. spr2.c
  29. tbsv.c tbmv.c
  30. tpsv.c tpmv.c
  31. )
  32. set(BLAS2_REAL_ONLY_SOURCES
  33. symv.c syr.c spmv.c spr.c
  34. )
  35. set(BLAS2_COMPLEX_LAPACK_SOURCES
  36. symv.c syr.c spmv.c spr.c
  37. )
  38. set(BLAS2_COMPLEX_ONLY_MANGLED_SOURCES
  39. hemv.c hbmv.c
  40. her.c her2.c
  41. hpmv.c hpr.c
  42. hpr2.c
  43. )
  44. # these do not have separate 'z' sources
  45. set(BLAS3_SOURCES
  46. gemm.c symm.c
  47. trsm.c syrk.c syr2k.c gemmt.c
  48. )
  49. set(BLAS3_MANGLED_SOURCES
  50. omatcopy.c imatcopy.c
  51. geadd.c
  52. )
  53. # generate the BLAS objs once with and once without cblas
  54. set (CBLAS_FLAGS "")
  55. if (NOT DEFINED NO_FBLAS)
  56. list(APPEND CBLAS_FLAGS 0)
  57. endif ()
  58. if (NOT DEFINED NO_CBLAS)
  59. list(APPEND CBLAS_FLAGS 1)
  60. endif ()
  61. foreach (CBLAS_FLAG ${CBLAS_FLAGS})
  62. # TODO: don't compile complex sources with cblas for now, the naming schemes are all different and they will have to be handled separately from SINGLE/DOUBLE
  63. set(DISABLE_COMPLEX 0)
  64. set(MANGLE_COMPLEX 3)
  65. if (CBLAS_FLAG EQUAL 1)
  66. # set(DISABLE_COMPLEX 1)
  67. # set(MANGLE_COMPLEX 1)
  68. endif ()
  69. GenerateNamedObjects("${BLAS1_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${DISABLE_COMPLEX})
  70. GenerateNamedObjects("${BLAS1_REAL_ONLY_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false 1)
  71. GenerateNamedObjects("${BLAS1_MANGLED_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${MANGLE_COMPLEX})
  72. GenerateNamedObjects("${BLAS2_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${MANGLE_COMPLEX})
  73. GenerateNamedObjects("${BLAS2_REAL_ONLY_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false 1)
  74. if (NOT DEFINED NO_LAPACK)
  75. GenerateNamedObjects("${BLAS2_COMPLEX_LAPACK_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${MANGLE_COMPLEX})
  76. endif ()
  77. GenerateNamedObjects("${BLAS2_COMPLEX_ONLY_MANGLED_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false 4)
  78. GenerateNamedObjects("${BLAS3_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${DISABLE_COMPLEX})
  79. GenerateNamedObjects("${BLAS3_MANGLED_SOURCES}" "" "" ${CBLAS_FLAG} "" "" false ${MANGLE_COMPLEX})
  80. GenerateNamedObjects("xerbla.c" "" "xerbla" ${CBLAS_FLAG} "" "" true)
  81. #sdsdot, dsdot
  82. if (BUILD_SINGLE OR BUILD_DOUBLE)
  83. GenerateNamedObjects("sdsdot.c" "" "sdsdot" ${CBLAS_FLAG} "" "" true "SINGLE")
  84. endif ()
  85. if (BUILD_DOUBLE)
  86. GenerateNamedObjects("dsdot.c" "" "dsdot" ${CBLAS_FLAG} "" "" true "SINGLE")
  87. endif ()
  88. # trmm is trsm with a compiler flag set
  89. GenerateNamedObjects("trsm.c" "TRMM" "trmm" ${CBLAS_FLAG})
  90. # max and imax are compiled 4 times
  91. GenerateNamedObjects("max.c" "" "" ${CBLAS_FLAG})
  92. GenerateNamedObjects("max.c" "USE_ABS" "amax" ${CBLAS_FLAG})
  93. GenerateNamedObjects("max.c" "USE_ABS;USE_MIN" "amin" ${CBLAS_FLAG})
  94. GenerateNamedObjects("max.c" "USE_MIN" "min" ${CBLAS_FLAG})
  95. GenerateNamedObjects("imax.c" "" "i*max" ${CBLAS_FLAG})
  96. GenerateNamedObjects("imax.c" "USE_ABS" "i*amax" ${CBLAS_FLAG})
  97. GenerateNamedObjects("imax.c" "USE_ABS;USE_MIN" "i*amin" ${CBLAS_FLAG})
  98. GenerateNamedObjects("imax.c" "USE_MIN" "i*min" ${CBLAS_FLAG})
  99. if (BUILD_BFLOAT16)
  100. GenerateNamedObjects("bf16dot.c" "" "sbdot" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  101. GenerateNamedObjects("gemm.c" "" "sbgemm" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  102. GenerateNamedObjects("gemmt.c" "" "sbgemmt" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  103. GenerateNamedObjects("sbgemv.c" "" "sbgemv" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  104. GenerateNamedObjects("tobf16.c" "SINGLE_PREC" "sbstobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  105. GenerateNamedObjects("tobf16.c" "DOUBLE_PREC" "sbdtobf16" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  106. GenerateNamedObjects("bf16to.c" "SINGLE_PREC" "sbf16tos" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  107. GenerateNamedObjects("bf16to.c" "DOUBLE_PREC" "dbf16tod" ${CBLAS_FLAG} "" "" true "BFLOAT16")
  108. endif ()
  109. # complex-specific sources
  110. foreach (float_type ${FLOAT_TYPES})
  111. if (${float_type} STREQUAL "COMPLEX" OR ${float_type} STREQUAL "ZCOMPLEX")
  112. GenerateNamedObjects("zaxpy.c" "" "axpyc" ${CBLAS_FLAG} "" "" false ${float_type})
  113. GenerateNamedObjects("zger.c" "" "geru" ${CBLAS_FLAG} "" "" false ${float_type})
  114. GenerateNamedObjects("zger.c" "CONJ" "gerc" ${CBLAS_FLAG} "" "" false ${float_type})
  115. GenerateNamedObjects("zdot.c" "CONJ" "dotc" ${CBLAS_FLAG} "" "" false ${float_type})
  116. GenerateNamedObjects("zdot.c" "" "dotu" ${CBLAS_FLAG} "" "" false ${float_type})
  117. GenerateNamedObjects("symm.c" "HEMM" "hemm" ${CBLAS_FLAG} "" "" false ${float_type})
  118. GenerateNamedObjects("syrk.c" "HEMM" "herk" ${CBLAS_FLAG} "" "" false ${float_type})
  119. GenerateNamedObjects("syr2k.c" "HEMM" "her2k" ${CBLAS_FLAG} "" "" false ${float_type})
  120. if (USE_GEMM3M)
  121. GenerateNamedObjects("gemm.c" "GEMM3M" "gemm3m" ${CBLAS_FLAG} "" "" false ${float_type})
  122. endif()
  123. endif ()
  124. if (${float_type} STREQUAL "COMPLEX")
  125. GenerateNamedObjects("zscal.c" "SSCAL" "sscal" ${CBLAS_FLAG} "" "" false "COMPLEX")
  126. GenerateNamedObjects("nrm2.c" "" "scnrm2" ${CBLAS_FLAG} "" "" true "COMPLEX")
  127. GenerateNamedObjects("zrot.c" "" "csrot" ${CBLAS_FLAG} "" "" true "COMPLEX")
  128. GenerateNamedObjects("max.c" "USE_ABS;USE_MIN" "scamin" ${CBLAS_FLAG} "" "" true "COMPLEX")
  129. GenerateNamedObjects("max.c" "USE_ABS" "scamax" ${CBLAS_FLAG} "" "" true "COMPLEX")
  130. GenerateNamedObjects("asum.c" "" "scasum" ${CBLAS_FLAG} "" "" true "COMPLEX")
  131. GenerateNamedObjects("sum.c" "" "scsum" ${CBLAS_FLAG} "" "" true "COMPLEX")
  132. endif ()
  133. if (${float_type} STREQUAL "ZCOMPLEX")
  134. GenerateNamedObjects("zscal.c" "SSCAL" "dscal" ${CBLAS_FLAG} "" "" false "ZCOMPLEX")
  135. GenerateNamedObjects("nrm2.c" "" "dznrm2" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  136. GenerateNamedObjects("zrot.c" "" "zdrot" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  137. GenerateNamedObjects("max.c" "USE_ABS;USE_MIN" "dzamin" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  138. GenerateNamedObjects("max.c" "USE_ABS" "dzamax" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  139. GenerateNamedObjects("asum.c" "" "dzasum" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  140. GenerateNamedObjects("sum.c" "" "dzsum" ${CBLAS_FLAG} "" "" true "ZCOMPLEX")
  141. endif ()
  142. endforeach ()
  143. endforeach ()
  144. #Special functions for CBLAS
  145. if (NOT DEFINED NO_CBLAS)
  146. foreach (float_type ${FLOAT_TYPES})
  147. if (${float_type} STREQUAL "COMPLEX" OR ${float_type} STREQUAL "ZCOMPLEX")
  148. #cblas_dotc_sub cblas_dotu_sub
  149. GenerateNamedObjects("zdot.c" "FORCE_USE_STACK" "dotu_sub" 1 "" "" false ${float_type})
  150. GenerateNamedObjects("zdot.c" "FORCE_USE_STACK;CONJ" "dotc_sub" 1 "" "" false ${float_type})
  151. endif()
  152. endforeach ()
  153. endif()
  154. if (NOT DEFINED NO_LAPACK)
  155. set(LAPACK_SOURCES
  156. lapack/gesv.c
  157. )
  158. # prepend z for complex versions
  159. set(LAPACK_MANGLED_SOURCES
  160. lapack/getrf.c lapack/getrs.c lapack/potrf.c lapack/getf2.c
  161. lapack/potf2.c lapack/laswp.c lapack/lauu2.c
  162. lapack/lauum.c lapack/trti2.c lapack/trtri.c
  163. )
  164. GenerateNamedObjects("${LAPACK_SOURCES}")
  165. if (NOT RELAPACK_REPLACE)
  166. GenerateNamedObjects("${LAPACK_MANGLED_SOURCES}" "" "" 0 "" "" 0 3)
  167. else ()
  168. GenerateNamedObjects("lapack/getrs.c" "" "" 0 "" "" 0 3)
  169. GenerateNamedObjects("lapack/getf2.c" "" "" 0 "" "" 0 3)
  170. GenerateNamedObjects("lapack/potf2.c" "" "" 0 "" "" 0 3)
  171. GenerateNamedObjects("lapack/laswp.c" "" "" 0 "" "" 0 3)
  172. GenerateNamedObjects("lapack/lauu2.c" "" "" 0 "" "" 0 3)
  173. GenerateNamedObjects("lapack/trti2.c" "" "" 0 "" "" 0 3)
  174. endif()
  175. endif ()
  176. if ( BUILD_COMPLEX AND NOT BUILD_SINGLE)
  177. GenerateNamedObjects("scal.c" "" "scal" 0 "" "" false "SINGLE")
  178. GenerateNamedObjects("copy.c" "" "copy" 0 "" "" false "SINGLE")
  179. GenerateNamedObjects("dot.c" "" "dot" 0 "" "" false "SINGLE")
  180. GenerateNamedObjects("rot.c" "" "rot" 0 "" "" false "SINGLE")
  181. GenerateNamedObjects("nrm2.c" "" "nrm2" 0 "" "" false "SINGLE")
  182. GenerateNamedObjects("gemv.c" "" "gemv" 0 "" "" false "SINGLE")
  183. GenerateNamedObjects("gemm.c" "" "gemm" 0 "" "" false "SINGLE")
  184. GenerateNamedObjects("asum.c" "" "asum" 0 "" "" false "SINGLE")
  185. GenerateNamedObjects("swap.c" "" "swap" 0 "" "" false "SINGLE")
  186. GenerateNamedObjects("axpy.c" "" "axpy" 0 "" "" false "SINGLE")
  187. GenerateNamedObjects("imax.c" "USE_ABS" "i*amax" 0 "" "" false "SINGLE")
  188. endif ()
  189. if ( BUILD_COMPLEX16 AND NOT BUILD_DOUBLE)
  190. GenerateNamedObjects("scal.c" "" "scal" 0 "" "" false "DOUBLE")
  191. GenerateNamedObjects("copy.c" "" "copy" 0 "" "" false "DOUBLE")
  192. GenerateNamedObjects("dot.c" "" "dot" 0 "" "" false "DOUBLE")
  193. GenerateNamedObjects("rot.c" "" "rot" 0 "" "" false "DOUBLE")
  194. GenerateNamedObjects("nrm2.c" "" "nrm2" 0 "" "" false "DOUBLE")
  195. GenerateNamedObjects("gemv.c" "" "gemv" 0 "" "" false "DOUBLE")
  196. GenerateNamedObjects("gemm.c" "" "gemm" 0 "" "" false "DOUBLE")
  197. GenerateNamedObjects("asum.c" "" "asum" 0 "" "" false "DOUBLE")
  198. GenerateNamedObjects("swap.c" "" "swap" 0 "" "" false "DOUBLE")
  199. GenerateNamedObjects("axpy.c" "" "axpy" 0 "" "" false "DOUBLE")
  200. GenerateNamedObjects("imax.c" "USE_ABS" "i*amax" 0 "" "" false "DOUBLE")
  201. endif ()
  202. add_library(interface OBJECT ${OPENBLAS_SRC})