diff options
Diffstat (limited to 'src/data/corpus.h')
-rw-r--r-- | src/data/corpus.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/src/data/corpus.h b/src/data/corpus.h index e8e9a9fd..281d43a2 100644 --- a/src/data/corpus.h +++ b/src/data/corpus.h @@ -4,6 +4,7 @@ #include <iostream> #include <random> +#include "3rd_party/threadpool.h" #include "common/definitions.h" #include "common/file_stream.h" #include "common/options.h" @@ -20,6 +21,8 @@ class Corpus : public CorpusBase { private: std::vector<UPtr<io::TemporaryFile>> tempFiles_; std::vector<size_t> ids_; + + UPtr<ThreadPool> threadPool_; // thread pool for parallelized data reading // for shuffle-in-ram bool shuffleInRAM_{false}; |