Function load_device_id

lite/test/test_network.cpp:1629–1641

}

void load_device_id(
        LiteDeviceType device_type, int device_id, const std::string& model_path) {
    lite::Config config;
    config.device_type = device_type;
    auto network = std::make_shared<lite::Network>(config);
    network->set_device_id(device_id);
    network->load_model(model_path);
    std::shared_ptr<Tensor> input_tensor = network->get_input_tensor(0);
    std::shared_ptr<Tensor> output_tensor = network->get_output_tensor(0);
    network->forward();
    network->wait();
    ASSERT_EQ(output_tensor->get_device_id(), device_id);
}
}  // namespace
#endif