diff options
Diffstat (limited to 'src/backends/neon/test/NeonTimerTest.cpp')
-rw-r--r-- | src/backends/neon/test/NeonTimerTest.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/backends/neon/test/NeonTimerTest.cpp b/src/backends/neon/test/NeonTimerTest.cpp index d2bb97c056..87e15679df 100644 --- a/src/backends/neon/test/NeonTimerTest.cpp +++ b/src/backends/neon/test/NeonTimerTest.cpp @@ -105,8 +105,9 @@ TEST_CASE("NeonTimerMeasure") std::ostringstream oss_cpu; oss_neon << "NeonKernelTimer/" << measurements.size()-1 << ": NEActivationLayerKernel"; oss_cpu << "NeonKernelTimer/" << measurements.size()-1 << ": CpuActivationKernel"; - CHECK((measurements[measurements.size()-1].m_Name == oss_neon.str() || - measurements[measurements.size()-1].m_Name == oss_cpu.str())); + bool kernelCheck = ((measurements[measurements.size()-1].m_Name.find(oss_neon.str()) != std::string::npos) + || (measurements[measurements.size()-1].m_Name.find(oss_cpu.str()) != std::string::npos)); + CHECK(kernelCheck); CHECK(measurements[measurements.size()-1].m_Value > 0.0); } |