nordic-dev.net/nixpkgs
nordic-dev.net/nixpkgs
2
0
Fork 0
mirror of https://github.com/NixOS/nixpkgs.git synced 2024-11-30 19:02:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

apptainer: add a cuda test

Browse Source
This commit is contained in:
Someone Serge 2024-01-06 22:50:52 +00:00
parent 07305dff4f
commit 0c4fe1f452
No known key found for this signature in database
GPG Key ID: 7B0E3B1390D61DA4
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
pkgs/applications/virtualization/singularity/generic.nix
View File

@ -33,6 +33,7 @@ in
, which
# Build inputs
, bash
, callPackage
, conmon
, coreutils
, cryptsetup
@ -269,6 +270,37 @@ in
contents = [ hello cowsay ];
singularity = finalAttrs.finalPackage;
};
image-saxpy = callPackage
({ singularity-tools, cudaPackages }:
singularity-tools.buildImage {
name = "saxpy";
contents = [ cudaPackages.saxpy ];
memSize = 2048;
diskSize = 2048;
singularity = finalAttrs.finalPackage;
})
{ };
};
gpuChecks = lib.optionalAttrs (projectName == "apptainer") {
saxpy =
callPackage
({ runCommand, writeShellScriptBin }:
let
unwrapped = writeShellScriptBin "apptainer-cuda-saxpy"
''
${lib.getExe finalAttrs.finalPackage} exec --nv --nvccli ${finalAttrs.passthru.tests.image-saxpy} saxpy
'';
in
runCommand "run-apptainer-cuda-saxpy"
{
requiredSystemFeatures = [ "cuda" ];
nativeBuildInputs = [ unwrapped ];
passthru = { inherit unwrapped; };
}
''
apptainer-cuda-saxpy
'')
{ };
};
};
})