| 24 | } |
| 25 | |
| 26 | bool NumRangeChecker::Checker::check(VarNode* var) { |
| 27 | auto&& val = var->dev_tensor(); |
| 28 | if (val.layout().is_contiguous()) { |
| 29 | *m_inp = var->dev_tensor(); |
| 30 | } else { |
| 31 | *m_inp = {}; |
| 32 | m_inp->copy_from(val); |
| 33 | } |
| 34 | m_func->execute(); |
| 35 | mgb_assert(m_out->shape().is_scalar()); |
| 36 | return m_out->ptr<float>()[0] >= 0.5; |
| 37 | } |
| 38 | |
| 39 | NumRangeChecker::NumRangeChecker(cg::ComputingGraph* graph, float range) |
| 40 | : PluginBase(graph), m_range{range} { |
no test coverage detected