Correct logic in MultiRootDeviceFixture

Related-To: NEO-4589

Signed-off-by: Krzysztof Gibala <krzysztof.gibala@intel.com>
This commit is contained in:
Krzysztof Gibala
2020-11-20 12:04:46 +01:00
committed by Compute-Runtime-Automation
parent 4989cf308d
commit baea633bdd
23 changed files with 102 additions and 54 deletions

View File

@ -3227,11 +3227,11 @@ TEST_F(KernelMultiRootDeviceTest, WhenGettingRootDeviceIndexThenCorrectRootDevic
tokenSPS.PerThreadPrivateMemorySize = 112;
kernelInfo->patchInfo.pAllocateStatelessPrivateSurface = &tokenSPS;
MockProgram program(context.get(), false, toClDeviceVector(*device));
MockProgram program(context.get(), false, toClDeviceVector(*device1));
std::unique_ptr<MockKernel> kernel(new MockKernel(&program, *kernelInfo));
kernel->initialize();
auto privateSurface = kernel->kernelDeviceInfos[device->getRootDeviceIndex()].privateSurface;
auto privateSurface = kernel->kernelDeviceInfos[device1->getRootDeviceIndex()].privateSurface;
ASSERT_NE(nullptr, privateSurface);
EXPECT_EQ(expectedRootDeviceIndex, privateSurface->getRootDeviceIndex());
}