diff --git a/src/runtime.c b/src/runtime.c index bc29215b0bf..169bed43a79 100644 --- 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 index 5f18f507bbe..fba387405c3 100644 --- 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);