microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-19 14:29:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Construct ProxyChannel with shared pointers (#184)

Browse Source
This commit is contained in:
Changho Hwang
2023-09-18 13:46:23 +08:00
committed by GitHub
parent a6b24dcbed
commit 6c0ee72916
5 changed files with 29 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/mscclpp/proxy_channel_py.cpp
View File

@@ -26,8 +26,8 @@ void register_proxy_channel(nb::module_& m) {
.def("proxy_channel", &ProxyService::proxyChannel, nb::arg("id"));
nb::class_<ProxyChannel>(m, "ProxyChannel")
.def(nb::init<SemaphoreId, Host2DeviceSemaphore::DeviceHandle, FifoDeviceHandle>(), nb::arg("semaphoreId"),
nb::arg("semaphore"), nb::arg("fifo"))
.def(nb::init<SemaphoreId, std::shared_ptr<Host2DeviceSemaphore>, std::shared_ptr<Proxy>>(),
nb::arg("semaphoreId"), nb::arg("semaphore"), nb::arg("proxy"))
.def("device_handle", &ProxyChannel::deviceHandle);
nb::class_<ProxyChannel::DeviceHandle>(m, "ProxyChannelDeviceHandle")