Skip to content

Commit 7aa42f7

Browse files
committed
nix: add missing install hooks to cuda
1 parent 7fa5ba0 commit 7aa42f7

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

flake.nix

+4
Original file line numberDiff line numberDiff line change
@@ -80,6 +80,8 @@
8080

8181
buildFlags = [ "LLAMA_CUBLAS=1" ];
8282
installPhase = ''
83+
runHook preInstall
84+
8385
mkdir -p $out/bin
8486
8587
# TODO(Green-Sky): add install target to Makefile, or wait for cmake support
@@ -91,6 +93,8 @@
9193
mv quantize $out/bin/
9294
mv llama-bench $out/bin/
9395
mv train-text-from-scratch $out/bin/
96+
97+
runHook postInstall
9498
'';
9599
};
96100
packages.rocm = pkgs.stdenv.mkDerivation {

0 commit comments

Comments
 (0)