38 lines
852 B
C++
38 lines
852 B
C++
|
// Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
|
||
|
|
// clang-format off
|
||
|
|
#include "catch.hpp"
|
||
|
|
// clang-format on
|
||
|
|
|
||
|
|
#include "core/model.h"
|
||
|
|
#include "core/net.h"
|
||
|
|
#include "test_resource.h"
|
||
|
|
|
||
|
|
using namespace mmdeploy;
|
||
|
|
|
||
|
|
TEST_CASE("test pplnn net", "[ppl_net]") {
|
||
|
|
auto& gResource = MMDeployTestResources::Get();
|
||
|
|
auto model_list = gResource.LocateModelResources("mmcls/pplnn");
|
||
|
|
REQUIRE(!model_list.empty());
|
||
|
|
|
||
|
|
Model model(model_list.front());
|
||
|
|
REQUIRE(model);
|
||
|
|
|
||
|
|
auto backend = "pplnn";
|
||
|
|
auto creator = Registry<Net>::Get().GetCreator(backend);
|
||
|
|
REQUIRE(creator);
|
||
|
|
|
||
|
|
Device device{"cpu"};
|
||
|
|
auto stream = Stream::GetDefault(device);
|
||
|
|
// clang-format off
|
||
|
|
Value net_config{
|
||
|
|
{"context", {
|
||
|
|
{"device", device},
|
||
|
|
{"model", model},
|
||
|
|
{"stream", stream}
|
||
|
|
}
|
||
|
|
},
|
||
|
|
{"name", model.meta().models[0].name}
|
||
|
|
};
|
||
|
|
}
|