L0Debug - capture SurfaceState heaps

Related-To: NEO-7103

Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>

Author: HoppeMateusz

opencl/test/unit_test/memory_manager/memory_manager_tests.cpp

@@ -3052,13 +3052,6 @@ TEST(MemoryManagerTest, givenStorageInfoWithParamsWhenGettingAllocDataForLocalMe
     EXPECT_FALSE(allocData.storageInfo.systemMemoryPlacement);
 }
 
-TEST(MemoryManagerTest, WhenCallingIsAllocationTypeToCaptureThenScratchAndPrivateTypesReturnTrue) {
-    MockMemoryManager mockMemoryManager;
-
-    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::SCRATCH_SURFACE));
-    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::PRIVATE_SURFACE));
-}
-
 TEST(MemoryTransferHelperTest, WhenBlitterIsSelectedButBlitCopyFailsThenFallbackToCopyOnCPU) {
     constexpr uint32_t dataSize = 16;
     uint8_t destData[dataSize] = {};

shared/source/memory_manager/memory_manager.cpp

@@ -856,6 +856,8 @@ bool MemoryManager::isAllocationTypeToCapture(AllocationType type) const {
     switch (type) {
     case AllocationType::SCRATCH_SURFACE:
     case AllocationType::PRIVATE_SURFACE:
+    case AllocationType::LINEAR_STREAM:
+    case AllocationType::INTERNAL_HEAP:
         return true;
     default:
         break;

shared/test/unit_test/memory_manager/CMakeLists.txt

@@ -17,6 +17,7 @@ target_sources(${TARGET_NAME} PRIVATE
                ${CMAKE_CURRENT_SOURCE_DIR}/local_memory_usage_tests.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/memory_manager_allocate_in_device_pool_tests.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/memory_manager_multi_device_tests.cpp
+               ${CMAKE_CURRENT_SOURCE_DIR}/memory_manager_tests.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/memory_pool_tests.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/multi_graphics_allocation_tests.cpp
                ${CMAKE_CURRENT_SOURCE_DIR}/page_table_tests.cpp

shared/test/unit_test/memory_manager/memory_manager_tests.cpp

@@ -0,0 +1,21 @@
+/*
+ * Copyright (C) 2022 Intel Corporation
+ *
+ * SPDX-License-Identifier: MIT
+ *
+ */
+
+#include "shared/test/common/mocks/mock_memory_manager.h"
+
+#include "gtest/gtest.h"
+
+using namespace NEO;
+
+TEST(MemoryManagerTest, WhenCallingIsAllocationTypeToCaptureThenScratchAndPrivateTypesReturnTrue) {
+    MockMemoryManager mockMemoryManager;
+
+    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::SCRATCH_SURFACE));
+    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::PRIVATE_SURFACE));
+    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::LINEAR_STREAM));
+    EXPECT_TRUE(mockMemoryManager.isAllocationTypeToCapture(AllocationType::INTERNAL_HEAP));
+}

shared/test/unit_test/os_interface/linux/drm_memory_manager_tests.cpp

@@ -3421,6 +3421,35 @@ TEST(DrmMemoryManager, givenTrackedAllocationTypeWhenFreeingThenRegisteredHandle
     EXPECT_EQ(1u, mockDrm->unregisterCalledCount);
 }
 
+TEST(DrmMemoryManager, givenEnabledResourceRegistrationWhenSshIsAllocatedThenItIsMarkedForCapture) {
+    auto executionEnvironment = new MockExecutionEnvironment();
+    executionEnvironment->prepareRootDeviceEnvironments(1u);
+    executionEnvironment->rootDeviceEnvironments[0]->setHwInfo(defaultHwInfo.get());
+    executionEnvironment->rootDeviceEnvironments[0]->initGmm();
+
+    auto mockDrm = new DrmMockResources(*executionEnvironment->rootDeviceEnvironments[0]);
+    executionEnvironment->rootDeviceEnvironments[0]->osInterface = std::make_unique<OSInterface>();
+    executionEnvironment->rootDeviceEnvironments[0]->osInterface->setDriverModel(std::unique_ptr<DriverModel>(mockDrm));
+    executionEnvironment->rootDeviceEnvironments[0]->memoryOperationsInterface = DrmMemoryOperationsHandler::create(*mockDrm, 0u);
+    executionEnvironment->memoryManager = std::make_unique<TestedDrmMemoryManager>(false, false, false, *executionEnvironment);
+
+    for (uint32_t i = 3; i < 3 + static_cast<uint32_t>(DrmResourceClass::MaxSize); i++) {
+        mockDrm->classHandles.push_back(i);
+    }
+    EXPECT_TRUE(mockDrm->resourceRegistrationEnabled());
+
+    auto device = std::unique_ptr<MockDevice>(MockDevice::create<MockDevice>(executionEnvironment, 0));
+
+    CommandContainer cmdContainer;
+    cmdContainer.initialize(device.get(), nullptr, true);
+
+    auto *ssh = cmdContainer.getIndirectHeap(NEO::HeapType::SURFACE_STATE);
+    auto bo = static_cast<DrmAllocation *>(ssh->getGraphicsAllocation())->getBO();
+
+    ASSERT_NE(nullptr, bo);
+    EXPECT_TRUE(bo->isMarkedForCapture());
+}
+
 TEST(DrmMemoryManager, givenNullBoWhenRegisteringBindExtHandleThenEarlyReturn) {
     auto executionEnvironment = std::make_unique<MockExecutionEnvironment>();
     executionEnvironment->prepareRootDeviceEnvironments(1u);