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.

opencl.cmake 2.3 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. if(ENABLE_GITEE)
  2. set(REQ_URL "https://gitee.com/mirrors/OpenCL-Headers/repository/archive/v2020.06.16.tar.gz")
  3. set(MD5 "8797a525aff953ea536ebe338a9f5ef6")
  4. set(PKG_GIT_TAG "")
  5. __download_pkg_with_git(OpenCL-Headers ${REQ_URL} ${PKG_GIT_TAG} ${MD5})
  6. set(REQ_URL "https://gitee.com/mirrors/OpenCL-CLHPP/repository/archive/v2.0.12.tar.gz")
  7. set(MD5 "a07b45d676b02644482bc2c3bb90b891")
  8. set(PKG_GIT_TAG "")
  9. __download_pkg_with_git(OpenCL-CLHPP ${REQ_URL} ${PKG_GIT_TAG} ${MD5})
  10. else()
  11. set(REQ_URL "https://github.com/KhronosGroup/OpenCL-Headers/archive/v2020.06.16.tar.gz")
  12. set(MD5 "fc7627b5a8a95ecbe3d5df43bc88aa44")
  13. __download_pkg(OpenCL-Headers ${REQ_URL} ${MD5})
  14. set(REQ_URL "https://github.com/KhronosGroup/OpenCL-CLHPP/archive/v2.0.12.tar.gz")
  15. set(MD5 "bd00fca8f861b3b65660d719f00a58dd")
  16. __download_pkg(OpenCL-CLHPP ${REQ_URL} ${MD5})
  17. endif()
  18. function(gene_opencl BASEPATH)
  19. string(CONCAT CL_SRC_DIR "${BASEPATH}" "/src/runtime/kernel/opencl/cl")
  20. message(STATUS "**********gene opencl*********base path: " "${BASEPATH}" ", cl path: " "${CL_SRC_DIR}")
  21. if(NOT EXISTS ${CL_SRC_DIR})
  22. return()
  23. endif()
  24. file(GLOB_RECURSE CL_LIST ${CL_SRC_DIR}/*.cl ${CL_SRC_DIR}/int8/*.cl)
  25. foreach(file_path ${CL_LIST})
  26. file(REMOVE ${file_path}.inc)
  27. string(REGEX REPLACE ".+/(.+)\\..*" "\\1" kernel_name "${file_path}")
  28. set(inc_file_ex "${file_path}.inc")
  29. execute_process(
  30. COMMAND bash -c "sed 's/\\\\/\\\\\\\\/g' "
  31. COMMAND bash -c "sed 's/\\\"/\\\\\\\"/g' "
  32. COMMAND bash -c "sed 's/$/\\\\n\\\" \\\\/' "
  33. COMMAND bash -c "sed 's/^/\\\"/' "
  34. WORKING_DIRECTORY ${CL_SRC_DIR}
  35. INPUT_FILE ${file_path}
  36. OUTPUT_FILE ${inc_file_ex}
  37. RESULT_VARIABLE RESULT)
  38. if(NOT RESULT EQUAL "0")
  39. message(FATAL_ERROR "error! when generate ${inc_file_ex}")
  40. endif()
  41. __exec_cmd(COMMAND sed -i "1i\\static const char *${kernel_name}_source =\\\"\\\\n\\\" \\\\"
  42. ${inc_file_ex} WORKING_DIRECTORY ${CL_SRC_DIR})
  43. __exec_cmd(COMMAND sed -i "$a\\\\\;" ${inc_file_ex} WORKING_DIRECTORY ${CL_SRC_DIR})
  44. endforeach()
  45. endfunction()