tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 cudaNamePrefix,
3 cudnn-frontend,
4 jq,
5 lib,
6 writeShellApplication,
7}:
8let
9 inherit (lib.meta) getExe';
10in
11writeShellApplication {
12 derivationArgs = {
13 __structuredAttrs = true;
14 strictDeps = true;
15 };
16 name = "${cudaNamePrefix}-tests-cudnn-frontend-tests";
17 runtimeInputs = [
18 cudnn-frontend.tests
19 jq
20 ];
21 text = ''
22 args=( "${getExe' cudnn-frontend.tests "tests"}" )
23
24 if (( $# != 0 ))
25 then
26 args+=( "$@" )
27 "''${args[@]}"
28 else
29 args+=(
30 --success
31 --rng-seed=0
32 --reporter=json
33 )
34 echo "Running with default arguments: ''${args[*]}" >&2
35 "''${args[@]}" | jq
36 fi
37 '';
38}