| 8 | } |
| 9 | |
| 10 | void 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 | |
| 24 | void VarValueChecker::Checker::check(VarNode* var) { |
| 25 | setup_inp(var); |
no test coverage detected