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

40 lines
1.0 KiB
C++

/*
* Copyright (C) 2018-2020 Intel Corporation
*
* SPDX-License-Identifier: MIT
*
*/
#pragma once
#include "shared/source/utilities/reference_tracked_object.h"
#include <vector>
namespace NEO {
class MemoryManager;
struct OsEnvironment;
struct RootDeviceEnvironment;
class ExecutionEnvironment : public ReferenceTrackedObject<ExecutionEnvironment> {
public:
ExecutionEnvironment();
~ExecutionEnvironment() override;
MOCKABLE_VIRTUAL bool initializeMemoryManager();
void calculateMaxOsContextCount();
void prepareRootDeviceEnvironments(uint32_t numRootDevices);
void setPerContextMemorySpace() {
requirePerContextMemorySpace = true;
}
bool isPerContextMemorySpaceRequired() { return requirePerContextMemorySpace; }
std::unique_ptr<MemoryManager> memoryManager;
std::unique_ptr<OsEnvironment> osEnvironment;
std::vector<std::unique_ptr<RootDeviceEnvironment>> rootDeviceEnvironments;
protected:
bool requirePerContextMemorySpace = false;
};
} // namespace NEO