ArmNN
 20.05
ClMemCopyTests.cpp
Go to the documentation of this file.
1 //
2 // Copyright © 2017 Arm Ltd. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5 
7 
10 
13 
14 #include <boost/test/unit_test.hpp>
15 
16 BOOST_AUTO_TEST_SUITE(ClMemCopy)
17 
18 BOOST_AUTO_TEST_CASE(CopyBetweenCpuAndGpu)
19 {
21  MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(false);
22  BOOST_TEST(CompareTensors(result.output, result.outputExpected));
23 }
24 
25 BOOST_AUTO_TEST_CASE(CopyBetweenGpuAndCpu)
26 {
28  MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(false);
29  BOOST_TEST(CompareTensors(result.output, result.outputExpected));
30 }
31 
32 BOOST_AUTO_TEST_CASE(CopyBetweenCpuAndGpuWithSubtensors)
33 {
35  MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(true);
36  BOOST_TEST(CompareTensors(result.output, result.outputExpected));
37 }
38 
39 BOOST_AUTO_TEST_CASE(CopyBetweenGpuAndCpuWithSubtensors)
40 {
42  MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(true);
43  BOOST_TEST(CompareTensors(result.output, result.outputExpected));
44 }
45 
BOOST_AUTO_TEST_SUITE(TensorflowLiteParser)
boost::test_tools::predicate_result CompareTensors(const boost::multi_array< T, n > &a, const boost::multi_array< T, n > &b, bool compareBoolean=false)
boost::multi_array< T, n > outputExpected
BOOST_AUTO_TEST_CASE(CopyBetweenCpuAndGpu)
boost::multi_array< T, n > output
BOOST_AUTO_TEST_SUITE_END()