MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / load

Function load

imperative/python/src/ops.cpp:237–253  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

235 Py_RETURN_NOTIMPLEMENTED;
236 }
237 static bool load(py::handle src, T& value) {
238 PyObject* obj = src.ptr();
239 if (PyObject_TypeCheck(obj, type)) {
240 value = reinterpret_cast<EnumWrapper*>(obj)->value;
241 return true;
242 }
243 if (py::isinstance<py::str>(src)) {
244 auto&& iter = mem2value.find(normalize_enum(py::cast<std::string>(src)));
245 if (iter != mem2value.end()) {
246 value = iter->second;
247 return true;
248 } else {
249 return false;
250 }
251 }
252 return false;
253 }
254 static PyObject* cast(const T& value) {
255 auto v = static_cast<std::underlying_type_t<T>>(value);
256 mgb_assert(v <= EnumTrait<T>::max);

Callers 15

init_graph_rtFunction · 0.70
as_comp_nodeFunction · 0.70
init_commonFunction · 0.70
as_comp_nodeFunction · 0.70
loadMethod · 0.70
_get_deviceFunction · 0.70
device2objFunction · 0.70
tp_richcompareFunction · 0.70
py_new_combined_enumFunction · 0.70
TESTFunction · 0.50
TESTFunction · 0.50
TESTFunction · 0.50

Calls 4

normalize_enumFunction · 0.85
ptrMethod · 0.45
findMethod · 0.45
endMethod · 0.45

Tested by 15

TESTFunction · 0.40
TESTFunction · 0.40
TESTFunction · 0.40
TESTFunction · 0.40
run_graphFunction · 0.40
check_rng_basicFunction · 0.40
TESTFunction · 0.40
TESTFunction · 0.40
operator()Method · 0.40
TESTFunction · 0.40
test_graph_load_dumpFunction · 0.40