Store multiple Kernels in MultiDeviceKernel

Related-To: NEO-5001
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2021-03-16 11:32:40 +00:00
committed by Compute-Runtime-Automation
parent d7f67ad22b
commit b07f0e81b7
27 changed files with 167 additions and 108 deletions

View File

@ -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 {

View File

@ -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);
}

View File

@ -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);

View File

@ -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 {