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.

lit.site.cfg.py.in 1.9 kB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849
  1. @LIT_SITE_CFG_IN_HEADER@
  2. import sys
  3. config.host_triple = "@LLVM_HOST_TRIPLE@"
  4. config.target_triple = "@TARGET_TRIPLE@"
  5. config.llvm_src_root = "@LLVM_SOURCE_DIR@"
  6. config.llvm_obj_root = "@LLVM_BINARY_DIR@"
  7. config.llvm_tools_dir = "@LLVM_BINARY_DIR@/bin"
  8. config.llvm_lib_dir = "@LLVM_LIBRARY_DIR@"
  9. config.llvm_shlib_dir = "@SHLIBDIR@"
  10. config.llvm_shlib_ext = "@SHLIBEXT@"
  11. config.llvm_exe_ext = "@EXEEXT@"
  12. config.lit_tools_dir = "@LLVM_LIT_TOOLS_DIR@"
  13. config.python_executable = "@PYTHON_EXECUTABLE@"
  14. config.gold_executable = "@GOLD_EXECUTABLE@"
  15. config.ld64_executable = "@LD64_EXECUTABLE@"
  16. config.enable_shared = @ENABLE_SHARED@
  17. config.enable_assertions = @ENABLE_ASSERTIONS@
  18. config.targets_to_build = "@TARGETS_TO_BUILD@"
  19. config.native_target = "@LLVM_NATIVE_ARCH@"
  20. config.llvm_bindings = "@LLVM_BINDINGS@".split(' ')
  21. config.host_os = "@HOST_OS@"
  22. config.host_cc = "@HOST_CC@"
  23. config.host_cxx = "@HOST_CXX@"
  24. # Note: ldflags can contain double-quoted paths, so must use single quotes here.
  25. config.host_ldflags = '@HOST_LDFLAGS@'
  26. config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@"
  27. config.llvm_host_triple = '@LLVM_HOST_TRIPLE@'
  28. config.host_arch = "@HOST_ARCH@"
  29. config.mgb_src_root = "@CMAKE_SOURCE_DIR@"
  30. config.mgb_obj_root = "@CMAKE_BINARY_DIR@"
  31. # Support substitution of the tools_dir with user parameters. This is
  32. # used when we can't determine the tool dir at configuration time.
  33. try:
  34. config.llvm_tools_dir = config.llvm_tools_dir % lit_config.params
  35. config.llvm_shlib_dir = config.llvm_shlib_dir % lit_config.params
  36. except KeyError:
  37. e = sys.exc_info()[1]
  38. key, = e.args
  39. lit_config.fatal("unable to find %r parameter, use '--param=%s=VALUE'" % (key,key))
  40. import lit.llvm
  41. lit.llvm.initialize(lit_config, config)
  42. # Let the main config do the real work.
  43. lit_config.load_config(config, "@CMAKE_SOURCE_DIR@/src/jit/test/mlir/utils/lit.cfg.py")

MegEngine 安装包中集成了使用 GPU 运行代码所需的 CUDA 环境,不用区分 CPU 和 GPU 版。 如果想要运行 GPU 程序,请确保机器本身配有 GPU 硬件设备并安装好驱动。 如果你想体验在云端 GPU 算力平台进行深度学习开发的感觉,欢迎访问 MegStudio 平台