compute-runtime/shared/test/common/mocks/mock_hw_helper.h

39 lines
1.1 KiB
C++

/*
* Copyright (C) 2020-2022 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/helpers/hw_helper.h"
#include "shared/source/helpers/hw_info.h"
#include "shared/source/memory_manager/allocation_properties.h"
namespace NEO {
template <typename GfxFamily>
class MockGfxCoreHelperWithFenceAllocation : public GfxCoreHelperHw<GfxFamily> {
public:
bool isFenceAllocationRequired(const HardwareInfo &hwInfo) const override {
return true;
}
};
template <typename GfxFamily>
class MockGfxCoreHelperWithLocalMemory : public GfxCoreHelperHw<GfxFamily> {
public:
bool isLocalMemoryEnabled(const HardwareInfo &hwInfo) const override {
return true;
}
};
template <typename GfxFamily>
struct MockGfxCoreHelperHwWithSetIsLockable : public GfxCoreHelperHw<GfxFamily> {
void setExtraAllocationData(AllocationData &allocationData, const AllocationProperties &properties, const HardwareInfo &hwInfo) const override {
allocationData.storageInfo.isLockable = setIsLockable;
}
bool setIsLockable = true;
};
} // namespace NEO