| 91 | py::handle py_comp_node_type; |
| 92 | |
| 93 | void init_common(py::module m) { |
| 94 | auto PyCompNode = |
| 95 | py::class_<CompNode>(m, "CompNode") |
| 96 | .def(py::init()) |
| 97 | .def(py::init( |
| 98 | py::overload_cast<const std::string&>(&CompNode::load))) |
| 99 | .def_property_readonly( |
| 100 | "logical_name", |
| 101 | [](const CompNode& cn) { return cn.to_string_logical(); }) |
| 102 | .def_property_readonly( |
| 103 | "physical_name", |
| 104 | [](const CompNode& cn) { return cn.to_string_physical(); }) |
| 105 | .def_property_readonly( |
| 106 | "physical_locator", |
| 107 | [](const CompNode& cn) { |
| 108 | py::list res; |
| 109 | auto locator = cn.locator(); |
| 110 | res.append(device_type2str(locator.type)); |
| 111 | res.append(locator.device); |
| 112 | res.append(locator.stream); |
| 113 | return res; |
| 114 | }) |
| 115 | .def("_log_mem_pool_details", |
| 116 | [](const CompNode& cn) -> void { cn.log_mem_pool_details(); }) |
| 117 | .def_property_readonly( |
| 118 | "get_mem_status_bytes", |
| 119 | [](const CompNode& cn) { |
| 120 | return cn.get_mem_status_bytes(); |
| 121 | }) |
| 122 | .def_property_readonly( |
| 123 | "get_used_memory", |
| 124 | [](const CompNode& cn) { return cn.get_used_memory(); }) |
| 125 | .def_property_readonly( |
| 126 | "get_max_used_memory", |
| 127 | [](const CompNode& cn) { return cn.get_max_used_memory(); }) |
| 128 | .def_property_readonly( |
| 129 | "get_reserved_memory", |
| 130 | [](const CompNode& cn) { return cn.get_reserved_memory(); }) |
| 131 | .def_property_readonly( |
| 132 | "get_max_reserved_memory", |
| 133 | [](const CompNode& cn) { |
| 134 | return cn.get_max_reserved_memory(); |
| 135 | }) |
| 136 | .def_static( |
| 137 | "reset_max_memory_stats", |
| 138 | [](const CompNode& cn) { |
| 139 | cn.reset_max_used_memory(); |
| 140 | cn.reset_max_reserved_memory(); |
| 141 | }) |
| 142 | .def("make_free_mem_block_device", |
| 143 | [](const CompNode& cn, int64_t size) { |
| 144 | cn.make_free_mem_block_device(static_cast<size_t>(size)); |
| 145 | }) |
| 146 | .def("create_event", &CompNode::create_event, |
| 147 | py::arg("flags") = 0ul) |
| 148 | .def_static("_set_default_device", &set_default_device) |
| 149 | .def_static("_get_default_device", &get_default_device) |
| 150 | .def("__str__", &CompNode::to_string_logical) |
no test coverage detected