Method exec

dnn/test/common/task_record_check.h:195–237

template <typename Opr, typename Proxy>
void TaskRecordChecker<Opr, Proxy>::exec(TensorLayoutArray layouts) {
    auto opr_cur = this->opr();
    opr_cur->param() = m_param;

    m_proxy.deduce_layout(opr_cur, layouts);
    for (size_t i = 0; i < layouts.size(); ++i) {
        if (layouts[i].dtype == dtype::Byte()) {
            layouts[i] = TensorLayout(layouts[i], dtype::Int8());
        }
    }

    // allocate input
    m_tensors_truth = alloc_tensors(m_handle.get(), layouts, 0);
    m_tensors_first = alloc_tensors(m_handle.get(), layouts, 0);
    m_tensors_second = alloc_tensors(m_handle.get(), layouts, 0);

    init_host_values();

    copy_tensors(*m_tensors_first, *m_tensors_truth);
    copy_tensors(*m_tensors_second, *m_tensors_truth);

    m_dispatcher->enable_execute_inplace();
    m_proxy.exec(opr_cur, *m_tensors_truth);

    m_dispatcher->clear_task();
    m_dispatcher->disable_execute_inplace();
    //! record the task
    m_proxy.exec(opr_cur, *m_tensors_first);
    m_dispatcher->run_task();

    //! if check record2, the opr should be free
    // free_opr();
    check_tensors(*m_tensors_truth, *m_tensors_first);

    //! change the src and out ptr and run again
    change_tensor_ptr(m_tensors_first, m_tensors_second, m_recovery_ptrs);
    m_dispatcher->run_task();
    check_tensors(*m_tensors_truth, *m_tensors_second);

    m_dispatcher->clear_task();
    recovery_tensor_ptr(m_tensors_first, m_recovery_ptrs);
    m_recovery_ptrs.clear();
}

template <typename Opr, typename Proxy>
void TaskRecordChecker<Opr, Proxy>::init_host_values() {