mirror of
https://github.com/microsoft/mscclpp.git
synced 2026-05-11 17:00:22 +00:00
8d1b984bed
* Changed device handle interfaces * Changed proxy service interfaces * Move device code into separate files * Fixed FIFO polling issues * Add configuration arguments in several interface functions --------- Co-authored-by: Changho Hwang <changhohwang@microsoft.com> Co-authored-by: Binyang Li <binyli@microsoft.com> Co-authored-by: root <root@a100-saemal0.qxveptpukjsuthqvv514inp03c.gx.internal.cloudapp.net>
14 lines
380 B
C++
14 lines
380 B
C++
#include <nanobind/nanobind.h>
|
|
namespace nb = nanobind;
|
|
|
|
namespace mscclpp {
|
|
int getDeviceNumaNode(int cudaDev);
|
|
void numaBind(int node);
|
|
}; // namespace mscclpp
|
|
|
|
void register_numa(nb::module_ &m) {
|
|
nb::module_ sub_m = m.def_submodule("numa", "numa functions");
|
|
sub_m.def("get_device_numa_node", &mscclpp::getDeviceNumaNode);
|
|
sub_m.def("numa_bind", &mscclpp::numaBind);
|
|
}
|