From 17dae8765ba738c0d68fd3ed6af9eae8ae40798b Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Fri, 12 Apr 2019 17:08:40 +0100 Subject: COMPMID-2113: Check correct extension for host memory import. Change-Id: I397a6a84e12501512f82fe8cacccd59f689ece26 Signed-off-by: Georgios Pinitas Reviewed-on: https://review.mlplatform.org/c/984 Reviewed-by: Michalis Spyrou Comments-Addressed: Arm Jenkins Tested-by: Arm Jenkins --- tests/validation/CL/UNIT/TensorAllocator.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/validation/CL/UNIT/TensorAllocator.cpp b/tests/validation/CL/UNIT/TensorAllocator.cpp index abe06c544b..7e47e3d983 100644 --- a/tests/validation/CL/UNIT/TensorAllocator.cpp +++ b/tests/validation/CL/UNIT/TensorAllocator.cpp @@ -55,7 +55,7 @@ cl_mem import_malloc_memory_helper(void *ptr, size_t size) cl_int err = CL_SUCCESS; cl_mem buf = clImportMemoryARM(CLKernelLibrary::get().context().get(), CL_MEM_READ_WRITE, import_properties, ptr, size, &err); - ARM_COMPUTE_EXPECT(err == CL_SUCCESS, framework::LogLevel::ERRORS); + ARM_COMPUTE_ASSERT(err == CL_SUCCESS); return buf; } @@ -108,7 +108,7 @@ TEST_CASE(ImportMemoryBuffer, framework::DatasetMode::ALL) TEST_CASE(ImportMemoryMalloc, framework::DatasetMode::ALL) { // Check if import extension is supported - if(!device_supports_extension(CLKernelLibrary::get().get_device(), "cl_arm_import_memory")) + if(!device_supports_extension(CLKernelLibrary::get().get_device(), "cl_arm_import_memory_host")) { return; } -- cgit v1.2.1