Add option to allocate private mem per dispatch

Signed-off-by: Jaroslaw Chodor <jaroslaw.chodor@intel.com>
Signed-off-by: Krystian Chmielewski <krystian.chmielewski@intel.com>
This commit is contained in:
Jaroslaw Chodor
2021-07-23 19:23:42 +02:00
committed by Compute-Runtime-Automation
parent cf4972d90e
commit 7c6c45f5b5
12 changed files with 137 additions and 20 deletions

View File

@@ -121,6 +121,8 @@ class Device : public ReferenceTrackedObject<Device> {
void initializeRayTracing();
void reduceMaxMemAllocSize();
virtual uint64_t getGlobalMemorySize(uint32_t deviceBitfield) const;
protected:
Device() = delete;
Device(ExecutionEnvironment *executionEnvironment);
@@ -145,7 +147,6 @@ class Device : public ReferenceTrackedObject<Device> {
MOCKABLE_VIRTUAL std::unique_ptr<CommandStreamReceiver> createCommandStreamReceiver() const;
MOCKABLE_VIRTUAL SubDevice *createSubDevice(uint32_t subDeviceIndex);
MOCKABLE_VIRTUAL SubDevice *createEngineInstancedSubDevice(uint32_t subDeviceIndex, aub_stream::EngineType engineType);
virtual uint64_t getGlobalMemorySize(uint32_t deviceBitfield) const;
double getPercentOfGlobalMemoryAvailable() const;
virtual void createBindlessHeapsHelper() {}
bool createSubDevices();