Adding openMP parallelization pragmas

CMakeLists.txt

@@ -4,7 +4,7 @@ project(ir2vec VERSION 2.1.2)
 set(IR2VEC_LIB "IR2Vec")
 set(IR2VEC_LIB_STATIC "IR2Vec_Static")
 
-set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O2")
+set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O2 -pg")
 
 set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_BINARY_DIR}/bin)
 

src/CMakeLists.txt

@@ -24,8 +24,11 @@ if(NOT LLVM_IR2VEC)
   # llvm_map_components_to_libnames(llvm_libs all)
   llvm_map_components_to_libnames(llvm_libs support core irreader analysis TransformUtils)
 
+  # Check if OpenMP is supported by the compiler
+  find_package(OpenMP REQUIRED)
+
   add_executable(${PROJECT_NAME} ${binsrc})
-  target_link_libraries (${PROJECT_NAME} ${llvm_libs} objlib)
+  target_link_libraries (${PROJECT_NAME} ${llvm_libs} objlib OpenMP::OpenMP_CXX)
   target_include_directories(${PROJECT_NAME} PRIVATE .)
 
   add_library(objlib OBJECT ${libsrc})
@@ -50,6 +53,9 @@ if(NOT LLVM_IR2VEC)
       PUBLIC_HEADER DESTINATION include
       RESOURCE DESTINATION ./)
 
+  # Add compiler flags for OpenMP
+  target_compile_options(${PROJECT_NAME} PRIVATE ${OpenMP_CXX_FLAGS})
+
   add_subdirectory(test-suite)
 
   add_custom_target(check

src/CollectIR.cpp

@@ -36,44 +36,32 @@ void CollectIR::traverseBasicBlock(BasicBlock &B) {
 
     if (type->isVoidTy()) {
       stype = " voidTy ";
-      res += stype;
     } else if (type->isFloatingPointTy()) {
       stype = " floatTy ";
-      res += stype;
     } else if (type->isIntegerTy()) {
       stype = " integerTy ";
-      res += stype;
     } else if (type->isFunctionTy()) {
       stype = " functionTy ";
-      res += stype;
     } else if (type->isStructTy()) {
       stype = " structTy ";
-      res += stype;
     } else if (type->isArrayTy()) {
       stype = " arrayTy ";
-      res += stype;
     } else if (type->isPointerTy()) {
       stype = " pointerTy ";
-      res += stype;
     } else if (type->isVectorTy()) {
       stype = " vectorTy ";
-      res += stype;
     } else if (type->isEmptyTy()) {
       stype = " emptyTy ";
-      res += stype;
     } else if (type->isLabelTy()) {
       stype = " labelTy ";
-      res += stype;
     } else if (type->isTokenTy()) {
       stype = " tokenTy ";
-      res += stype;
     } else if (type->isMetadataTy()) {
       stype = " metadataTy ";
-      res += stype;
     } else {
       stype = " unknownTy ";
-      res += stype;
     }
+    res += stype;
 
     IR2VEC_DEBUG(errs() << "Type taken : " << stype << "\n";);