aboutsummaryrefslogtreecommitdiff
path: root/src/core/platform.h
diff options
context:
space:
mode:
Diffstat (limited to 'src/core/platform.h')
-rw-r--r--src/core/platform.h4
1 files changed, 4 insertions, 0 deletions
diff --git a/src/core/platform.h b/src/core/platform.h
index 07bfc14c..b709e3ba 100644
--- a/src/core/platform.h
+++ b/src/core/platform.h
@@ -190,6 +190,10 @@ extern int nni_plat_init(int (*)(void));
// will be called until nni_platform_init is called.
extern void nni_plat_fini(void);
+// nni_plat_ncpu returns the number of logical CPUs on the system. This is
+// used to scale the number of independent threads started.
+extern int nni_plat_ncpu(void);
+
//
// TCP Support.
//