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


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

{
  backendStdenv,
  lib,
  mkTester,
  sample-data,
  ...
}:
{
  default = mkTester "sample_int8_api" [
    "sample_int8_api"
    "--model=${sample-data.outPath + "/resnet50/ResNet50.onnx"}"
    "--data=${sample-data.outPath + "/int8_api"}"
  ];
}
# Only Xavier and Orin have a DLA
// lib.optionalAttrs (lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ]) {
  dla = mkTester "sample_int8_api-dla" [
    "sample_int8_api"
    "--model=${sample-data.outPath + "/resnet50/ResNet50.onnx"}"
    "--data=${sample-data.outPath + "/int8_api"}"
    "--useDLACore=0"
  ];
}