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

Method init_cpu

src/core/impl/comp_node_env.cpp:380–389  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

378#endif
379
380void CompNodeEnv::init_cpu(const CpuEnv& env, CompNode comp_node) {
381 m_comp_node = comp_node;
382
383 mgb_assert(!m_user_data_container);
384 m_property.type = DeviceType::CPU;
385 m_cpu_env = env;
386 m_user_data_container = std::make_unique<UserDataContainer>();
387 m_property.mem_alignment =
388 MegDNNHandle::get(*this).handle()->alignment_requirement();
389}
390
391#if MGB_CAMBRICON
392namespace {

Callers 2

CompNodeDefaultImplMethod · 0.80
CompNodeRecorderImplMethod · 0.80

Calls 3

getFunction · 0.85
alignment_requirementMethod · 0.45
handleMethod · 0.45

Tested by

no test coverage detected