read subatches in parallel

marius-team · Nov 21, 2023 · cd55db7 · cd55db7
1 parent 1a071e6
commit cd55db7
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/cpp/src/pipeline/pipeline_gpu.cpp b/src/cpp/src/pipeline/pipeline_gpu.cpp
@@ -151,6 +151,7 @@ void BatchToDeviceWorker::run() {
             }
 
             if (batch->sub_batches_.size() > 0) {
+                #pragma omp parallel for
                 for (int i = 0; i < batch->sub_batches_.size(); i++) {
                     if (!batch->sub_batches_[i]->node_features_.defined())
                         batch->sub_batches_[i]->node_features_ = pipeline_->dataloader_->graph_storage_->getNodeFeatures(batch->sub_batches_[i]->unique_node_indices_);