Internal changes to non-public code.

PiperOrigin-RevId: 694185819
google · Nov 7, 2024 · 1779abd · 1779abd
1 parent 23efce6
commit 1779abd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/runtime.c b/src/runtime.c
@@ -643,7 +643,8 @@ enum xnn_status xnn_create_runtime_v4(
 #endif
   if (use_slinky) {
     #ifdef XNN_SLINKY_AVAILABLE
-    // slinky_init_pipeline(runtime);
+    const int max_threads = 16;
+    // slinky_init_pipeline(runtime, max_threads);
     #else
     xnn_log_warning("Slinky requested but not available");
     #endif

diff --git a/src/xnnpack/subgraph.h b/src/xnnpack/subgraph.h
@@ -47,7 +47,7 @@ extern "C" {
 struct slinky_pipeline;
 typedef struct slinky_pipeline* slinky_pipeline_t;
 
-void slinky_init_pipeline(xnn_runtime_t runtime);
+void slinky_init_pipeline(xnn_runtime_t runtime, int max_threads);
 void slinky_setup_inputs_and_outputs(xnn_runtime_t runtime);
 void slinky_destroy_pipeline(xnn_runtime_t runtime);
 bool slinky_evaluate(xnn_runtime_t runtime, enum xnn_status* status);