Add 'projects/rocprofiler-systems/' from commit '92e1d84c72c9321d79a1866e0090fae0215e6557'

git-subtree-dir: projects/rocprofiler-systems git-subtree-mainline: ee9e74df21 git-subtree-split: 92e1d84c72
2025-07-17 18:13:44 +00:00
parent ee9e74df21 92e1d84c72
commit 6755fa3a36
699 changed files with 179131 additions and 0 deletions
@@ -0,0 +1,137 @@
+
+#include <rocprofiler-systems/user.h>
+
+#include <chrono>
+#include <cstdio>
+#include <cstdlib>
+#include <cstring>
+#include <pthread.h>
+#include <set>
+#include <string>
+#include <sys/wait.h>
+#include <thread>
+#include <unistd.h>
+#include <vector>
+
+void
+print_info(const char* _name)
+{
+    fflush(stdout);
+    fflush(stderr);
+    printf("[%s] pid = %i, ppid = %i\n", _name, getpid(), getppid());
+    fflush(stdout);
+    fflush(stderr);
+}
+
+int
+run(const char* _name, int nchildren)
+{
+    auto _barrier  = pthread_barrier_t{};
+    auto _threads  = std::vector<std::thread>{};
+    auto _children = std::vector<pid_t>{};
+    _children.resize(nchildren, 0);
+    pthread_barrier_init(&_barrier, nullptr, nchildren + 1);
+    for(int i = 0; i < nchildren; ++i)
+    {
+        rocprofsys_user_push_region("launch_child");
+        auto _run = [&_barrier, &_children, i, _name](uint64_t _nsec) {
+            pthread_barrier_wait(&_barrier);
+            _children.at(i) = fork();
+            if(_children.at(i) == 0)
+            {
+                // child code
+                print_info(_name);
+                printf("[%s][%i] child job starting...\n", _name, getpid());
+                auto _sleep = [=]() {
+                    rocprofsys_user_push_region("child_process_child_thread");
+                    std::this_thread::sleep_for(std::chrono::seconds{ _nsec });
+                    rocprofsys_user_pop_region("child_process_child_thread");
+                };
+                rocprofsys_user_push_region("child_process");
+                std::thread{ _sleep }.join();
+                rocprofsys_user_push_region("child_process");
+                printf("[%s][%i] child job complete\n", _name, getpid());
+                exit(EXIT_SUCCESS);
+            }
+            else
+            {
+                pthread_barrier_wait(&_barrier);
+            }
+        };
+        _threads.emplace_back(_run, i + 1);
+        rocprofsys_user_pop_region("launch_child");
+    }
+
+    // all child threads should start executing their fork once this returns
+    pthread_barrier_wait(&_barrier);
+    // wait for the threads to successfully fork
+    pthread_barrier_wait(&_barrier);
+
+    rocprofsys_user_push_region("wait_for_children");
+
+    int   _status   = 0;
+    pid_t _wait_pid = 0;
+    // parent waits for all the child processes
+    for(auto& itr : _children)
+    {
+        while(itr == 0)
+        {
+        }
+        printf("[%s][%i] performing waitpid(%i, ...)\n", _name, getpid(), itr);
+        while((_wait_pid = waitpid(itr, &_status, WUNTRACED | WNOHANG)) <= 0)
+        {
+            if(_wait_pid == 0) continue;
+
+            printf("[%s][%i] returned from waitpid(%i) with pid = %i (status = %i) :: ",
+                   _name, getpid(), itr, _wait_pid, _status);
+            if(WIFEXITED(_status))
+            {
+                printf("exited, status=%d\n", WEXITSTATUS(_status));
+            }
+            else if(WIFSIGNALED(_status))
+            {
+                printf("killed by signal %d\n", WTERMSIG(_status));
+            }
+            else if(WIFSTOPPED(_status))
+            {
+                printf("stopped by signal %d\n", WSTOPSIG(_status));
+            }
+            else if(WIFCONTINUED(_status))
+            {
+                printf("continued\n");
+            }
+            else
+            {
+                printf("unknown\n");
+            }
+        }
+    }
+
+    printf("[%s][%i] joining threads ...\n", _name, getpid());
+    for(auto& itr : _threads)
+        itr.join();
+
+    rocprofsys_user_pop_region("wait_for_children");
+
+    printf("[%s][%i] returning (error code: %i) ...\n", _name, getpid(), _status);
+    return _status;
+}
+
+int
+main(int argc, char** argv)
+{
+    int _nfork = 4;
+    int _nrep  = 1;
+    if(argc > 1) _nfork = std::stoi(argv[1]);
+    if(argc > 2) _nrep = std::stoi(argv[2]);
+
+    print_info(argv[0]);
+    for(int i = 0; i < _nrep; ++i)
+    {
+        auto _ec = run(argv[0], _nfork);
+        if(_ec != 0) return _ec;
+    }
+
+    printf("[%s][%i] job complete\n", argv[0], getpid());
+    return EXIT_SUCCESS;
+}