diff --git a/shared/source/memory_manager/memory_manager.cpp b/shared/source/memory_manager/memory_manager.cpp index ceeaccbd5c..f8a689cc35 100644 --- a/shared/source/memory_manager/memory_manager.cpp +++ b/shared/source/memory_manager/memory_manager.cpp @@ -473,7 +473,6 @@ bool MemoryManager::getAllocationData(AllocationData &allocationData, const Allo case AllocationType::gpuTimestampDeviceBuffer: case AllocationType::preemption: case AllocationType::syncDispatchToken: - case AllocationType::timestampPacketTagBuffer: allow64KbPages = true; default: break; diff --git a/shared/test/unit_test/memory_manager/memory_manager_allocate_in_preferred_pool_tests.cpp b/shared/test/unit_test/memory_manager/memory_manager_allocate_in_preferred_pool_tests.cpp index ef4ef246e3..8c23640df6 100644 --- a/shared/test/unit_test/memory_manager/memory_manager_allocate_in_preferred_pool_tests.cpp +++ b/shared/test/unit_test/memory_manager/memory_manager_allocate_in_preferred_pool_tests.cpp @@ -243,6 +243,7 @@ INSTANTIATE_TEST_SUITE_P(Allow32BitAnd64kbPagesTypes, ::testing::ValuesIn(allocationTypesWith32BitAnd64KbPagesAllowed)); static const AllocationType allocationTypesWith32BitAnd64KbPagesNotAllowed[] = {AllocationType::commandBuffer, + AllocationType::timestampPacketTagBuffer, AllocationType::profilingTagBuffer, AllocationType::image, AllocationType::instructionHeap, @@ -611,7 +612,6 @@ HWTEST_F(GetAllocationDataTestHw, givenTimestampPacketTagBufferTypeWhenGetAlloca mockMemoryManager.getAllocationData(allocData, properties, nullptr, mockMemoryManager.createStorageInfoFromProperties(properties)); EXPECT_FALSE(allocData.flags.useSystemMemory); EXPECT_TRUE(allocData.flags.requiresCpuAccess); - EXPECT_TRUE(allocData.flags.allow64kbPages); } TEST(MemoryManagerTest, givenProfilingTagBufferTypeWhenGetAllocationDataIsCalledThenSystemMemoryIsRequested) {