pkgs/development/cuda-modules/packages/tensorrt-samples/testers/sample_non_zero_plugin.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28

{
  atLeast,
  lib,
  mkTester,
  sample-data,
  ...
}:
lib.optionalAttrs (atLeast "10") (
  {
    default = mkTester "sample_non_zero_plugin" [
      "sample_non_zero_plugin"
      "--datadir=${sample-data.outPath + "/mnist"}"
    ];

    fp16 = mkTester "sample_non_zero_plugin-fp16" [
      "sample_non_zero_plugin"
      "--datadir=${sample-data.outPath + "/mnist"}"
      "--fp16"
    ];
  }
  // lib.optionalAttrs (atLeast "10.0.1") {
    columnOrder = mkTester "sample_non_zero_plugin-columnOrder" [
      "sample_non_zero_plugin"
      "--datadir=${sample-data.outPath + "/mnist"}"
      "--columnOrder"
    ];
  }
)