intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-25 21:42:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Simplify ReferenceTrackedObject API

Browse Source 
Signed-off-by: Daniel Chabrowski <daniel.chabrowski@intel.com>
This commit is contained in:
Daniel Chabrowski
2022-06-24 14:46:11 +00:00
committed by Compute-Runtime-Automation
parent 48c2aa23db
commit f31fef4417
2 changed files with 14 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
shared/source/utilities/reference_tracked_object.h
View File

@@ -32,20 +32,15 @@ class RefCounter {
DEBUG_BREAK_IF(curr < 1);
}
bool dec() {
CT curr = --val;
void dec() {
[[maybe_unused]] CT curr = --val;
DEBUG_BREAK_IF(curr < 0);
return (curr == 0);
}
CT decAndReturnCurrent() {
return --val;
}
bool peekIsZero() const {
return (val == 0);
}
protected:
std::atomic<CT> val;
};
@@ -86,7 +81,7 @@ class unique_ptr_if_unused : // NOLINT(readability-identifier-naming)
}
template <typename DT = DataType>
static typename std::enable_if<std::is_base_of<ReferenceTrackedObject<DataType>, DT>::value, DeleterFuncType>::type getObjDeleter(DataType *inPtr) {
static typename std::enable_if_t<std::is_base_of_v<ReferenceTrackedObject<DataType>, DT>, DeleterFuncType> getObjDeleter(DataType *inPtr) {
if (inPtr != nullptr) {
return inPtr->getCustomDeleter();
}
@@ -94,7 +89,7 @@ class unique_ptr_if_unused : // NOLINT(readability-identifier-naming)
}
template <typename DT = DataType>
static typename std::enable_if<!std::is_base_of<ReferenceTrackedObject<DataType>, DT>::value, DeleterFuncType>::type getObjDeleter(DataType *inPtr) {
static typename std::enable_if_t<!std::is_base_of_v<ReferenceTrackedObject<DataType>, DT>, DeleterFuncType> getObjDeleter(DataType *inPtr) {
return nullptr;
}
@@ -121,6 +116,8 @@ class unique_ptr_if_unused : // NOLINT(readability-identifier-naming)
template <typename DerivedClass>
class ReferenceTrackedObject {
public:
using DeleterFuncType = void (*)(DerivedClass *);
virtual ~ReferenceTrackedObject();
int32_t getRefInternalCount() const {
@@ -153,15 +150,10 @@ class ReferenceTrackedObject {
return decRefInternal();
}
using DeleterFuncType = void (*)(DerivedClass *);
DeleterFuncType getCustomDeleter() const {
return nullptr;
}
bool peekHasZeroRefcounts() const {
return refInternal.peekIsZero();
}
private:
DeleterFuncType tryGetCustomDeleter() const {
const DerivedClass *asDerivedObj = static_cast<const DerivedClass *>(this);
@@ -171,6 +163,7 @@ class ReferenceTrackedObject {
RefCounter<> refInternal;
RefCounter<> refApi;
};
template <typename DerivedClass>
inline ReferenceTrackedObject<DerivedClass>::~ReferenceTrackedObject() {
DEBUG_BREAK_IF(refInternal.peek() > 1);