Handle fork in target application (#191)

* Always print PID in log messages

* omnitrace-dl updates

- omnitrace_preload does not call omnitrace_init or omnitrace_init_tooling
- omnitrace_preload will call omnitrace_set_mpi if OMNITRACE_USE_MPI
  or OMNITRACE_USE_MPIP in the env is true but not call it otherwise
  because doing so either overrides OMNITRACE_USE_PID (when true) or
  disable mpip from initialization (when false) and the MPI
  init can be caught later and override OMNITRACE_USE_PID

* config updates

- set_setting_value sets user update type
- remove volatile from get_settings_configured
- don't override settings::default_process_suffix
- don't kill process in omnitrace_exit_action
- set_state ignores updating state if >= State::Finalized

* Handle state > State::Finalized

* fork gotcha updates

- unsets LD_PRELOAD
- sets OMNITRACE_ROOT_PROCESS
- sets OMNITRACE_CHILD_PROCESS

* libomnitrace library.cpp updates

- basic_bundle for fini metrics
- handle finalization from child process

* sampling updates

- sampling::shutdown handles when child process

* Add example and test using fork

* Update run-ci script to support not submitting

* Tweak test envs

* Update build flags when codecov enabled

* remove unnecessary includes of sampling header

* Replace mpi copy/fini static lambda with free-funcs

* Update codecov job

* Fix OMPT segfaults after finalization

* Miscellaneous updates after rebase

* fixes for causal profiling

* revert some run-ci.sh changes

* Disable storing env in sampling::shutdown

* formatting fix

* Update timemory submodule

- fixed occasional synchronization issues with allocator offloading
- exclude protozero:: from internal samples

* improve root/child process detection

- avoid omnitrace_finalize in MPI when child process
- revert some testing tweaks
This commit is contained in:
Jonathan R. Madsen
2023-02-08 01:31:38 -06:00
zatwierdzone przez GitHub
rodzic 0da62c980e
commit 32b15fe7b7
39 zmienionych plików z 440 dodań i 133 usunięć
+46 -5
Wyświetl plik
@@ -25,6 +25,7 @@
#include "state.hpp"
#include <timemory/log/color.hpp>
#include <timemory/process/threading.hpp>
#include <timemory/utility/filepath.hpp>
#include <iomanip>
@@ -43,6 +44,29 @@ struct source_location_history
size_t size = 0;
};
const std::string&
get_file_name()
{
static auto _fname = tim::get_env<std::string>("OMNITRACE_LOG_FILE", "");
return _fname;
}
std::atomic<FILE*>&
get_file_pointer()
{
static auto _v = std::atomic<FILE*>{ []() {
const auto&_fname= get_file_name();
if(!_fname.empty()) tim::log::monochrome() = true;
return (_fname.empty())
? stderr
: filepath::fopen(
settings::format(_fname, filepath::basename(filepath::realpath(
"/proc/self/exe", nullptr, false))),
"w");
}() };
return _v;
}
auto&
get_source_location_history()
{
@@ -88,11 +112,28 @@ lock::~lock()
FILE*
get_file()
{
static FILE* _v = []() {
auto&& _fname = tim::get_env<std::string>("OMNITRACE_LOG_FILE", "");
if(!_fname.empty()) tim::log::monochrome() = true;
return (_fname.empty()) ? stderr : tim::filepath::fopen(_fname, "w");
}();
return get_file_pointer();
}
void
close_file()
{
if(get_file() != stderr)
{
auto* _file = get_file_pointer().load();
get_file_pointer().store(stderr);
fclose(_file);
// Write the trace into a file.
if(get_verbose() >= 0)
operation::file_output_message<tim::project::omnitrace>{}(
get_file_name(), std::string{ "debug" });
}
}
int64_t
get_tid()
{
static thread_local auto _v = threading::get_id();
return _v;
}
} // namespace debug