timemory submodule (#1)

- library now wraps fork() to warn about fork + OpenMPI

[ROCm/rocprofiler-systems commit: fc7cc1e68f]

projects/rocprofiler-systems/examples/transpose/CMakeLists.txt

@@ -20,6 +20,7 @@ endif()
 
 string(TOUPPER "${CMAKE_BUILD_TYPE}" BUILD_TYPE)
 set(transpose_CXX_FLAGS "-W -Wall ${CMAKE_CXX_FLAGS_${BUILD_TYPE}} ${CMAKE_CXX_FLAGS}")
+set(transpose_LINK_FLAGS "")
 
 if(CMAKE_CXX_COMPILER_IS_CLANG AND TARGET hosttrace::hosttrace-compile-options)
     set(transpose_CXX_FLAGS "${transpose_CXX_FLAGS} ${hosttrace_CXX_FLAGS}")
@@ -27,6 +28,40 @@ if(CMAKE_CXX_COMPILER_IS_CLANG AND TARGET hosttrace::hosttrace-compile-options)
     set(transpose_CXX_FLAGS "${transpose_CXX_FLAGS} ${COMPILE_OPTS}")
 endif()
 
+option(TRANSPOSE_USE_MPI "Enable MPI support in transpose exe" ${TIMEMORY_USE_MPI})
+
+if(TRANSPOSE_USE_MPI)
+    find_package(MPI REQUIRED)
+endif()
+
+if(TARGET MPI::MPI_C)
+    set(transpose_CXX_FLAGS "${transpose_CXX_FLAGS} -DUSE_MPI")
+    get_target_property(COMPILE_OPTS MPI::MPI_C INTERFACE_COMPILE_OPTIONS)
+    foreach(_COPT ${COMPILE_OPTS})
+        if(NOT "${_COPT}" MATCHES "COMPILE_LANG")
+            set(transpose_CXX_FLAGS "${transpose_CXX_FLAGS} ${_COPT}")
+        endif()
+    endforeach()
+    get_target_property(INCLUDE_DIRS MPI::MPI_C INTERFACE_INCLUDE_DIRECTORIES)
+    foreach(_IDIR ${INCLUDE_DIRS})
+        set(transpose_CXX_FLAGS "${transpose_CXX_FLAGS} -I${_IDIR}")
+        get_filename_component(_LIBDIR "${_IDIR}" DIRECTORY)
+        foreach(_LDIR lib lib64)
+            set(_LIBDIR_SAVE "${_LIBDIR}")
+            if(NOT EXISTS "${_LIBDIR}/${_LDIR}")
+                get_filename_component(_LIBDIR "${_LIBDIR}" DIRECTORY)
+            endif()
+            if(EXISTS "${_LIBDIR}/${_LDIR}")
+                set(transpose_LINK_FLAGS "${transpose_LINK_FLAGS} -L${_LIBDIR}/${_LDIR} -lmpi")
+            endif()
+            set(_LIBDIR "${_LIBDIR_SAVE}")
+        endforeach()
+    endforeach()
+    if(MPI_C_LINK_FLAGS)
+        set(transpose_LINK_FLAGS "${transpose_LINK_FLAGS} ${MPI_C_LINK_FLAGS}")
+    endif()
+endif()
+
 # remove generator expressions
 string(REPLACE "$<" "" transpose_CXX_FLAGS "${transpose_CXX_FLAGS}")
 string(REPLACE ">:" ":" transpose_CXX_FLAGS "${transpose_CXX_FLAGS}")

projects/rocprofiler-systems/examples/transpose/Makefile

@@ -1,11 +1,13 @@
 CXXFLAGS += -std=c++17 -O3 -Wno-unused-result -g
+LINK_FLAGS :=
+
 all: transpose
 
 debug: CXXFLAGS += -ggdb -g
 debug: transpose
 
 transpose: transpose.cpp
-	hipcc ${CXXFLAGS} -o $@ $^ 
+	hipcc ${CXXFLAGS} -o $@ $^ ${LINK_FLAGS}
 
 .PHONY: clean
 

projects/rocprofiler-systems/examples/transpose/Makefile.in

@@ -1,11 +1,13 @@
 CXXFLAGS += -std=c++@CMAKE_CXX_STANDARD@ @transpose_CXX_FLAGS@
+LINKFLAGS := @transpose_LINK_FLAGS@
+
 all: transpose
 
 debug: CXXFLAGS += -ggdb -g
 debug: transpose
 
 transpose: transpose.cpp
-	hipcc ${CXXFLAGS} -o $@ $^
+	hipcc ${CXXFLAGS} -o $@ $^ ${LINKFLAGS}
 
 .PHONY: clean
 

projects/rocprofiler-systems/examples/transpose/transpose.cpp

@@ -148,9 +148,19 @@ run(int argc, char** argv)
     free(out_matrix);
 }
 
+#if defined(USE_MPI)
+#    include <mpi.h>
+#endif
+
 int
 main(int argc, char** argv)
 {
+#if defined(USE_MPI)
+    MPI_Init(&argc, &argv);
+#endif
     run(argc, argv);
+#if defined(USE_MPI)
+    MPI_Finalize();
+#endif
     return 0;
 }