diff options
author | Kevin May <kevin.may@arm.com> | 2019-05-22 12:02:53 +0100 |
---|---|---|
committer | Kevin May <kevin.may@arm.com> | 2019-05-23 16:22:21 +0100 |
commit | edc5ffa62ab7c552539956eba8bb1d56fb48057c (patch) | |
tree | 553ae127123aa59ebc2823db91e6b7639edee6fa /test/Lstm.cpp | |
parent | c6494ccc24d5db72e1a6afc0f71c18b5016027e8 (diff) | |
download | android-nn-driver-edc5ffa62ab7c552539956eba8bb1d56fb48057c.tar.gz |
IVGCVSW-3116 Fix failing NN Driver Tests on Android Q
* Refactor Convolution2d Unit test for fp16Enabled
* Disable GpuAcc for Unit tests in Android Q
Signed-off-by: Kevin May <kevin.may@arm.com>
Change-Id: I9613eb325841bdf7d25abf51259fe2f5d973c929
Diffstat (limited to 'test/Lstm.cpp')
-rw-r--r-- | test/Lstm.cpp | 7 |
1 files changed, 5 insertions, 2 deletions
diff --git a/test/Lstm.cpp b/test/Lstm.cpp index b1b7c9d5..56812326 100644 --- a/test/Lstm.cpp +++ b/test/Lstm.cpp @@ -1491,8 +1491,11 @@ void LstmCifgPeepholeNoProjectionBatch2(armnn::Compute compute) outputDimensions, outputValue, compute); } - -static const boost::array<armnn::Compute, 2> COMPUTE_DEVICES = {{ armnn::Compute::CpuRef, armnn::Compute::GpuAcc }}; +#ifndef ARMCOMPUTECL_ENABLED + static const boost::array<armnn::Compute, 1> COMPUTE_DEVICES = {{ armnn::Compute::CpuRef }}; +#else + static const boost::array<armnn::Compute, 2> COMPUTE_DEVICES = {{ armnn::Compute::CpuRef, armnn::Compute::GpuAcc }}; +#endif BOOST_DATA_TEST_CASE(LstmNoCifgNoPeepholeNoProjectionTest, COMPUTE_DEVICES) { |