mirror of
https://github.com/intel/compute-runtime.git
synced 2025-11-15 10:14:56 +08:00
7b86c8da2e
This commit adds support for ZEX_NUMBER_OF_CCS flag which can be used for limiting number of CCS engines Format is as follows: ZEX_NUMBER_OF_CCS=RootDeviceIndex:NumberOfCCS;RootDeviceIndex:NumberOfCCS... i.e. setting Root Device Index 0 to 4 CCS, and Root Device Index 1 To 1 CCS ZEX_NUMBER_OF_CCS=0:4,1:1 Related-To: NEO-7195 Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
93 lines
2.8 KiB
C++
93 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2019-2022 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/source/built_ins/sip_kernel_type.h"
|
|
#include "shared/source/helpers/affinity_mask.h"
|
|
#include "shared/source/helpers/options.h"
|
|
|
|
#include <cstdint>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <string>
|
|
|
|
namespace NEO {
|
|
|
|
class AubCenter;
|
|
class BindlessHeapsHelper;
|
|
class BuiltIns;
|
|
class CompilerInterface;
|
|
class Debugger;
|
|
class Device;
|
|
class ExecutionEnvironment;
|
|
class GmmClientContext;
|
|
class GmmHelper;
|
|
class GmmPageTableMngr;
|
|
class HwDeviceId;
|
|
class MemoryManager;
|
|
class MemoryOperationsHandler;
|
|
class OSInterface;
|
|
class OSTime;
|
|
class SipKernel;
|
|
class SWTagsManager;
|
|
struct HardwareInfo;
|
|
|
|
struct RootDeviceEnvironment {
|
|
protected:
|
|
std::unique_ptr<HardwareInfo> hwInfo;
|
|
|
|
public:
|
|
RootDeviceEnvironment(ExecutionEnvironment &executionEnvironment);
|
|
RootDeviceEnvironment(RootDeviceEnvironment &) = delete;
|
|
MOCKABLE_VIRTUAL ~RootDeviceEnvironment();
|
|
|
|
MOCKABLE_VIRTUAL const HardwareInfo *getHardwareInfo() const;
|
|
HardwareInfo *getMutableHardwareInfo() const;
|
|
void setHwInfo(const HardwareInfo *hwInfo);
|
|
bool isFullRangeSvm() const;
|
|
|
|
MOCKABLE_VIRTUAL void initAubCenter(bool localMemoryEnabled, const std::string &aubFileName, CommandStreamReceiverType csrType);
|
|
bool initOsInterface(std::unique_ptr<HwDeviceId> &&hwDeviceId, uint32_t rootDeviceIndex);
|
|
void initOsTime();
|
|
void initGmm();
|
|
void initDebugger();
|
|
void initDebuggerL0(Device *neoDevice);
|
|
MOCKABLE_VIRTUAL void prepareForCleanup() const;
|
|
MOCKABLE_VIRTUAL bool initAilConfiguration();
|
|
GmmHelper *getGmmHelper() const;
|
|
GmmClientContext *getGmmClientContext() const;
|
|
MOCKABLE_VIRTUAL CompilerInterface *getCompilerInterface();
|
|
BuiltIns *getBuiltIns();
|
|
BindlessHeapsHelper *getBindlessHeapsHelper() const;
|
|
void createBindlessHeapsHelper(MemoryManager *memoryManager, bool availableDevices, uint32_t rootDeviceIndex, DeviceBitfield deviceBitfield);
|
|
void limitNumberOfCcs(uint32_t numberOfCcs);
|
|
bool isNumberOfCcsLimited() const;
|
|
|
|
std::unique_ptr<SipKernel> sipKernels[static_cast<uint32_t>(SipKernelType::COUNT)];
|
|
std::unique_ptr<GmmHelper> gmmHelper;
|
|
std::unique_ptr<OSInterface> osInterface;
|
|
std::unique_ptr<MemoryOperationsHandler> memoryOperationsInterface;
|
|
std::unique_ptr<AubCenter> aubCenter;
|
|
std::unique_ptr<BindlessHeapsHelper> bindlessHeapsHelper;
|
|
std::unique_ptr<OSTime> osTime;
|
|
|
|
std::unique_ptr<CompilerInterface> compilerInterface;
|
|
std::unique_ptr<BuiltIns> builtins;
|
|
std::unique_ptr<Debugger> debugger;
|
|
std::unique_ptr<SWTagsManager> tagsManager;
|
|
ExecutionEnvironment &executionEnvironment;
|
|
|
|
AffinityMaskHelper deviceAffinityMask{true};
|
|
|
|
protected:
|
|
bool limitedNumberOfCcs = false;
|
|
|
|
private:
|
|
std::mutex mtx;
|
|
};
|
|
} // namespace NEO
|