diff options
author | Hiroshi SHIBATA <hsbt@ruby-lang.org> | 2019-07-09 21:04:07 +0900 |
---|---|---|
committer | Hiroshi SHIBATA <hsbt@ruby-lang.org> | 2019-07-09 21:04:07 +0900 |
commit | f9a2440866a2df05fcbacf31c4f4e5e18b996e15 (patch) | |
tree | b1dae201ef70c678742ba99798bbe857629b8f62 /test/lib/jit_support.rb | |
parent | d905ff61e69ce4f9bc134285f3ac1859e1eb74a6 (diff) | |
download | ruby-f9a2440866a2df05fcbacf31c4f4e5e18b996e15.tar.gz |
Restore support library for only test files.
Diffstat (limited to 'test/lib/jit_support.rb')
-rw-r--r-- | test/lib/jit_support.rb | 68 |
1 files changed, 68 insertions, 0 deletions
diff --git a/test/lib/jit_support.rb b/test/lib/jit_support.rb new file mode 100644 index 0000000000..c1624cc3b2 --- /dev/null +++ b/test/lib/jit_support.rb @@ -0,0 +1,68 @@ +require 'rbconfig' + +module JITSupport + JIT_TIMEOUT = 600 # 10min for each... + JIT_SUCCESS_PREFIX = 'JIT success \(\d+\.\dms\)' + UNSUPPORTED_COMPILERS = [ + %r[\Aicc\b], + %r[\A/opt/developerstudio\d+\.\d+/bin/cc\z], + ] + + module_function + # Run Ruby script with --jit-wait (Synchronous JIT compilation). + # Returns [stdout, stderr] + def eval_with_jit(env = nil, script, **opts) + stdout, stderr = nil, nil + # retry 3 times while cc1 error happens. + 3.times do |i| + stdout, stderr, status = eval_with_jit_without_retry(env, script, **opts) + assert_equal(true, status.success?, "Failed to run script with JIT:\n#{code_block(script)}\nstdout:\n#{code_block(stdout)}\nstderr:\n#{code_block(stderr)}") + break unless retried_stderr?(stderr) + end + [stdout, stderr] + end + + def eval_with_jit_without_retry(env = nil, script, verbose: 0, min_calls: 5, save_temps: false, max_cache: 1000, wait: true, timeout: JIT_TIMEOUT) + args = [ + '--disable-gems', "--jit-verbose=#{verbose}", + "--jit-min-calls=#{min_calls}", "--jit-max-cache=#{max_cache}", + ] + args << '--jit-wait' if wait + args << '--jit-save-temps' if save_temps + args << '-e' << script + base_env = { 'MJIT_SEARCH_BUILD_DIR' => 'true' } # workaround to skip requiring `make install` for `make test-all` + if preloadenv = RbConfig::CONFIG['PRELOADENV'] and !preloadenv.empty? + so = "mjit_build_dir.#{RbConfig::CONFIG['SOEXT']}" + base_env[preloadenv] = File.realpath(so) rescue nil + end + args.unshift(env ? base_env.merge!(env) : base_env) + EnvUtil.invoke_ruby(args, + '', true, true, timeout: timeout, + ) + end + + def supported? + return @supported if defined?(@supported) + @supported = UNSUPPORTED_COMPILERS.all? do |regexp| + !regexp.match?(RbConfig::CONFIG['CC']) + end + end + + def remove_mjit_logs(stderr) + if RubyVM::MJIT.enabled? # utility for -DFORCE_MJIT_ENABLE + stderr.gsub(/^MJIT warning: Skipped to compile unsupported instruction: \w+\n/m, '') + else + stderr + end + end + + def code_block(code) + %Q["""\n#{code}\n"""\n\n] + end + + # We're retrying cc1 not found error on gcc, which should be solved in the future but ignored for now. + def retried_stderr?(stderr) + RbConfig::CONFIG['CC'].start_with?('gcc') && + stderr.include?("error trying to exec 'cc1': execvp: No such file or directory") + end +end |