Add check for host inlining

Author: eregon

ci.jsonnet

@@ -65,13 +65,14 @@ local part_definitions = {
         "*.log",
       ],
 
+      mx_options:: [],
       mx_build_options:: [],
     },
 
     build: {
       setup+: [["mx", "sversions"]] +
               # aot-build.log is used for the build-stats metrics, in other cases it does no harm
-              jt(["build", "--env", self.mx_env, "--"] + self.mx_build_options + ["|", "tee", "aot-build.log"]) +
+              jt(["build", "--env", self.mx_env] + self.mx_options + ["--"] + self.mx_build_options + ["|", "tee", "aot-build.log"]) +
               [
                 # make sure jt always uses what was just built
                 ["set-export", "RUBY_BIN", jt(["--use", self.mx_env, "--silent", "launcher"])[0]],
@@ -204,6 +205,14 @@ local part_definitions = {
         HOST_VM_CONFIG: "graal-enterprise",
       },
     },
+    host_inlining_log: {
+      # Same as in mx.truffleruby/native-host-inlining
+      mx_options+:: [
+        "--extra-image-builder-argument=rubyvm:-H:Log=TruffleHostInliningPhase,~CanonicalizerPhase,~GraphBuilderPhase",
+        "--extra-image-builder-argument=rubyvm:-H:+TruffleHostInliningPrintExplored",
+        "--extra-image-builder-argument=rubyvm:-Dgraal.LogFile=host-inlining.txt",
+      ],
+    },
   },
 
   jdk: {
@@ -533,13 +542,13 @@ local composition_environment = utils.add_inclusion_tracking(part_definitions, "
       "ruby-test-compiler-graal-enterprise-17": $.platform.linux + $.jdk.v17 + $.env.jvm_ee + gate + $.use.truffleruby + $.run.test_compiler,
       "ruby-test-compiler-graal-enterprise-20": $.platform.linux + $.jdk.v20 + $.env.jvm_ee + gate + $.use.truffleruby + $.run.test_compiler,
 
-      "ruby-test-svm-graal-core-linux-17":              $.platform.linux          + $.jdk.v17 + $.env.native    + $.env.gdb_svm + gate + native_tests,
+      "ruby-test-svm-graal-core-linux-17":              $.platform.linux          + $.jdk.v17 + $.env.native    + $.env.gdb_svm + gate + native_tests + $.env.host_inlining_log,
       "ruby-test-svm-graal-core-linux-20":              $.platform.linux          + $.jdk.v20 + $.env.native    + $.env.gdb_svm + gate + native_tests,
       "ruby-test-svm-graal-core-darwin-amd64-17":       $.platform.darwin_amd64   + $.jdk.v17 + $.env.native    + $.env.gdb_svm + gate + native_tests,
       "ruby-test-svm-graal-core-darwin-amd64-20":       $.platform.darwin_amd64   + $.jdk.v20 + $.env.native    + $.env.gdb_svm + gate + native_tests,
       "ruby-test-svm-graal-core-darwin-aarch64-17":     $.platform.darwin_aarch64 + $.jdk.v17 + $.env.native    +                 gate + native_tests,
       "ruby-test-svm-graal-core-darwin-aarch64-20":     $.platform.darwin_aarch64 + $.jdk.v20 + $.env.native    +                 gate + native_tests,
-      "ruby-test-svm-graal-enterprise-linux":           $.platform.linux          + $.jdk.v17 + $.env.native_ee + $.env.gdb_svm + gate + native_tests,
+      "ruby-test-svm-graal-enterprise-linux":           $.platform.linux          + $.jdk.v17 + $.env.native_ee + $.env.gdb_svm + gate + native_tests + $.env.host_inlining_log,
       "ruby-test-svm-graal-enterprise-darwin-aarch64 ": $.platform.darwin_aarch64 + $.jdk.v17 + $.env.native_ee +                 gate + native_tests,
     },
 

test/truffle/compiler/host-inlining.sh

@@ -0,0 +1,37 @@
+#!/usr/bin/env bash
+
+source test/truffle/common.sh.inc
+
+file=${1:-host-inlining.txt}
+
+if [ -f "$file" ]; then
+  # shellcheck disable=SC2016
+  ruby tool/extract_host_inlining.rb 'org.truffleruby.language.methods.CallForeignMethodNodeGen$ConvertForOperatorAndReDispatchNodeGen.execute' "$file" > out.txt
+  # shellcheck disable=SC2016
+  grep -F 'Root[org.truffleruby.language.methods.CallForeignMethodNodeGen$ConvertForOperatorAndReDispatchNodeGen.execute]' out.txt
+  if ! grep -E 'Out of budget|too big to explore' out.txt; then
+    echo 'ConvertForOperatorAndReDispatchNodeGen.execute should be out of budget (too much code), did host inlining output change?'
+    cat out.txt
+    exit 1
+  fi
+
+  ruby tool/extract_host_inlining.rb org.truffleruby.language.dispatch.RubyCallNode.execute "$file" > out.txt
+  grep -F 'Root[org.truffleruby.language.dispatch.RubyCallNode.execute]' out.txt
+  if grep -E 'Out of budget|too big to explore' out.txt; then
+    echo 'RubyCallNode.execute no longer fits in host inlining budget'
+    cat out.txt
+    exit 1
+  fi
+
+  ruby tool/extract_host_inlining.rb --simplify org.truffleruby.language.dispatch.RubyCallNode.execute "$file" | grep CUTOFF > simplified.txt
+  if grep -v -E 'not direct call|not inlinable|marked to be not used for inlining|leads to unwind' simplified.txt; then
+    echo 'RubyCallNode.execute has unexpected CUTOFFs'
+    cat out.txt
+    exit 1
+  fi
+
+  rm out.txt
+  rm simplified.txt
+else
+  echo "$file does not exist, skipping test"
+fi