mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-12 09:17:06 +00:00
e7d5e652df
Co-authored-by: Olli Saarikivi <olsaarik@microsoft.com> Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Binyang Li <binyli@microsoft.com>
26 lines
863 B
C++
26 lines
863 B
C++
// Copyright (c) Microsoft Corporation.
|
|
// Licensed under the MIT license.
|
|
|
|
#include <nanobind/nanobind.h>
|
|
|
|
#include <mscclpp/fifo.hpp>
|
|
|
|
namespace nb = nanobind;
|
|
using namespace mscclpp;
|
|
|
|
void register_fifo(nb::module_& m) {
|
|
nb::class_<ProxyTrigger>(m, "ProxyTrigger").def_rw("fst", &ProxyTrigger::fst).def_rw("snd", &ProxyTrigger::snd);
|
|
|
|
nb::class_<DeviceProxyFifo>(m, "DeviceProxyFifo")
|
|
.def_rw("triggers", &DeviceProxyFifo::triggers)
|
|
.def_rw("tail_replica", &DeviceProxyFifo::tailReplica)
|
|
.def_rw("head", &DeviceProxyFifo::head);
|
|
|
|
nb::class_<HostProxyFifo>(m, "HostProxyFifo")
|
|
.def(nb::init<>())
|
|
.def("poll", &HostProxyFifo::poll, nb::arg("trigger"))
|
|
.def("pop", &HostProxyFifo::pop)
|
|
.def("flush_tail", &HostProxyFifo::flushTail, nb::arg("sync") = false)
|
|
.def("device_fifo", &HostProxyFifo::deviceFifo);
|
|
}
|