diff options
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) { |