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

Method init

src/plugin/impl/var_value_checker.cpp:10–22  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

8}
9
10void VarValueChecker::Checker::init(
11 VarNode* var, const std::shared_ptr<DeviceTensorND>& expected) {
12 if (!m_inp) {
13 m_inp = std::make_shared<DeviceTensorND>();
14 }
15 setup_inp(var);
16 auto graph = ComputingGraph::make();
17 auto ex = opr::SharedDeviceTensor::make(*graph, expected, {"expected"}),
18 get = opr::SharedDeviceTensor::make(
19 *graph, m_inp, {ssprintf("get:%s", cg::dump_var_info({var}).c_str())}),
20 out = opr::AssertEqual::make(ex, get, {false});
21 m_func = graph->compile({{out, {}}});
22}
23
24void VarValueChecker::Checker::check(VarNode* var) {
25 setup_inp(var);

Callers 1

on_var_computedMethod · 0.45

Calls 2

makeFunction · 0.50
compileMethod · 0.45

Tested by

no test coverage detected