Skip to content

Commit 40e3faf

Browse files
author
Wish
committed
add init nv plugins
1 parent af42123 commit 40e3faf

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

python/trtpy/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,7 @@ def set_log_level(level : LogLevel): ...
281281
def get_log_level()->LogLevel: ...
282282
def set_devie(device : int): ...
283283
def get_devie()->int: ...
284+
def init_nv_plugins(): ...
284285

285286
os_name = platform.system()
286287
if os_name == "Windows":

src/application/app_python/interface.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -770,5 +770,6 @@ PYBIND11_MODULE(libtrtpyc, m) {
770770
m.def("set_log_level", [](iLogger::LogLevel level){iLogger::set_log_level(level);});
771771
m.def("get_log_level", [](){return iLogger::get_log_level();});
772772
m.def("random_color", [](int idd){return iLogger::random_color(idd);});
773+
m.def("init_nv_plugins", [](){TRT::init_nv_plugins();});
773774
}
774775
#endif // HAS_PYTHON

0 commit comments

Comments
 (0)