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

Function install_custom

imperative/python/src/ops.cpp:736–745  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

734#undef CUSTOM_CASE_TO_PARSE_NON_LIST
735
736py::list install_custom(const std::string& name, const std::string& path) {
737#if MGB_CUSTOM_OP
738 const auto& ops_in_lib = custom::CustomOpManager::inst()->install(name, path);
739 py::list ret = py::cast(ops_in_lib);
740 return ret;
741#else
742 mgb_assert(false, "CustomOp disabled, please build megengine with CustomOp open");
743 return py::list{};
744#endif
745}
746
747py::list uninstall_custom(const std::string& name) {
748#if MGB_CUSTOM_OP

Callers

nothing calls this directly

Calls 2

instFunction · 0.85
castFunction · 0.85

Tested by

no test coverage detected