From 075457db003569498b35c44145ec4f7801e419f2 Mon Sep 17 00:00:00 2001
From: Wouter Deconinck <wdconinc@gmail.com>
Date: Mon, 4 Apr 2022 17:21:34 -0500
Subject: [PATCH] Build tests/benchmarks only when self.run_tests (#29862)

---
 var/spack/repos/builtin/packages/cpuinfo/package.py     | 3 +++
 var/spack/repos/builtin/packages/fp16/package.py        | 2 ++
 var/spack/repos/builtin/packages/nnpack/package.py      | 1 +
 var/spack/repos/builtin/packages/pthreadpool/package.py | 2 ++
 var/spack/repos/builtin/packages/qnnpack/package.py     | 2 ++
 var/spack/repos/builtin/packages/xnnpack/package.py     | 2 ++
 6 files changed, 12 insertions(+)

diff --git a/var/spack/repos/builtin/packages/cpuinfo/package.py b/var/spack/repos/builtin/packages/cpuinfo/package.py
index 9ddda0ceed3..045676967e7 100644
--- a/var/spack/repos/builtin/packages/cpuinfo/package.py
+++ b/var/spack/repos/builtin/packages/cpuinfo/package.py
@@ -48,4 +48,7 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'googletest')),
             self.define('GOOGLEBENCHMARK_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
+            self.define('CPUINFO_BUILD_UNIT_TESTS', self.run_tests),
+            self.define('CPUINFO_BUILD_MOCK_TESTS', self.run_tests),
+            self.define('CPUINFO_BUILD_BENCHMARKS', self.run_tests),
         ]
diff --git a/var/spack/repos/builtin/packages/fp16/package.py b/var/spack/repos/builtin/packages/fp16/package.py
index a953bb73dcf..45445545efb 100644
--- a/var/spack/repos/builtin/packages/fp16/package.py
+++ b/var/spack/repos/builtin/packages/fp16/package.py
@@ -54,4 +54,6 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'googletest')),
             self.define('GOOGLEBENCHMARK_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
+            self.define('FP16_BUILD_TESTS', self.run_tests),
+            self.define('FP16_BUILD_BENCHMARKS', self.run_tests),
         ]
diff --git a/var/spack/repos/builtin/packages/nnpack/package.py b/var/spack/repos/builtin/packages/nnpack/package.py
index ed0f21a36e5..9a0d1c7b88b 100644
--- a/var/spack/repos/builtin/packages/nnpack/package.py
+++ b/var/spack/repos/builtin/packages/nnpack/package.py
@@ -115,4 +115,5 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'pthreadpool')),
             self.define('GOOGLETEST_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'googletest')),
+            self.define('NNPACK_BUILD_TESTS', self.run_tests),
         ]
diff --git a/var/spack/repos/builtin/packages/pthreadpool/package.py b/var/spack/repos/builtin/packages/pthreadpool/package.py
index 6b75b13adc8..93ce659e252 100644
--- a/var/spack/repos/builtin/packages/pthreadpool/package.py
+++ b/var/spack/repos/builtin/packages/pthreadpool/package.py
@@ -57,4 +57,6 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'googletest')),
             self.define('GOOGLEBENCHMARK_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
+            self.define('PTHREADPOOL_BUILD_TESTS', self.run_tests),
+            self.define('PTHREADPOOL_BUILD_BENCHMARKS', self.run_tests),
         ]
diff --git a/var/spack/repos/builtin/packages/qnnpack/package.py b/var/spack/repos/builtin/packages/qnnpack/package.py
index 4b2f2f2e778..40c93b5eebd 100644
--- a/var/spack/repos/builtin/packages/qnnpack/package.py
+++ b/var/spack/repos/builtin/packages/qnnpack/package.py
@@ -87,4 +87,6 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
             self.define('GOOGLETEST_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'googletest')),
+            self.define('QNNPACK_BUILD_TESTS', self.run_tests),
+            self.define('QNNPACK_BUILD_BENCHMARKS', self.run_tests),
         ]
diff --git a/var/spack/repos/builtin/packages/xnnpack/package.py b/var/spack/repos/builtin/packages/xnnpack/package.py
index 8c982e3ebe8..9bea7b21c7e 100644
--- a/var/spack/repos/builtin/packages/xnnpack/package.py
+++ b/var/spack/repos/builtin/packages/xnnpack/package.py
@@ -101,4 +101,6 @@ def cmake_args(self):
                         join_path(self.stage.source_path, 'deps', 'googlebenchmark')),
             self.define('PSIMD_SOURCE_DIR',
                         join_path(self.stage.source_path, 'deps', 'psimd')),
+            self.define('XNNPACK_BUILD_TESTS', self.run_tests),
+            self.define('XNNPACK_BUILD_BENCHMARKS', self.run_tests),
         ]