Create API to determine how many threads to useHEADmaster

PiperOrigin-RevId: 474367386

1 files changed, 9 insertions, 0 deletions

diff --git a/ruy/ctx_test.cc b/ruy/ctx_test.cc
index e55dcfc..c40f2d6 100644
--- a/ruy/ctx_test.cc
+++ b/ruy/ctx_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "ruy/ctx_impl.h"
 #include "ruy/gtest_wrapper.h"
+#include "ruy/strategy_controls.h"
 #include "ruy/path.h"
 #include "ruy/platform.h"
 
@@ -67,6 +68,14 @@ TEST(ContextInternalTest, ThreadSpecificResources) {
   }
 }
 
+TEST(ContextInternalTest, SetNumThreadsStrategy) {
+  CtxImpl ctx;
+  EXPECT_EQ(ctx.num_threads_strategy(), NumThreadsStrategy::kDefault);
+  ctx.set_num_threads_strategy(NumThreadsStrategy::kForceMaxNumThreads);
+  EXPECT_EQ(ctx.num_threads_strategy(),
+            NumThreadsStrategy::kForceMaxNumThreads);
+}
+
 }  // namespace
 }  // namespace ruy