Use MemoryManager retrieved from ExecutionEnvironment in Kernel dtor

Change-Id: I5f3880e1a95b3cbd262847d97776b0b92a580181

Author: zzdanowicz

runtime/kernel/kernel.cpp

@@ -100,12 +100,12 @@ Kernel::~Kernel() {
     crossThreadDataSize = 0;
 
     if (privateSurface) {
-        device.getMemoryManager()->checkGpuUsageAndDestroyGraphicsAllocations(privateSurface);
+        program->peekExecutionEnvironment().memoryManager->checkGpuUsageAndDestroyGraphicsAllocations(privateSurface);
         privateSurface = nullptr;
     }
 
     if (kernelReflectionSurface) {
-        device.getMemoryManager()->freeGraphicsMemory(kernelReflectionSurface);
+        program->peekExecutionEnvironment().memoryManager->freeGraphicsMemory(kernelReflectionSurface);
         kernelReflectionSurface = nullptr;
     }
 

unit_tests/kernel/kernel_tests.cpp

@@ -2411,3 +2411,16 @@ TEST(KernelTest, whenAllocationReadOnlyNonFlushRequiredThenAssignNullPointerToCa
     EXPECT_EQ(nullptr, kernel.mockKernel->kernelArgRequiresCacheFlush[0]);
 }
 
+TEST(KernelTest, givenKernelUsesPrivateMemoryWhenDeviceReleasedBeforeKernelThenKernelUsesMemoryManagerFromEnvironment) {
+    auto device = std::unique_ptr<MockDevice>(MockDevice::createWithNewExecutionEnvironment<MockDevice>(platformDevices[0]));
+    auto executionEnvironment = device->getExecutionEnvironment();
+
+    auto mockKernel = std::make_unique<MockKernelWithInternals>(*device);
+    GraphicsAllocation *privateSurface = device->getExecutionEnvironment()->memoryManager->allocateGraphicsMemoryWithProperties(MockAllocationProperties{MemoryConstants::pageSize});
+    mockKernel->mockKernel->setPrivateSurface(privateSurface, 10);
+
+    executionEnvironment->incRefInternal();
+    device.reset(nullptr);
+    mockKernel.reset(nullptr);
+    executionEnvironment->decRefInternal();
+}