intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2026-01-07 21:27:04 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: allow using different engine types for secondary contexts

Browse Source 
Related-To: NEO-7824

Signed-off-by: Bartosz Dunajski <bartosz.dunajski@intel.com>
This commit is contained in:
Bartosz Dunajski
2024-05-23 17:05:30 +00:00
committed by Compute-Runtime-Automation
parent b39e99a17a
commit 0a74d718f2
8 changed files with 120 additions and 99 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
shared/source/device/device.h
View File

@@ -102,6 +102,7 @@ class Device : public ReferenceTrackedObject<Device> {
EngineGroupsT &getRegularEngineGroups() {
return this->regularEngineGroups;
}
const EngineGroupT *tryGetRegularEngineGroup(EngineGroupType engineGroupType) const;
size_t getEngineGroupIndexFromEngineGroupType(EngineGroupType engineGroupType) const;
EngineControl &getEngine(uint32_t index);
EngineControl &getDefaultEngine();
@@ -192,7 +193,7 @@ class Device : public ReferenceTrackedObject<Device> {
return getPreemptionMode() == PreemptionMode::MidThread || getDebugger() != nullptr;
}
MOCKABLE_VIRTUAL EngineControl *getSecondaryEngineCsr(uint32_t engineIndex, EngineTypeUsage engineTypeUsage);
MOCKABLE_VIRTUAL EngineControl *getSecondaryEngineCsr(EngineTypeUsage engineTypeUsage);
bool isSecondaryContextEngineType(aub_stream::EngineType type) {
return EngineHelpers::isCcs(type);
}
@@ -219,7 +220,7 @@ class Device : public ReferenceTrackedObject<Device> {
void addEngineToEngineGroup(EngineControl &engine);
MOCKABLE_VIRTUAL bool createEngine(uint32_t deviceCsrIndex, EngineTypeUsage engineTypeUsage);
MOCKABLE_VIRTUAL bool createSecondaryEngine(CommandStreamReceiver *primaryCsr, uint32_t index, EngineTypeUsage engineTypeUsage);
MOCKABLE_VIRTUAL bool createSecondaryEngine(CommandStreamReceiver *primaryCsr, EngineTypeUsage engineTypeUsage);
MOCKABLE_VIRTUAL std::unique_ptr<CommandStreamReceiver> createCommandStreamReceiver() const;
MOCKABLE_VIRTUAL SubDevice *createSubDevice(uint32_t subDeviceIndex);
@@ -241,7 +242,7 @@ class Device : public ReferenceTrackedObject<Device> {
std::vector<std::unique_ptr<CommandStreamReceiver>> commandStreamReceivers;
EnginesT allEngines;
std::vector<SecondaryContexts> secondaryEngines;
std::unordered_map<aub_stream::EngineType, SecondaryContexts> secondaryEngines;
std::vector<std::unique_ptr<CommandStreamReceiver>> secondaryCsrs;
EngineGroupsT regularEngineGroups;