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.

pslite.cmake 549 B

1234567891011121314
  1. set(pslite_USE_STATIC_LIBS ON)
  2. if (${ENABLE_IBVERBS} STREQUAL "ON")
  3. set(pslite_CXXFLAGS "USE_IBVERBS=1")
  4. endif()
  5. mindspore_add_pkg(pslite
  6. LIBS ps
  7. URL https://github.com/dmlc/ps-lite/archive/34fd45cae457d59850fdcb2066467778d0673f21.zip
  8. MD5 393c0e27b68bfaf96718caa3aa96f5a3
  9. PATCHES ${CMAKE_SOURCE_DIR}/third_party/patch/pslite/ps_lite.patch001
  10. ONLY_MAKE True
  11. ONLY_MAKE_INCS include/*
  12. ONLY_MAKE_LIBS build/*)
  13. include_directories(${pslite_INC})
  14. add_library(mindspore::pslite ALIAS pslite::ps)