saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-23 19:32:23 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

add path for _kernel_explorer.so

Browse source
This commit is contained in:
Tianlei Wu 2025-02-01 00:01:46 +00:00
parent 93d73bf831
commit 714c812e36
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/python/tools/kernel_explorer/kernels/kernel_explorer.py
View file

@ -26,7 +26,10 @@ if build_dir is None:
if not os.path.exists(build_dir):
raise ValueError(f"KERNEL_EXPLORER_BUILD_DIR ({build_dir}) points to nonexistent path")
# add path for onnxruntime_pybind11_state
# add path for _kernel_explorer.so
sys.path.insert(0, os.path.join(build_dir))
# add path for onnxruntime_pybind11_state.so
sys.path.insert(0, os.path.join(build_dir, "onnxruntime", "capi"))
# pylint: disable=wrong-import-position