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

Add automatic PyTorch library discovery for Python applications (#2623)

Browse Source 
* Add automatic PyTorch library discovery for Python applications (#2623)
This commit is contained in:
Milan Radosavljevic
2026-01-20 08:42:49 +01:00
committed by GitHub
parent c83b3aae07
commit b533f56197
9 changed files with 425 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rocprofiler-systems/source/bin/rocprof-sys-sample/impl.cpp
+6
View File
@@ -933,3 +933,9 @@ parse_args(int argc, char** argv, std::vector<char*>& _env)
return _outv;
}
void
add_torch_library_path(std::vector<char*>& envp, const std::vector<char*>& argv)
{
rocprofsys::common::add_torch_library_path(envp, argv, verbose > 0, updated_envs);
}
projects/rocprofiler-systems/source/bin/rocprof-sys-sample/rocprof-sys-sample.cpp
+2
View File
@@ -51,6 +51,8 @@ main(int argc, char** argv)
_argv.emplace_back(argv[i]);
}
add_torch_library_path(_env, _argv);
print_updated_environment(_env);
if(!_argv.empty())
projects/rocprofiler-systems/source/bin/rocprof-sys-sample/rocprof-sys-sample.hpp
+3
View File
@@ -35,3 +35,6 @@ get_initial_environment();
std::vector<char*>
parse_args(int argc, char** argv, std::vector<char*>& envp);
void
add_torch_library_path(std::vector<char*>& envp, const std::vector<char*>& argv);