compute-runtime/shared/test/common/mocks/ult_device_factory.h

/*
 * Copyright (C) 2020-2025 Intel Corporation
 *
 * SPDX-License-Identifier: MIT
 *
 */

#pragma once
#include <cstdint>
#include <vector>

namespace NEO {
class ExecutionEnvironment;
class MockDevice;
class SubDevice;

struct UltDeviceFactory {
    UltDeviceFactory(uint32_t rootDevicesCount, uint32_t subDevicesCount);
    UltDeviceFactory(uint32_t rootDevicesCount, uint32_t subDevicesCount, ExecutionEnvironment &executionEnvironment);
    ~UltDeviceFactory();

    static bool prepareDeviceEnvironments(ExecutionEnvironment &executionEnvironment, uint32_t rootDevicesCount);
    static bool initializeMemoryManager(ExecutionEnvironment &executionEnvironment);

    std::vector<MockDevice *> rootDevices;
    std::vector<SubDevice *> subDevices;
};

} // namespace NEO