60 lines
1.4 KiB
C++
60 lines
1.4 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/context/context.h"
|
|
|
|
#include "cl_api_tests.h"
|
|
|
|
using namespace NEO;
|
|
|
|
typedef api_tests clRetainMemObjectTests;
|
|
|
|
namespace ULT {
|
|
TEST_F(clRetainMemObjectTests, returnsSuccess) {
|
|
unsigned char *hostMem = nullptr;
|
|
cl_mem_flags flags = CL_MEM_USE_HOST_PTR;
|
|
static const unsigned int bufferSize = 16;
|
|
cl_mem buffer = nullptr;
|
|
cl_int retVal;
|
|
cl_uint theRef;
|
|
|
|
hostMem = new unsigned char[bufferSize];
|
|
memset(hostMem, 0xaa, bufferSize);
|
|
|
|
buffer = clCreateBuffer(
|
|
pContext,
|
|
flags,
|
|
bufferSize,
|
|
hostMem,
|
|
&retVal);
|
|
|
|
ASSERT_EQ(CL_SUCCESS, retVal);
|
|
EXPECT_NE(nullptr, buffer);
|
|
|
|
retVal = clRetainMemObject(buffer);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
|
|
retVal = clGetMemObjectInfo(buffer, CL_MEM_REFERENCE_COUNT,
|
|
sizeof(cl_uint), &theRef, NULL);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
EXPECT_EQ(2u, theRef);
|
|
|
|
retVal = clReleaseMemObject(buffer);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
|
|
retVal = clGetMemObjectInfo(buffer, CL_MEM_REFERENCE_COUNT,
|
|
sizeof(cl_uint), &theRef, NULL);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
EXPECT_EQ(1u, theRef);
|
|
|
|
retVal = clReleaseMemObject(buffer);
|
|
EXPECT_EQ(CL_SUCCESS, retVal);
|
|
|
|
delete[] hostMem;
|
|
}
|
|
} // namespace ULT
|