compute-runtime/shared/source/gmm_helper/gmm_helper.h

49 lines
1.3 KiB
C++

/*
* Copyright (C) 2018-2023 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include <memory>
namespace NEO {
class GmmClientContext;
struct HardwareInfo;
struct RootDeviceEnvironment;
class GmmHelper {
public:
GmmHelper() = delete;
GmmHelper(const RootDeviceEnvironment &rootDeviceEnvironment);
MOCKABLE_VIRTUAL ~GmmHelper();
const HardwareInfo *getHardwareInfo();
uint32_t getMOCS(uint32_t type) const;
static void applyMocsEncryptionBit(uint32_t &index);
void forceAllResourcesUncached() { allResourcesUncached = true; };
static constexpr uint64_t maxPossiblePitch = (1ull << 31);
uint64_t canonize(uint64_t address) const;
uint64_t decanonize(uint64_t address) const;
uint32_t getAddressWidth() { return addressWidth; };
void setAddressWidth(uint32_t width) { addressWidth = width; };
bool isValidCanonicalGpuAddress(uint64_t address);
GmmClientContext *getClientContext() const;
const RootDeviceEnvironment &getRootDeviceEnvironment() const;
static std::unique_ptr<GmmClientContext> (*createGmmContextWrapperFunc)(const RootDeviceEnvironment &);
protected:
uint32_t addressWidth;
const RootDeviceEnvironment &rootDeviceEnvironment;
std::unique_ptr<GmmClientContext> gmmClientContext;
bool allResourcesUncached = false;
};
} // namespace NEO