blob: 55866b2e5b4845b01bc4a6ff5f8c6bba1a3867a6 (
plain)
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
29
30
31
32
33
34
35
36
|
{
backendStdenv,
lib,
mkTester,
sample-data,
...
}:
# NOTE: Disabled because it generates way too much output.
lib.optionalAttrs false (
{
default = mkTester "sample_progress_monitor" [
"sample_progress_monitor"
"--datadir=${sample-data.outPath + "/mnist"}"
];
int8 = mkTester "sample_progress_monitor-int8" [
"sample_progress_monitor"
"--datadir=${sample-data.outPath + "/mnist"}"
"--int8"
];
fp16 = mkTester "sample_progress_monitor-fp16" [
"sample_progress_monitor"
"--datadir=${sample-data.outPath + "/mnist"}"
"--fp16"
];
}
# Only Xavier and Orin have a DLA
// lib.optionalAttrs (lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ]) {
dla = mkTester "sample_progress_monitor-dla" [
"sample_progress_monitor"
"--datadir=${sample-data.outPath + "/mnist"}"
"--useDLACore=0"
];
}
)
|
