tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 backendStdenv,
3 lib,
4 mkTester,
5 sample-data,
6 ...
7}:
8{
9 default = mkTester "sample_int8_api" [
10 "sample_int8_api"
11 "--model=${sample-data.outPath + "/resnet50/ResNet50.onnx"}"
12 "--data=${sample-data.outPath + "/int8_api"}"
13 ];
14}
15# Only Xavier and Orin have a DLA
16// lib.optionalAttrs (lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ]) {
17 dla = mkTester "sample_int8_api-dla" [
18 "sample_int8_api"
19 "--model=${sample-data.outPath + "/resnet50/ResNet50.onnx"}"
20 "--data=${sample-data.outPath + "/int8_api"}"
21 "--useDLACore=0"
22 ];
23}