mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-01 04:23:00 +08:00
01c11fdf98
Change-Id: I693e73d7fcb7c586506d3e102b77b95a0a0d3783 Signed-off-by: Filip Hazubski <filip.hazubski@intel.com>
50 lines
1.7 KiB
C++
50 lines
1.7 KiB
C++
/*
|
|
* Copyright (C) 2018-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "common/helpers/bit_helpers.h"
|
|
#include "runtime/command_stream/preemption_mode.h"
|
|
#include "runtime/utilities/reference_tracked_object.h"
|
|
|
|
#include "engine_node.h"
|
|
|
|
#include <limits>
|
|
#include <memory>
|
|
|
|
namespace OCLRT {
|
|
class OSInterface;
|
|
|
|
class OsContext : public ReferenceTrackedObject<OsContext> {
|
|
public:
|
|
OsContext() = delete;
|
|
|
|
static OsContext *create(OSInterface *osInterface, uint32_t contextId, uint32_t deviceBitfield, EngineInstanceT engineType, PreemptionMode preemptionMode);
|
|
uint32_t getContextId() const { return contextId; }
|
|
uint32_t getNumSupportedDevices() const { return numSupportedDevices; }
|
|
uint8_t getDeviceBitfield() const { return deviceBitfield; }
|
|
PreemptionMode getPreemptionMode() const { return preemptionMode; }
|
|
EngineInstanceT &getEngineType() { return engineType; }
|
|
|
|
protected:
|
|
OsContext(uint32_t contextId, uint32_t deviceBitfield, EngineInstanceT engineType, PreemptionMode preemptionMode)
|
|
: contextId(contextId), deviceBitfield(deviceBitfield), preemptionMode(preemptionMode), engineType(engineType) {
|
|
constexpr uint32_t maxIndex = std::numeric_limits<decltype(deviceBitfield)>::digits;
|
|
for (uint32_t deviceIndex = 0; deviceIndex < maxIndex; deviceIndex++) {
|
|
if (isBitSet(deviceBitfield, deviceIndex)) {
|
|
numSupportedDevices++;
|
|
}
|
|
}
|
|
};
|
|
|
|
const uint32_t contextId;
|
|
const uint32_t deviceBitfield;
|
|
const PreemptionMode preemptionMode;
|
|
uint32_t numSupportedDevices = 0;
|
|
EngineInstanceT engineType = {EngineType::ENGINE_RCS, 0};
|
|
};
|
|
} // namespace OCLRT
|