|
8 | 8 | #include "core/memory_manager/unified_memory_manager.h" |
9 | 9 | #include "runtime/built_ins/builtins_dispatch_builder.h" |
10 | 10 | #include "test.h" |
| 11 | +#include "unit_tests/command_queue/command_enqueue_fixture.h" |
11 | 12 | #include "unit_tests/command_queue/command_queue_fixture.h" |
12 | 13 | #include "unit_tests/fixtures/device_fixture.h" |
13 | 14 | #include "unit_tests/mocks/mock_builtin_dispatch_info_builder.h" |
@@ -147,3 +148,69 @@ HWTEST_P(EnqueueSvmMemFillTest, givenEnqueueSVMMemFillWhenUsingFillBufferBuilder |
147 | 148 | INSTANTIATE_TEST_CASE_P(size_t, |
148 | 149 | EnqueueSvmMemFillTest, |
149 | 150 | ::testing::Values(1, 2, 4, 8, 16, 32, 64, 128)); |
| 151 | + |
| 152 | +struct EnqueueSvmMemFillHw : public ::testing::Test { |
| 153 | + |
| 154 | + void SetUp() override { |
| 155 | + |
| 156 | + device.reset(MockDevice::createWithNewExecutionEnvironment<MockDevice>(*platformDevices)); |
| 157 | + if (is32bit || !device->isFullRangeSvm()) { |
| 158 | + GTEST_SKIP(); |
| 159 | + } |
| 160 | + |
| 161 | + context = std::make_unique<MockContext>(device.get()); |
| 162 | + svmPtr = context->getSVMAllocsManager()->createSVMAlloc(device->getRootDeviceIndex(), 256, {}); |
| 163 | + ASSERT_NE(nullptr, svmPtr); |
| 164 | + } |
| 165 | + |
| 166 | + void TearDown() override { |
| 167 | + if (is32bit || !device->isFullRangeSvm()) { |
| 168 | + return; |
| 169 | + } |
| 170 | + context->getSVMAllocsManager()->freeSVMAlloc(svmPtr); |
| 171 | + } |
| 172 | + |
| 173 | + std::unique_ptr<MockDevice> device; |
| 174 | + std::unique_ptr<MockContext> context; |
| 175 | + uint64_t bigSize = 5ull * MemoryConstants::gigaByte; |
| 176 | + uint64_t smallSize = 4ull * MemoryConstants::gigaByte - 1; |
| 177 | + void *svmPtr = nullptr; |
| 178 | + const uint64_t pattern[4] = {0x0011223344556677, |
| 179 | + 0x8899AABBCCDDEEFF, |
| 180 | + 0xFFEEDDCCBBAA9988, |
| 181 | + 0x7766554433221100}; |
| 182 | + size_t patternSize = 0; |
| 183 | +}; |
| 184 | + |
| 185 | +using EnqueueSvmMemFillHwTest = EnqueueSvmMemFillHw; |
| 186 | + |
| 187 | +HWTEST_F(EnqueueSvmMemFillHwTest, givenEnqueueSVMMemFillWhenUsingCopyBufferToBufferStatelessBuilderThenSuccessIsReturned) { |
| 188 | + auto cmdQ = std::make_unique<CommandQueueStateless<FamilyType>>(context.get(), device.get()); |
| 189 | + auto svmData = context->getSVMAllocsManager()->getSVMAlloc(svmPtr); |
| 190 | + svmData->size = static_cast<size_t>(bigSize); |
| 191 | + |
| 192 | + auto retVal = cmdQ->enqueueSVMMemFill( |
| 193 | + svmPtr, // void *svm_ptr |
| 194 | + pattern, // const void *pattern |
| 195 | + patternSize, // size_t pattern_size |
| 196 | + static_cast<size_t>(bigSize), // size_t size |
| 197 | + 0, // cl_uint num_events_in_wait_list |
| 198 | + nullptr, // cl_event *event_wait_list |
| 199 | + nullptr // cL_event *event |
| 200 | + ); |
| 201 | + EXPECT_EQ(CL_SUCCESS, retVal); |
| 202 | +} |
| 203 | + |
| 204 | +HWTEST_F(EnqueueSvmMemFillHwTest, givenEnqueueSVMMemFillWhenUsingCopyBufferToBufferStatefulBuilderThenSuccessIsReturned) { |
| 205 | + auto cmdQ = std::make_unique<CommandQueueStateful<FamilyType>>(context.get(), device.get()); |
| 206 | + auto retVal = cmdQ->enqueueSVMMemFill( |
| 207 | + svmPtr, // void *svm_ptr |
| 208 | + pattern, // const void *pattern |
| 209 | + patternSize, // size_t pattern_size |
| 210 | + static_cast<size_t>(smallSize), // size_t size |
| 211 | + 0, // cl_uint num_events_in_wait_list |
| 212 | + nullptr, // cl_event *event_wait_list |
| 213 | + nullptr // cL_event *event |
| 214 | + ); |
| 215 | + EXPECT_EQ(CL_SUCCESS, retVal); |
| 216 | +} |
0 commit comments