@@ -21,14 +21,21 @@ TEST_P(urEnqueueMemBufferMapTest, SuccessRead) {
21
21
}
22
22
}
23
23
24
- TEST_P (urEnqueueMemBufferMapTest, SuccessWrite) {
24
+ using urEnqueueMemBufferMapTestWithWriteFlagParam =
25
+ uur::urMemBufferQueueTestWithParam<ur_map_flag_t >;
26
+ UUR_TEST_SUITE_P (urEnqueueMemBufferMapTestWithWriteFlagParam,
27
+ ::testing::Values (UR_MAP_FLAG_WRITE,
28
+ UR_MAP_FLAG_WRITE_INVALIDATE_REGION),
29
+ uur::deviceTestWithParamPrinter<ur_map_flag_t>);
30
+
31
+ TEST_P (urEnqueueMemBufferMapTestWithWriteFlagParam, SuccessWrite) {
25
32
const std::vector<uint32_t > input (count, 0 );
26
33
ASSERT_SUCCESS (urEnqueueMemBufferWrite (queue, buffer, true , 0 , size,
27
34
input.data (), 0 , nullptr , nullptr ));
28
35
29
36
uint32_t *map = nullptr ;
30
- ASSERT_SUCCESS (urEnqueueMemBufferMap (queue, buffer, true , UR_MAP_FLAG_WRITE ,
31
- 0 , size, 0 , nullptr , nullptr ,
37
+ ASSERT_SUCCESS (urEnqueueMemBufferMap (queue, buffer, true , getParam (), 0 ,
38
+ size, 0 , nullptr , nullptr ,
32
39
(void **)&map));
33
40
for (unsigned i = 0 ; i < count; ++i) {
34
41
map[i] = 42 ;
0 commit comments