Fixed compilation error in the tensor thread pool

This commit is contained in:
Benoit Steiner 2016-05-17 12:33:46 -07:00
parent 0bb61b04ca
commit ebf6ada5ee

View File

@ -253,8 +253,8 @@ struct ThreadPoolDevice {
}
// Split into halves and submit to the pool.
Index mid = first + divup((last - first) / 2, block_size) * block_size;
enqueue_func([=, &handleRange]() { handleRange(mid, last); });
enqueue_func([=, &handleRange]() { handleRange(first, mid); });
pool_->Schedule([=, &handleRange]() { handleRange(mid, last); });
pool_->Schedule([=, &handleRange]() { handleRange(first, mid); });
};
handleRange(0, n);
barrier.Wait();