AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update to KOKKOS_TOOLS_LIBS env var (#69)

Browse Source
This commit is contained in:
Maarten Arnst
2024-12-20 01:20:54 +01:00
committed by Sajina Kandy
parent 27f210ea5d
commit 043a8010a9
6 changed files with 14 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
source/lib/core/argparse.cpp
+2 -2
View File
@@ -616,7 +616,7 @@ add_core_arguments(parser_t& _parser, parser_data& _data)
UPD_PREPEND);
if(_v.count("all") > 0 || _v.count("kokkosp") > 0)
update_env(_data, "KOKKOS_PROFILE_LIBRARY", _data.omni_libpath,
update_env(_data, "KOKKOS_TOOLS_LIBS", _data.omni_libpath,
UPD_PREPEND);
});
@@ -649,7 +649,7 @@ add_core_arguments(parser_t& _parser, parser_data& _data)
remove_env(_data, "OMP_TOOL_LIBRARIES");
if(_v.count("all") > 0 || _v.count("kokkosp") > 0)
remove_env(_data, "KOKKOS_PROFILE_LIBRARY");
remove_env(_data, "KOKKOS_TOOLS_LIBS");
});
_data.processed_environs.emplace("exclude");
source/lib/core/config.cpp
+3 -3
View File
@@ -1137,7 +1137,7 @@ configure_mode_settings(const std::shared_ptr<settings>& _config)
if(_config->get<bool>("ROCPROFSYS_USE_KOKKOSP"))
{
auto _current_kokkosp_lib = tim::get_env<std::string>("KOKKOS_PROFILE_LIBRARY");
auto _current_kokkosp_lib = tim::get_env<std::string>("KOKKOS_TOOLS_LIBS");
if(_current_kokkosp_lib.find("librocprof-sys-dl.so") == std::string::npos &&
_current_kokkosp_lib.find("librocprof-sys.so") == std::string::npos)
{
@@ -1149,9 +1149,9 @@ configure_mode_settings(const std::shared_ptr<settings>& _config)
_message =
JOIN("", " (forced. Previous value: '", _current_kokkosp_lib, "')");
}
ROCPROFSYS_BASIC_VERBOSE_F(1, "Setting KOKKOS_PROFILE_LIBRARY=%s%s\n",
ROCPROFSYS_BASIC_VERBOSE_F(1, "Setting KOKKOS_TOOLS_LIBS=%s%s\n",
"librocprof-sys.so", _message.c_str());
tim::set_env("KOKKOS_PROFILE_LIBRARY", "librocprof-sys.so", _force);
tim::set_env("KOKKOS_TOOLS_LIBS", "librocprof-sys.so", _force);
}
}