cudaPackages.tensorrt-samples.passthru.testers: properly guard bf16 and dla testers (f01183a9) · Commits · nix / nixpkgs

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

+5 −3

Original line number	Diff line number	Diff line
		{
		atLeast,
		backendStdenv,
		lib,
		mkTester,
		@@ -26,15 +27,16 @@ lib.optionalAttrs (older "10.8") (
		"--datadir=${sample-data.outPath + "/mnist"}"
		"--fp16"
		];

		}
		// lib.optionalAttrs (atLeast "10") {
		bf16 = mkTester "sample_algorithm_selector-bf16" [
		"sample_algorithm_selector"
		"--datadir=${sample-data.outPath + "/mnist"}"
		"--bf16"
		];
		}
		# Only Orin has a DLA
		// lib.optionalAttrs (lib.elem "8.7" backendStdenv.cudaCapabilities) {
		# Only Xavier and Orin have a DLA
		// lib.optionalAttrs (lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ]) {
		dla = mkTester "sample_algorithm_selector-dla" [
		"sample_algorithm_selector"
		"--datadir=${sample-data.outPath + "/mnist"}"

+4 −1

Original line number	Diff line number	Diff line
		{
		atLeast,
		lib,
		mkTester,
		sample-data,
		...
		@@ -21,7 +23,8 @@
		"--datadir=${sample-data.outPath + "/mnist"}"
		"--fp16"
		];

		}
		// lib.optionalAttrs (atLeast "10") {
		bf16 = mkTester "sample_dynamic_reshape-bf16" [
		"sample_dynamic_reshape"
		"--datadir=${sample-data.outPath + "/mnist"}"

+2 −2

Original line number	Diff line number	Diff line
		@@ -12,8 +12,8 @@
		"--data=${sample-data.outPath + "/int8_api"}"
		];
		}
		# Only Orin has a DLA
		// lib.optionalAttrs (lib.elem "8.7" backendStdenv.cudaCapabilities) {
		# 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"}"

+2 −2

Original line number	Diff line number	Diff line
		@@ -11,8 +11,8 @@
		"--datadir=${sample-data.outPath + "/mnist"}"
		];
		}
		# Only Orin has a DLA
		// lib.optionalAttrs (lib.elem "8.7" backendStdenv.cudaCapabilities) {
		# Only Xavier and Orin have a DLA
		// lib.optionalAttrs (lib.subtractLists [ "7.2" "8.7" ] backendStdenv.cudaCapabilities == [ ]) {
		dla = mkTester "sample_io_formats-dla" [
		"sample_io_formats"
		"--datadir=${sample-data.outPath + "/mnist"}"

+20 −19

Original line number	Diff line number	Diff line
		{
		atLeast,
		finalAttrs,
		lib,
		mkTester,
		sample-data,
		...
		}:
		lib.optionalAttrs (atLeast "10") (
		{
		default = mkTester "sample_non_zero_plugin" [
		"sample_non_zero_plugin"
		@@ -25,3 +25,4 @@
		"--columnOrder"
		];
		}
		)