@@ -21,14 +21,21 @@ TEST_P(urEnqueueMemBufferMapTest, SuccessRead) {
2121 }
2222}
2323
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) {
2532 const std::vector<uint32_t > input (count, 0 );
2633 ASSERT_SUCCESS (urEnqueueMemBufferWrite (queue, buffer, true , 0 , size,
2734 input.data (), 0 , nullptr , nullptr ));
2835
2936 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 ,
3239 (void **)&map));
3340 for (unsigned i = 0 ; i < count; ++i) {
3441 map[i] = 42 ;
0 commit comments