mirror of https://github.com/llvm/torch-mlir
55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
# -*- Python -*-
|
|
# This file is licensed under a pytorch-style license
|
|
# See frontends/pytorch/LICENSE for license information.
|
|
|
|
@LIT_SITE_CFG_IN_HEADER@
|
|
|
|
import sys
|
|
|
|
config.host_triple = "@LLVM_HOST_TRIPLE@"
|
|
config.target_triple = "@TARGET_TRIPLE@"
|
|
config.llvm_src_root = "@LLVM_SOURCE_DIR@"
|
|
config.llvm_obj_root = "@LLVM_BINARY_DIR@"
|
|
config.llvm_tools_dir = "@LLVM_TOOLS_DIR@"
|
|
config.llvm_lib_dir = "@LLVM_LIBRARY_DIR@"
|
|
config.llvm_shlib_dir = "@SHLIBDIR@"
|
|
config.llvm_shlib_ext = "@SHLIBEXT@"
|
|
config.llvm_exe_ext = "@EXEEXT@"
|
|
config.lit_tools_dir = "@LLVM_LIT_TOOLS_DIR@"
|
|
config.python_executable = "@PYTHON_EXECUTABLE@"
|
|
config.gold_executable = "@GOLD_EXECUTABLE@"
|
|
config.ld64_executable = "@LD64_EXECUTABLE@"
|
|
config.enable_shared = @ENABLE_SHARED@
|
|
config.enable_assertions = @ENABLE_ASSERTIONS@
|
|
config.targets_to_build = "@TARGETS_TO_BUILD@"
|
|
config.native_target = "@LLVM_NATIVE_ARCH@"
|
|
config.llvm_bindings = "@LLVM_BINDINGS@".split(' ')
|
|
config.host_os = "@HOST_OS@"
|
|
config.host_cc = "@HOST_CC@"
|
|
config.host_cxx = "@HOST_CXX@"
|
|
# Note: ldflags can contain double-quoted paths, so must use single quotes here.
|
|
config.host_ldflags = '@HOST_LDFLAGS@'
|
|
config.llvm_use_sanitizer = "@LLVM_USE_SANITIZER@"
|
|
config.llvm_host_triple = '@LLVM_HOST_TRIPLE@'
|
|
config.host_arch = "@HOST_ARCH@"
|
|
config.npcomp_src_root = "@CMAKE_SOURCE_DIR@"
|
|
config.npcomp_obj_root = "@CMAKE_BINARY_DIR@"
|
|
config.enable_c10_dispatch = not @NPCOMP_ENABLE_TORCH_TYPE_DISPATCH@
|
|
|
|
# Support substitution of the tools_dir with user parameters. This is
|
|
# used when we can't determine the tool dir at configuration time.
|
|
try:
|
|
config.llvm_tools_dir = config.llvm_tools_dir % lit_config.params
|
|
config.llvm_shlib_dir = config.llvm_shlib_dir % lit_config.params
|
|
except KeyError:
|
|
e = sys.exc_info()[1]
|
|
key, = e.args
|
|
lit_config.fatal("unable to find %r parameter, use '--param=%s=VALUE'" % (key,key))
|
|
|
|
|
|
import lit.llvm
|
|
lit.llvm.initialize(lit_config, config)
|
|
|
|
# Let the main config do the real work.
|
|
lit_config.load_config(config, "@CMAKE_SOURCE_DIR@/frontends/pytorch/test/lit.cfg.py")
|