mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-03 23:03:02 +08:00
e784ba39fb
-add rootDeviceId to the command stream receiver Related-To: NEO-3857 Change-Id: I6c7f334ebe3d19cf0c58a4db65d013b7a8b7f982 Signed-off-by: Jobczyk, Lukasz <lukasz.jobczyk@intel.com>
49 lines
1.6 KiB
C++
49 lines
1.6 KiB
C++
/*
|
|
* Copyright (C) 2017-2019 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "runtime/command_stream/aub_command_stream_receiver_hw.h"
|
|
#include "runtime/execution_environment/execution_environment.h"
|
|
#include "runtime/memory_manager/os_agnostic_memory_manager.h"
|
|
#include "unit_tests/mocks/mock_experimental_command_buffer.h"
|
|
|
|
namespace NEO {
|
|
|
|
template <typename GfxFamily>
|
|
class UltAubCommandStreamReceiver : public AUBCommandStreamReceiverHw<GfxFamily>, public NonCopyableOrMovableClass {
|
|
using BaseClass = AUBCommandStreamReceiverHw<GfxFamily>;
|
|
|
|
public:
|
|
using BaseClass::osContext;
|
|
|
|
UltAubCommandStreamReceiver(const std::string &fileName, bool standalone, ExecutionEnvironment &executionEnvironment, uint32_t rootDeviceIndex)
|
|
: BaseClass(fileName, standalone, executionEnvironment, rootDeviceIndex) {
|
|
}
|
|
|
|
UltAubCommandStreamReceiver(ExecutionEnvironment &executionEnvironment, uint32_t rootDeviceIndex)
|
|
: BaseClass("aubfile", true, executionEnvironment, rootDeviceIndex) {
|
|
}
|
|
|
|
static CommandStreamReceiver *create(bool withAubDump, ExecutionEnvironment &executionEnvironment, uint32_t rootDeviceIndex) {
|
|
auto csr = new UltAubCommandStreamReceiver<GfxFamily>("aubfile", true, executionEnvironment, rootDeviceIndex);
|
|
|
|
if (!csr->subCaptureManager->isSubCaptureMode()) {
|
|
csr->openFile("aubfile");
|
|
}
|
|
|
|
return csr;
|
|
}
|
|
|
|
uint32_t blitBuffer(const BlitProperties &blitProperties) override {
|
|
blitBufferCalled++;
|
|
return BaseClass::blitBuffer(blitProperties);
|
|
}
|
|
|
|
uint32_t blitBufferCalled = 0;
|
|
};
|
|
} // namespace NEO
|