mirror of
https://github.com/CHN-beta/nixos.git
synced 2024-10-24 20:18:56 +08:00
23 lines
644 B
Nix
23 lines
644 B
Nix
{ buildGoModule, cudatoolkit, src, config, cudaCapabilities ? config.cudaCapabilities, gcc, makeWrapper }:
|
|
# TODO: use addDriverRunpath
|
|
buildGoModule
|
|
{
|
|
name = "mumax";
|
|
inherit src;
|
|
vendorHash = null;
|
|
nativeBuildInputs = [ cudatoolkit gcc makeWrapper ];
|
|
CUDA_CC = builtins.concatStringsSep " " cudaCapabilities;
|
|
CPATH = "${cudatoolkit}/include";
|
|
LIBRARY_PATH = "${cudatoolkit}/lib/stubs";
|
|
doCheck = false;
|
|
postInstall =
|
|
''
|
|
rm $out/bin/{doc,test}
|
|
for i in $out/bin/*; do
|
|
if [ -f $i ]; then
|
|
wrapProgram $i --prefix LD_LIBRARY_PATH ":" "/run/opengl-driver/lib:${cudatoolkit}/lib"
|
|
fi
|
|
done
|
|
'';
|
|
}
|