Fix Benchmark/Integration CI (rust-lang#503)

Author: wsmoses

enzyme/benchmarks/lit.site.cfg.py.in

@@ -39,19 +39,33 @@ except KeyError:
 # directories.
 config.excludes = ['Inputs']
 
-config.substitutions.append(('%shlibext', config.llvm_shlib_ext))
-config.substitutions.append(('%lli', config.llvm_tools_dir + "/lli"))
-config.substitutions.append(('%lli', config.llvm_tools_dir + "/lli"))
 config.substitutions.append(('%bench', config.bench_flags))
 config.substitutions.append(('%blink', config.bench_link))
 config.substitutions.append(('%bldpath', config.bench_ldpath))
+config.substitutions.append(('%desired_wd', ''
+                                         + "@ENZYME_SOURCE_DIR@/benchmarks/"))
+
+config.substitutions.append(('%shlibext', config.llvm_shlib_ext))
+config.substitutions.append(('%lli', config.llvm_tools_dir + "/lli" + (" --jit-kind=mcjit" if int(config.llvm_ver) >= 13 else "")
+))
+config.substitutions.append(('%opt', config.llvm_tools_dir + "/opt"))
+config.substitutions.append(('%llvmver', config.llvm_ver))
+config.substitutions.append(('%FileCheck', config.llvm_tools_dir + "/FileCheck"))
 config.substitutions.append(('%clang', config.llvm_tools_dir + "/clang"))
 config.substitutions.append(('%loadEnzyme', ''
-                                 + ' -load=@ENZYME_BINARY_DIR@/Enzyme/LLVMEnzyme-' + config.llvm_ver + config.llvm_shlib_ext
-                                 + ' -enzyme-preopt=0' ))
-config.substitutions.append(('%desired_wd', ''
-                                     + "@ENZYME_SOURCE_DIR@/benchmarks/"
-                                                                      ))
+                                 + (" --enable-new-pm=0" if int(config.llvm_ver) >= 13 else "")
+                                 + ' -load=@ENZYME_BINARY_DIR@/Enzyme/LLVMEnzyme-' + config.llvm_ver + config.llvm_shlib_ext 
+                                 + (" --enzyme-attributor=0" if int(config.llvm_ver) >= 13 else "")
+                                 + ' -enzyme-preopt=0'
+                                 ))
+config.substitutions.append(('%loadBC', ''
+                                 + ' @ENZYME_BINARY_DIR@/BCLoad/BCPass-' + config.llvm_ver + config.llvm_shlib_ext
+                                 ))
+config.substitutions.append(('%BClibdir', '@ENZYME_SOURCE_DIR@/bclib/'))
+config.substitutions.append(('%loadClangEnzyme', ''
+                                 + (" -fno-experimental-new-pass-manager" if int(config.llvm_ver) >= 13 else "")
+                                 + ' -Xclang -load -Xclang @ENZYME_BINARY_DIR@/Enzyme/ClangEnzyme-' + config.llvm_ver + config.llvm_shlib_ext
+                                 ))
 
 # Let the main config do the real work.
 lit_config.load_config(config, "@ENZYME_SOURCE_DIR@/benchmarks/lit.cfg.py")

enzyme/test/lit.site.cfg.py.in

@@ -37,7 +37,8 @@ except KeyError:
 config.excludes = ['Inputs']
 
 config.substitutions.append(('%shlibext', config.llvm_shlib_ext))
-config.substitutions.append(('%lli', config.llvm_tools_dir + "/lli"))
+config.substitutions.append(('%lli', config.llvm_tools_dir + "/lli" + (" --jit-kind=mcjit" if int(config.llvm_ver) >= 13 else "")
+))
 config.substitutions.append(('%opt', config.llvm_tools_dir + "/opt"))
 config.substitutions.append(('%llvmver', config.llvm_ver))
 config.substitutions.append(('%FileCheck', config.llvm_tools_dir + "/FileCheck"))