enable OpenMP for operators and projectors

elsa/operators/CMakeLists.txt

@@ -25,6 +25,13 @@ add_library(elsa::${ELSA_MODULE_NAME} ALIAS ${ELSA_MODULE_TARGET_NAME})
 
 target_link_libraries(${ELSA_MODULE_TARGET_NAME} elsa_core elsa_logging)
 
+# use OpenMP is available
+find_package(OpenMP)
+if(OpenMP_CXX_FOUND)
+    target_link_libraries(${ELSA_MODULE_TARGET_NAME} PRIVATE OpenMP::OpenMP_CXX)
+endif()
+
+
 target_include_directories(${ELSA_MODULE_TARGET_NAME}
         PUBLIC
         $<INSTALL_INTERFACE:include/${ELSA_MODULE_NAME}>

elsa/operators/FiniteDifferences.cpp

@@ -102,6 +102,7 @@ namespace elsa
         IndexVector_t decrementedCoefficients = numberOfCoefficients -
                                                 IndexVector_t::Ones(this->getRangeDescriptor().getNumberOfDimensions());
 
+#pragma omp parallel
         for (int currDim = 0; currDim < numDim; ++currDim) {
             if (!_activeDims[currDim])
                 continue;
@@ -109,6 +110,7 @@ namespace elsa
             index_t modulus = numberOfCoefficients.head(currDim + 1).prod();
             index_t divisor = numberOfCoefficients.head(currDim).prod();
 
+#pragma omp for nowait
             for (index_t id = 0; id < sizeOfDomain; ++id) {
                 index_t icCount = (id % modulus) / divisor; //_domainDescriptor.index(id, ic);
                 index_t ir = id + _dimCounter[currDim] * _coordDelta[currDim];
@@ -147,6 +149,7 @@ namespace elsa
         IndexVector_t numberOfCoefficients = this->getDomainDescriptor().getNumberOfCoefficientsPerDimension();
         IndexVector_t decrementedCoefficients = numberOfCoefficients - IndexVector_t::Ones(numDim);
 
+#pragma omp parallel
         for (index_t currDim = 0; currDim < numDim; ++currDim) {
             if (!_activeDims[currDim])
                 continue;
@@ -154,6 +157,7 @@ namespace elsa
             index_t modulus = numberOfCoefficients.head(currDim + 1).prod();
             index_t divisor = numberOfCoefficients.head(currDim).prod();
 
+#pragma omp for nowait
             for (index_t id = 0; id < sizeOfDomain; ++id) {
                 index_t icCount = (id % modulus) / divisor;
                 index_t ir = id + _dimCounter[currDim] * _coordDelta[currDim];

elsa/projectors/CMakeLists.txt

@@ -35,6 +35,14 @@ add_library(elsa::${ELSA_MODULE_NAME} ALIAS ${ELSA_MODULE_TARGET_NAME})
 
 target_link_libraries(${ELSA_MODULE_TARGET_NAME} elsa_core elsa_logging)
 
+# use OpenMP is available
+find_package(OpenMP)
+if(OpenMP_CXX_FOUND)
+    target_link_libraries(${ELSA_MODULE_TARGET_NAME} PRIVATE OpenMP::OpenMP_CXX)
+endif()
+
+
+
 target_include_directories(${ELSA_MODULE_TARGET_NAME}
         PUBLIC
         $<INSTALL_INTERFACE:include/${ELSA_MODULE_NAME}>