Skip to content

T
tic
Commit e374abf8 by Pariksheet Pinjari Committed by Tianqi Chen
Options

Bug fix for Android platforms (https://github.com/dmlc/tvm/pull/971) (#986)

parent f11f6b4e
Showing with 6 additions and 0 deletions
src/runtime/thread_pool.cc
...@@ -355,6 +355,7 @@ class ThreadPool { ...@@ -355,6 +355,7 @@ class ThreadPool {
// bind worker threads to disjoint cores // bind worker threads to disjoint cores
void SetThreadAffinity() { void SetThreadAffinity() {
#if defined(__ANDROID__) #if defined(__ANDROID__)
#ifndef CPU_SET
#define CPU_SETSIZE 1024 #define CPU_SETSIZE 1024
#define __NCPUBITS (8 * sizeof (uint64_t)) #define __NCPUBITS (8 * sizeof (uint64_t))
typedef struct { typedef struct {
...@@ -366,14 +367,19 @@ class ThreadPool { ...@@ -366,14 +367,19 @@ class ThreadPool {
#define CPU_ZERO(cpusetp) \ #define CPU_ZERO(cpusetp) \
memset((cpusetp), 0, sizeof(cpu_set_t)) memset((cpusetp), 0, sizeof(cpu_set_t))
#endif #endif
#endif
for (int i=0; i < num_workers_; ++i) { for (int i=0; i < num_workers_; ++i) {
#if defined(__linux__) || defined(__ANDROID__) #if defined(__linux__) || defined(__ANDROID__)
cpu_set_t cpuset; cpu_set_t cpuset;
CPU_ZERO(&cpuset); CPU_ZERO(&cpuset);
CPU_SET(i, &cpuset); CPU_SET(i, &cpuset);
#if defined(__ANDROID__)
sched_setaffinity(threads_[i].native_handle(), sizeof(cpu_set_t), &cpuset);
#else
pthread_setaffinity_np(threads_[i].native_handle(), pthread_setaffinity_np(threads_[i].native_handle(),
sizeof(cpu_set_t), &cpuset); sizeof(cpu_set_t), &cpuset);
#endif #endif
#endif
} }
} }
// Number of workers // Number of workers
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment