Add automatic PyTorch library discovery for Python applications (#2623)
* Add automatic PyTorch library discovery for Python applications (#2623)
Этот коммит содержится в:
1 родитель
c83b3aae07
Коммит
b533f56197
9 files changed
+425
-2
No files matched your search
@@ -192,6 +192,10 @@ prepare_environment_for_run(parser_data_t& _data)
|
||||
rocprofsys::argparse::add_ld_preload(_data);
|
||||
rocprofsys::argparse::add_ld_library_path(_data);
|
||||
}
|
||||
|
||||
rocprofsys::argparse::add_torch_library_path(_data, _data.verbose > 0);
|
||||
|
||||
rocprofsys::common::consolidate_env_entries(_data.current);
|
||||
}
|
||||
|
||||
void
|
||||
|
||||
Reference in new issue
Block a user