intel/compute-runtime
1
0
Fork 0
mirror of https://github.com/intel/compute-runtime.git synced 2025-12-24 21:18:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable memory transfer for mem obj kernel args

Browse Source 
correct building programs with multiple root devices

Related-To: NEO-5001
Signed-off-by: Mateusz Jablonski <mateusz.jablonski@intel.com>
This commit is contained in:
Mateusz Jablonski
2020-12-16 15:29:07 +00:00
committed by Compute-Runtime-Automation
parent 1ef9a1c35f
commit c3c92416db
7 changed files with 99 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
opencl/test/unit_test/kernel/kernel_arg_buffer_tests.cpp
View File

@@ -130,6 +130,30 @@ TEST_F(MultiDeviceKernelArgBufferTest, GivenValidBufferWhenSettingKernelArgThenB
}
}
TEST_F(MultiDeviceKernelArgBufferTest, WhenMakingKernelArgResidentThenMemoryIsTransferredToProperDevice) {
auto pKernel = std::unique_ptr<MockKernel>(Kernel::create<MockKernel>(pProgram.get(), kernelInfos, nullptr));
EXPECT_NE(nullptr, pKernel);
cl_mem val = pBuffer.get();
auto pVal = &val;
auto retVal = pKernel->setArg(0, sizeof(cl_mem *), pVal);
EXPECT_EQ(CL_SUCCESS, retVal);
auto csr1 = deviceFactory.rootDevices[1]->getDefaultEngine().commandStreamReceiver;
auto csr2 = deviceFactory.rootDevices[2]->getDefaultEngine().commandStreamReceiver;
pKernel->makeResident(*csr1);
EXPECT_EQ(1u, pBuffer->getMultiGraphicsAllocation().getLastUsedRootDeviceIndex());
pKernel->makeResident(*csr2);
EXPECT_EQ(2u, pBuffer->getMultiGraphicsAllocation().getLastUsedRootDeviceIndex());
pKernel->makeResident(*csr1);
EXPECT_EQ(1u, pBuffer->getMultiGraphicsAllocation().getLastUsedRootDeviceIndex());
}
TEST_F(KernelArgBufferTest, GivenSvmPtrStatelessWhenSettingKernelArgThenArgumentsAreSetCorrectly) {
Buffer *buffer = new MockBuffer();