mirror of
https://github.com/intel/compute-runtime.git
synced 2025-12-30 01:35:20 +08:00
5a8f455a84
improve mocking: - add method to create RootDevice in Platform - add method to create SubDevice in RootDevice - add method to create CommandStreamReceiver in Device Related-To: NEO-3691 Change-Id: Ie9fe3de260492604333c8ca93796bfbffae518c4 Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
40 lines
1.1 KiB
C++
40 lines
1.1 KiB
C++
/*
|
|
* Copyright (C) 2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "runtime/device/device.h"
|
|
|
|
namespace NEO {
|
|
|
|
class SubDevice;
|
|
|
|
class RootDevice : public Device {
|
|
public:
|
|
RootDevice(ExecutionEnvironment *executionEnvironment, uint32_t rootDeviceIndex);
|
|
~RootDevice() override;
|
|
bool createDeviceImpl() override;
|
|
uint32_t getNumAvailableDevices() const override;
|
|
uint32_t getRootDeviceIndex() const override;
|
|
Device *getDeviceById(uint32_t deviceId) const override;
|
|
/* We hide the retain and release function of BaseObject. */
|
|
void retain() override;
|
|
unique_ptr_if_unused<Device> release() override;
|
|
|
|
uint32_t getNumSubDevices() const;
|
|
|
|
protected:
|
|
DeviceBitfield getDeviceBitfieldForOsContext() const override;
|
|
bool createEngines() override;
|
|
|
|
MOCKABLE_VIRTUAL bool initializeRootCommandStreamReceiver();
|
|
MOCKABLE_VIRTUAL SubDevice *createSubDevice(uint32_t subDeviceIndex);
|
|
|
|
std::vector<std::unique_ptr<SubDevice>> subdevices;
|
|
const uint32_t rootDeviceIndex;
|
|
};
|
|
} // namespace NEO
|