Function test_new_operator

internal/tests/executor/tests/test_fc.cpp:194–262

}

void test_new_operator(int rep)
{
    Node* node0 = create_fc_node();
    Node* node1 = create_fc_node();

    caffe_run_fully_connected(node0, 0, nullptr);

    SocInfo* soc_info = TestGetSocInfo();
    NodeOps* fc_ops = NodeOpsRegistryManager::FindNodeOps(soc_info, node1);

    fc_ops->SetHelper(std::malloc, std::free, nullptr);

    if(!fc_ops->Prerun(node1))
    {
        std::cout << "Prerun failed\n";
    }

    if(!fc_ops->Run(node1))
    {
        std::cout << "Run failed\n";
    }

    /* compare date */

    Tensor* tensor = node0->GetOutputTensor(0);
    float* data0 = ( float* )get_tensor_mem(tensor);

    tensor = node1->GetOutputTensor(0);

    float* data1 = ( float* )get_tensor_mem(tensor);

    std::vector<int> output_dim = tensor->GetShape().GetDim();

    std::vector<int> mismatch;

    if(!CompareFloatTensor(data0, data1, output_dim, mismatch))
    {
        std::cout << "MISMATCH: ";
        for(unsigned int i = 0; i < mismatch.size(); i++)
            std::cout << " " << mismatch[i];
        std::cout << "\n";

        DumpFloat("/tmp/data0", data0, tensor->GetShape().GetSize());
        DumpFloat("/tmp/data1", data1, tensor->GetShape().GetSize());

        return;
    }

    /* performance benchmark ... */

    unsigned long start = get_cur_time();

    for(int i = 0; i < rep; i++)
        fc_ops->Run(node1);

    unsigned long end = get_cur_time();

    std::printf("OUR IMPLEMENTATION ...\n");