diff options
Diffstat (limited to 'src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp')
-rw-r--r-- | src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp b/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp index 92c8e14e4b..c4c9f7f31c 100644 --- a/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp +++ b/src/backends/backendsCommon/test/JsonPrinterTestImpl.cpp @@ -259,6 +259,12 @@ void RunSoftmaxProfilerJsonPrinterTest(const std::vector<armnn::BackendId>& back } else if (firstBackend == armnn::Compute::CpuAcc) { - CHECK(result.find("NeonKernelTimer/: CpuLogitsDLogSoftmaxKernel_#") != std::string::npos); + CHECK(result.find("NeonKernelTimer") != std::string::npos); // Validate backend + + bool softmaxCheck = ((result.find("softmax") != std::string::npos) || // Validate softmax + (result.find("Softmax") != std::string::npos) || + (result.find("SoftMax") != std::string::npos)); + CHECK(softmaxCheck); + } } |