Something went wrong. Try again.
What happens if I push all of nixpkgs
Something went wrong. Try again.
1.7 kB · 66 lines
Nix
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667{ lib, runCommand, python3Packages, makeWrapper,}:{ feature ? "cuda", name ? if feature == null then "cpu" else feature, libraries ? [ ], # [PythonPackage] | (PackageSet -> [PythonPackage]) ...}@args:
let inherit (builtins) isFunction all; librariesFun = if isFunction libraries then libraries else (_: libraries);in
assert lib.assertMsg ( isFunction libraries || all (python3Packages.hasPythonModule) libraries) "writeGpuTestPython was passed `libraries` from the wrong python release";
content:
let interpreter = python3Packages.python.withPackages librariesFun; tester = runCommand "tester-${name}" ( lib.removeAttrs args [ "libraries" "name" ] // { inherit content; nativeBuildInputs = args.nativeBuildInputs or [ ] ++ [ makeWrapper ]; passAsFile = args.passAsFile or [ ] ++ [ "content" ]; } ) '' mkdir -p "$out"/bin cat << EOF >"$out/bin/$name" #!${lib.getExe interpreter} EOF cat "$contentPath" >>"$out/bin/$name" chmod +x "$out/bin/$name"
if [[ -n "''${makeWrapperArgs+''${makeWrapperArgs[@]}}" ]] ; then wrapProgram "$out/bin/$name" ''${makeWrapperArgs[@]} fi ''; tester' = tester.overrideAttrs (oldAttrs: { passthru.gpuCheck = runCommand "test-${name}" { nativeBuildInputs = [ tester' ]; requiredSystemFeatures = lib.optionals (feature != null) [ feature ]; } '' set -e ${tester.meta.mainProgram or (lib.getName tester')} touch $out ''; });intester'