diff options
| author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-07 17:52:36 +0100 |
|---|---|---|
| committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2021-05-08 20:15:32 +0100 |
| commit | e5f0b2409c2e557a5a78e2f4659d203154289b23 (patch) | |
| tree | 0e32680ed15ed5157c78d5deeabda2c0ceeeb4a3 /src/armnn/test/TensorHandleStrategyTest.cpp | |
| parent | ae12306486efc55293a40048618abe5e8b19151b (diff) | |
| download | armnn-e5f0b2409c2e557a5a78e2f4659d203154289b23.tar.gz | |
IVGCVSW-5818 Enable import on GPU
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I4e4eb107aa2bfa09625840d738001f33152e6792
Diffstat (limited to 'src/armnn/test/TensorHandleStrategyTest.cpp')
| -rw-r--r-- | src/armnn/test/TensorHandleStrategyTest.cpp | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/src/armnn/test/TensorHandleStrategyTest.cpp b/src/armnn/test/TensorHandleStrategyTest.cpp index c7aa30f701..47d0666414 100644 --- a/src/armnn/test/TensorHandleStrategyTest.cpp +++ b/src/armnn/test/TensorHandleStrategyTest.cpp @@ -139,7 +139,8 @@ public: { "TestHandleFactoryA1", "TestHandleFactoryA2", - "TestHandleFactoryB1" + "TestHandleFactoryB1", + "TestHandleFactoryD1" }; } @@ -252,7 +253,7 @@ public: std::vector<ITensorHandleFactory::FactoryId> GetHandleFactoryPreferences() const override { return std::vector<ITensorHandleFactory::FactoryId>{ - "TestHandleFactoryD1" + "TestHandleFactoryD1", }; } @@ -279,6 +280,7 @@ BOOST_AUTO_TEST_CASE(RegisterFactories) BOOST_TEST(backendA.GetHandleFactoryPreferences()[0] == "TestHandleFactoryA1"); BOOST_TEST(backendA.GetHandleFactoryPreferences()[1] == "TestHandleFactoryA2"); BOOST_TEST(backendA.GetHandleFactoryPreferences()[2] == "TestHandleFactoryB1"); + BOOST_TEST(backendA.GetHandleFactoryPreferences()[3] == "TestHandleFactoryD1"); TensorHandleFactoryRegistry registry; backendA.RegisterTensorHandleFactories(registry); @@ -351,7 +353,7 @@ BOOST_AUTO_TEST_CASE(TensorHandleSelectionStrategy) OutputSlot& softmaxLayer4Out = softmaxLayer4->GetOutputSlot(0); // Check that the correct factory was selected - BOOST_TEST(inputLayerOut.GetTensorHandleFactoryId() == "TestHandleFactoryA1"); + BOOST_TEST(inputLayerOut.GetTensorHandleFactoryId() == "TestHandleFactoryD1"); BOOST_TEST(softmaxLayer1Out.GetTensorHandleFactoryId() == "TestHandleFactoryB1"); BOOST_TEST(softmaxLayer2Out.GetTensorHandleFactoryId() == "TestHandleFactoryB1"); BOOST_TEST(softmaxLayer3Out.GetTensorHandleFactoryId() == "TestHandleFactoryC1"); |