microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 09:17:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

update python binding (#136)

Browse Source 
update pythons binding for `device_handle`
This commit is contained in:
Binyang2014
2023-07-24 11:00:33 +08:00
committed by GitHub
parent a58e2e9623
commit 9a488f0da2
5 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/core_py.cpp
View File

@@ -14,6 +14,7 @@ using namespace mscclpp;
extern void register_error(nb::module_& m);
extern void register_proxy_channel(nb::module_& m);
extern void register_sm_channel(nb::module_& m);
extern void register_fifo(nb::module_& m);
extern void register_semaphore(nb::module_& m);
extern void register_config(nb::module_& m);
@@ -145,6 +146,7 @@ void register_core(nb::module_& m) {
NB_MODULE(mscclpp, m) {
register_error(m);
register_proxy_channel(m);
register_sm_channel(m);
register_fifo(m);
register_semaphore(m);
register_config(m);