tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{
2 atLeast,
3 lib,
4 mkTester,
5 sample-data,
6 ...
7}:
8{
9 default = mkTester "sample_dynamic_reshape" [
10 "sample_dynamic_reshape"
11 "--datadir=${sample-data.outPath + "/mnist"}"
12 ];
13
14 # TODO(@connorbaker): Neither the sample data nor the sample sources provide train-images-idx3-ubyte, so we can't run the int8 test.
15 # int8 = mkTester "sample_dynamic_reshape-int8" [
16 # "sample_dynamic_reshape"
17 # "--datadir=${sample-data.outPath + "/mnist"}"
18 # "--int8"
19 # ];
20
21 fp16 = mkTester "sample_dynamic_reshape-fp16" [
22 "sample_dynamic_reshape"
23 "--datadir=${sample-data.outPath + "/mnist"}"
24 "--fp16"
25 ];
26}
27// lib.optionalAttrs (atLeast "10") {
28 bf16 = mkTester "sample_dynamic_reshape-bf16" [
29 "sample_dynamic_reshape"
30 "--datadir=${sample-data.outPath + "/mnist"}"
31 "--bf16"
32 ];
33}