diff options
Diffstat (limited to 'test/1.1')
-rw-r--r-- | test/1.1/Mean.cpp | 6 | ||||
-rw-r--r-- | test/1.1/Transpose.cpp | 6 |
2 files changed, 10 insertions, 2 deletions
diff --git a/test/1.1/Mean.cpp b/test/1.1/Mean.cpp index 7d482433..cf9ddcb2 100644 --- a/test/1.1/Mean.cpp +++ b/test/1.1/Mean.cpp @@ -18,7 +18,11 @@ using namespace armnn_driver; namespace { -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 void MeanTestImpl(const TestTensor& input, const hidl_vec<uint32_t>& axisDimensions, diff --git a/test/1.1/Transpose.cpp b/test/1.1/Transpose.cpp index 6dc3e0ee..e32a25fe 100644 --- a/test/1.1/Transpose.cpp +++ b/test/1.1/Transpose.cpp @@ -21,7 +21,11 @@ using namespace armnn_driver; namespace { -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 void TransposeTestImpl(const TestTensor & inputs, int32_t perm[], const TestTensor & expectedOutputTensor, armnn::Compute computeDevice) |