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. // Split into halves and submit to the pool.
Index mid = first + divup((last - first) / 2, block_size) * block_size; Index mid = first + divup((last - first) / 2, block_size) * block_size;
enqueue_func([=, &handleRange]() { handleRange(mid, last); }); pool_->Schedule([=, &handleRange]() { handleRange(mid, last); });
enqueue_func([=, &handleRange]() { handleRange(first, mid); }); pool_->Schedule([=, &handleRange]() { handleRange(first, mid); });
}; };
handleRange(0, n); handleRange(0, n);
barrier.Wait(); barrier.Wait();