mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
0791b25214
Related-To: NEO-4519 Change-Id: Id055631d68f1d73ae39c4976843080b8f8e05f1c Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
103 lines
2.8 KiB
C++
103 lines
2.8 KiB
C++
/*
|
|
* Copyright (C) 2017-2020 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*
|
|
*/
|
|
|
|
#pragma once
|
|
#include "shared/test/unit_test/cmd_parse/hw_parse.h"
|
|
|
|
#include "opencl/test/unit_test/command_queue/enqueue_fixture.h"
|
|
#include "opencl/test/unit_test/fixtures/hello_world_fixture.h"
|
|
#include "opencl/test/unit_test/mocks/mock_platform.h"
|
|
|
|
namespace NEO {
|
|
|
|
template <typename FactoryType>
|
|
struct MediaKernelFixture : public HelloWorldFixture<FactoryType>,
|
|
public HardwareParse,
|
|
public ::testing::Test {
|
|
typedef HelloWorldFixture<FactoryType> Parent;
|
|
|
|
using Parent::pCmdBuffer;
|
|
using Parent::pCmdQ;
|
|
using Parent::pContext;
|
|
using Parent::pCS;
|
|
using Parent::pDevice;
|
|
using Parent::pKernel;
|
|
using Parent::pProgram;
|
|
using Parent::retVal;
|
|
|
|
MediaKernelFixture() {}
|
|
|
|
template <typename FamilyType>
|
|
void enqueueRegularKernel() {
|
|
auto retVal = EnqueueKernelHelper<>::enqueueKernel(
|
|
pCmdQ,
|
|
pKernel);
|
|
ASSERT_EQ(CL_SUCCESS, retVal);
|
|
|
|
parseCommands<FamilyType>(*pCmdQ);
|
|
|
|
itorWalker1 = find<typename FamilyType::WALKER_TYPE *>(cmdList.begin(), cmdList.end());
|
|
ASSERT_NE(cmdList.end(), itorWalker1);
|
|
}
|
|
|
|
template <typename FamilyType>
|
|
void enqueueVmeKernel() {
|
|
auto retVal = EnqueueKernelHelper<>::enqueueKernel(
|
|
pCmdQ,
|
|
pVmeKernel);
|
|
ASSERT_EQ(CL_SUCCESS, retVal);
|
|
|
|
parseCommands<FamilyType>(*pCmdQ);
|
|
|
|
itorWalker1 = find<typename FamilyType::WALKER_TYPE *>(cmdList.begin(), cmdList.end());
|
|
ASSERT_NE(cmdList.end(), itorWalker1);
|
|
}
|
|
|
|
void SetUp() override {
|
|
skipVmeTest = !platform()->peekExecutionEnvironment()->rootDeviceEnvironments[0]->getHardwareInfo()->capabilityTable.supportsVme;
|
|
if (skipVmeTest) {
|
|
GTEST_SKIP();
|
|
}
|
|
Parent::kernelFilename = "vme_kernels";
|
|
Parent::kernelName = "non_vme_kernel";
|
|
|
|
Parent::SetUp();
|
|
HardwareParse::SetUp();
|
|
|
|
ASSERT_NE(nullptr, pKernel);
|
|
ASSERT_EQ(false, pKernel->isVmeKernel());
|
|
|
|
cl_int retVal;
|
|
|
|
// create the VME kernel
|
|
pVmeKernel = Kernel::create<MockKernel>(
|
|
pProgram,
|
|
*pProgram->getKernelInfo("device_side_block_motion_estimate_intel"),
|
|
&retVal);
|
|
|
|
ASSERT_NE(nullptr, pVmeKernel);
|
|
ASSERT_EQ(true, pVmeKernel->isVmeKernel());
|
|
}
|
|
|
|
void TearDown() override {
|
|
if (skipVmeTest) {
|
|
return;
|
|
}
|
|
pVmeKernel->release();
|
|
|
|
HardwareParse::TearDown();
|
|
Parent::TearDown();
|
|
}
|
|
|
|
GenCmdList::iterator itorWalker1;
|
|
GenCmdList::iterator itorWalker2;
|
|
|
|
Kernel *pVmeKernel = nullptr;
|
|
bool skipVmeTest = false;
|
|
};
|
|
} // namespace NEO
|