nix: add missing install hooks to cuda

ggerganov · Sep 15, 2023 · 7aa42f7 · 7aa42f7
1 parent 7fa5ba0
commit 7aa42f7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/flake.nix b/flake.nix
@@ -80,6 +80,8 @@
 
           buildFlags = [ "LLAMA_CUBLAS=1" ];
           installPhase = ''
+            runHook preInstall
+
             mkdir -p $out/bin
 
             # TODO(Green-Sky): add install target to Makefile, or wait for cmake support
@@ -91,6 +93,8 @@
             mv quantize $out/bin/
             mv llama-bench $out/bin/
             mv train-text-from-scratch $out/bin/
+
+            runHook postInstall
           '';
         };
         packages.rocm = pkgs.stdenv.mkDerivation {