diff options
-rw-r--r-- | framework/Framework.cpp | 16 | ||||
-rw-r--r-- | framework/SConscript | 3 |
2 files changed, 19 insertions, 0 deletions
diff --git a/framework/Framework.cpp b/framework/Framework.cpp index 1f09550830..7708537fd6 100644 --- a/framework/Framework.cpp +++ b/framework/Framework.cpp @@ -26,6 +26,10 @@ #include "Exceptions.h" #include "support/ToolchainSupport.h" +#ifdef ARM_COMPUTE_CL +#include "arm_compute/core/CL/OpenCL.h" +#endif /* ARM_COMPUTE_CL */ + #include <chrono> #include <iostream> #include <sstream> @@ -237,6 +241,18 @@ void Framework::run_test(TestCaseFactory &test_factory) throw; } } +#ifdef ARM_COMPUTE_CL + catch(const ::cl::Error &error) + { + std::cerr << "FATAL CL ERROR: " << error.what() << " with code " << error.err() << "\n"; + result.status = TestResult::Status::FAILED; + + if(_throw_errors) + { + throw; + } + } +#endif /* ARM_COMPUTE_CL */ catch(const std::exception &error) { std::cerr << "FATAL ERROR: Received unhandled error: '" << error.what() << "'\n"; diff --git a/framework/SConscript b/framework/SConscript index bf98241ca9..36632515a3 100644 --- a/framework/SConscript +++ b/framework/SConscript @@ -47,6 +47,9 @@ if env['os'] == 'android' and framework_env['pmu']: print("pmu=1 is not supported for os=android") Exit(1) +if(env['opencl']): + framework_env.Append(CPPDEFINES=['ARM_COMPUTE_CL']) + framework_env.Append(CPPPATH = ["."]) framework_env.Append(CPPFLAGS=['-Wno-overloaded-virtual']) |