mirror of
https://github.com/intel/compute-runtime.git
synced 2026-01-12 17:33:00 +08:00
29 lines
965 B
C++
29 lines
965 B
C++
/*
|
|
* Copyright (C) 2017-2018 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#include "runtime/command_stream/preemption.h"
|
|
#include "runtime/os_interface/os_context.h"
|
|
#include "unit_tests/fixtures/memory_manager_fixture.h"
|
|
#include "unit_tests/mocks/mock_csr.h"
|
|
#include "unit_tests/mocks/mock_memory_manager.h"
|
|
|
|
using namespace OCLRT;
|
|
|
|
void MemoryManagerWithCsrFixture::SetUp() {
|
|
csr = new MockCommandStreamReceiver(this->executionEnvironment);
|
|
memoryManager = new MockMemoryManager(executionEnvironment);
|
|
executionEnvironment.memoryManager.reset(memoryManager);
|
|
csr->tagAddress = ¤tGpuTag;
|
|
executionEnvironment.commandStreamReceivers.resize(1);
|
|
executionEnvironment.commandStreamReceivers[0][0].reset(csr);
|
|
|
|
csr->setOsContext(*memoryManager->createAndRegisterOsContext(gpgpuEngineInstances[0], PreemptionHelper::getDefaultPreemptionMode(*platformDevices[0])));
|
|
}
|
|
|
|
void MemoryManagerWithCsrFixture::TearDown() {
|
|
}
|