llvmPackages.bolt: add experimental bolt hook

lib/systems/default.nix

@@ -93,6 +93,8 @@ let
       # Derived meta-data
       useLLVM = final.isFreeBSD || final.isOpenBSD;
 
+      useBolt = false;
+
       libc =
         /**/ if final.isDarwin                then "libSystem"
         else if final.isMinGW                 then "msvcrt"

pkgs/build-support/cc-wrapper/default.nix

@@ -728,6 +728,14 @@ stdenvNoCC.mkDerivation {
       substituteAll ${./add-clang-cc-cflags-before.sh} $out/nix-support/add-local-cc-cflags-before.sh
     ''
 
+    ###
+    ### Bolt support
+    ### Binaries need to be linked with relocations.
+    ###
+    + optionalString targetPlatform.useBolt ''
+      echo " --emit-relocs" >> $out/nix-support/cc-ldflags
+    ''
+
     ##
     ## Extra custom steps
     ##

pkgs/development/compilers/llvm/common/bolt/default.nix

@@ -71,13 +71,18 @@ stdenv.mkDerivation (finalAttrs: {
   postInstall = ''
     mkdir -p $dev/lib
     mv $out/lib/libLLVMBOLT*.a $dev/lib
+
+    mkdir -p $out/nix-support
+    cp $setupHook $out/nix-support/setup-hook
   '';
 
   outputs = [
     "out"
     "dev"
   ];
 
+  setupHook = ./setup-hook.sh;
+
   meta = llvm_meta // {
     homepage = "https://github.com/llvm/llvm-project/tree/main/bolt";
     description = "LLVM post-link optimizer.";

pkgs/development/compilers/llvm/common/bolt/setup-hook.sh

@@ -0,0 +1,16 @@
+# shellcheck shell=bash
+
+function boltFixupPhase {
+  if [ -d "$prefix/bin" ]; then
+    for e in $(find "$prefix/bin" -executable -type f); do
+      if $(file "$e" | cut -f2 -d ':' | grep -q ELF); then
+        llvm-bolt "$e" -o $(dirname "$e")/.bolt-$(basename "$e") "${boltFlags[@]-}"
+        mv $(dirname "$e")/.bolt-$(basename "$e") "$e"
+      fi
+    done
+  fi
+}
+
+if [ -z "${dontUseBolt-}" ]; then
+    appendToVar preFixupPhases boltFixupPhase
+fi

pkgs/stdenv/cross/default.nix

@@ -69,6 +69,9 @@ lib.init bootStages
           extraNativeBuildInputs =
             old.extraNativeBuildInputs
             ++ lib.optionals (hostPlatform.isLinux && !buildPlatform.isLinux) [ buildPackages.patchelf ]
+            ++ lib.optionals hostPlatform.useBolt [
+              buildPackages.llvmPackages.bolt
+            ]
             ++ lib.optional (
               let
                 f =

pkgs/top-level/release-attrpaths-superset.nix

@@ -51,6 +51,7 @@ let
     # cross packagesets
     pkgsLLVM = true;
     pkgsLLVMLibc = true;
+    pkgsBolt = true;
     pkgsMusl = true;
     pkgsStatic = true;
     pkgsCross = true;

pkgs/top-level/release.nix

@@ -345,6 +345,7 @@ let
       agdaPackages = packagePlatforms pkgs.agdaPackages;
 
       pkgsLLVM.stdenv = [ "x86_64-linux" "aarch64-linux" ];
+      pkgsBolt.stdenv = [ "x86_64-linux" "aarch64-linux" ];
       pkgsLLVMLibc.stdenv = [ "x86_64-linux" "aarch64-linux" ];
       pkgsArocc.stdenv = [ "x86_64-linux" "aarch64-linux" ];
       pkgsZig.stdenv = [ "x86_64-linux" "aarch64-linux" ];

pkgs/top-level/stage.nix

@@ -211,6 +211,20 @@ let
       };
     };
 
+    pkgsBolt = nixpkgsFun {
+      overlays = [
+        (self': super': {
+          pkgsBolt = super';
+        })
+      ] ++ overlays;
+      # Bootstrap a cross stdenv and apply the Bolt post-link optimizer.
+      # This is currently not possible when compiling natively,
+      # so we don't need to check hostPlatform != buildPlatform.
+      crossSystem = stdenv.hostPlatform // {
+        useBolt = true;
+      };
+    };
+
     pkgsLLVMLibc = nixpkgsFun {
       overlays = [ (self': super': {
         pkgsLLVMLibc = super';