mirror of
https://github.com/intel/compute-runtime.git
synced 2025-09-15 13:01:45 +08:00
Store multiple Kernels in MultiDeviceKernel
Related-To: NEO-5001 Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:

committed by
Compute-Runtime-Automation

parent
d7f67ad22b
commit
b07f0e81b7
@ -93,7 +93,7 @@ class ExecutionModelSchedulerTest : public ClDeviceFixture,
|
||||
|
||||
parentKernel = MockParentKernel::create(*context);
|
||||
ASSERT_NE(nullptr, parentKernel);
|
||||
pMultiDeviceKernel = new MockMultiDeviceKernel(parentKernel);
|
||||
pMultiDeviceKernel = new MockMultiDeviceKernel(MockMultiDeviceKernel::toKernelVector(parentKernel));
|
||||
}
|
||||
|
||||
void TearDown() override {
|
||||
|
@ -43,7 +43,7 @@ struct ExecutionModelKernelFixture : public ProgramFromBinaryFixture,
|
||||
pProgram->getKernelInfosForKernel(kernelName),
|
||||
&retVal);
|
||||
|
||||
pMultiDeviceKernel = new MockMultiDeviceKernel(pKernel);
|
||||
pMultiDeviceKernel = new MockMultiDeviceKernel(MockMultiDeviceKernel::toKernelVector(pKernel));
|
||||
ASSERT_EQ(CL_SUCCESS, retVal);
|
||||
ASSERT_NE(nullptr, pKernel);
|
||||
}
|
||||
|
@ -63,9 +63,10 @@ void KernelImageArgTest::SetUp() {
|
||||
ClDeviceFixture::SetUp();
|
||||
context.reset(new MockContext(pClDevice));
|
||||
program = std::make_unique<MockProgram>(context.get(), false, toClDeviceVector(*pClDevice));
|
||||
pKernel = new MockKernel(program.get(), MockKernel::toKernelInfoContainer(*pKernelInfo, rootDeviceIndex));
|
||||
ASSERT_EQ(CL_SUCCESS, pKernel->initialize());
|
||||
pMultiDeviceKernel = std::make_unique<MultiDeviceKernel>(pKernel);
|
||||
int32_t retVal = CL_INVALID_VALUE;
|
||||
pMultiDeviceKernel.reset(MultiDeviceKernel::create<MockKernel>(program.get(), MockKernel::toKernelInfoContainer(*pKernelInfo, rootDeviceIndex), &retVal));
|
||||
pKernel = static_cast<MockKernel *>(pMultiDeviceKernel->getKernel(rootDeviceIndex));
|
||||
ASSERT_EQ(CL_SUCCESS, retVal);
|
||||
|
||||
pKernel->setKernelArgHandler(0, &Kernel::setArgImage);
|
||||
pKernel->setKernelArgHandler(1, &Kernel::setArgImage);
|
||||
|
@ -81,7 +81,7 @@ struct MediaKernelFixture : public HelloWorldFixture<FactoryType>,
|
||||
|
||||
ASSERT_NE(nullptr, pVmeKernel);
|
||||
ASSERT_EQ(true, pVmeKernel->isVmeKernel());
|
||||
pMultiDeviceVmeKernel = new MockMultiDeviceKernel(pVmeKernel);
|
||||
pMultiDeviceVmeKernel = new MockMultiDeviceKernel(MockMultiDeviceKernel::toKernelVector(pVmeKernel));
|
||||
}
|
||||
|
||||
void TearDown() override {
|
||||
|
Reference in New Issue
Block a user