@@ -324,14 +324,16 @@ TEST(
324
324
EXPECT_EQ (ZE_RESULT_SUCCESS, status);
325
325
326
326
std::stringstream capture;
327
- std::streambuf *old_buf;
328
-
329
- old_buf = std::cerr.rdbuf ();
327
+ std::streambuf *old_cerr_buf = std::cerr.rdbuf ();
328
+ std::streambuf *old_cout_buf = std::cout.rdbuf ();
330
329
std::cerr.rdbuf (capture.rdbuf ());
330
+ std::cout.rdbuf (capture.rdbuf ());
331
331
332
332
// Action_2: Host to Device, is dependent on Action_1. It also creates a deadlock by having Action_0 dependent on it.
333
333
status = zeCommandListAppendMemoryCopy (command_list, device_mem_ptr, host_mem_ptr, buffer_size, event[2 ], 1 , &event[1 ]);
334
- std::cerr.rdbuf (old_buf);
334
+
335
+ std::cerr.rdbuf (old_cerr_buf);
336
+ std::cout.rdbuf (old_cout_buf);
335
337
336
338
auto found = capture.str ().find (" Warning: There may be a potential event deadlock" );
337
339
EXPECT_NE (found, std::string::npos);
@@ -355,12 +357,15 @@ TEST(
355
357
capture.str (" " );
356
358
capture.clear ();
357
359
358
- old_buf = std::cerr.rdbuf ();
360
+ old_cerr_buf = std::cerr.rdbuf ();
361
+ old_cout_buf = std::cout.rdbuf ();
359
362
std::cerr.rdbuf (capture.rdbuf ());
363
+ std::cout.rdbuf (capture.rdbuf ());
360
364
361
365
// Explicitly break the dependency by signaling the last event.
362
366
status = zeEventHostSignal (event[2 ]);
363
- std::cerr.rdbuf (old_buf);
367
+ std::cerr.rdbuf (old_cerr_buf);
368
+ std::cout.rdbuf (old_cout_buf);
364
369
365
370
found = capture.str ().find (" Warning: zeEventHostSignal is using the same ze_event_handle_t" );
366
371
EXPECT_NE (found, std::string::npos);
@@ -542,14 +547,15 @@ TEST(
542
547
EXPECT_EQ (ZE_RESULT_SUCCESS, status);
543
548
544
549
std::stringstream capture;
545
- std::streambuf *old_buf;
546
-
547
- old_buf = std::cerr.rdbuf ();
550
+ std::streambuf *old_cerr_buf = std::cerr.rdbuf ();
551
+ std::streambuf *old_cout_buf = std::cout.rdbuf ();
548
552
std::cerr.rdbuf (capture.rdbuf ());
553
+ std::cout.rdbuf (capture.rdbuf ());
549
554
550
- // Host to Device, signals event[0] and is dependent on nothing.
555
+ // Host to Device, signals event[0] and is dependent on nothing.
551
556
status = zeCommandListAppendMemoryCopy (command_list, device_mem_ptr, host_mem_ptr, buffer_size, event[0 ], 0 , nullptr );
552
- std::cerr.rdbuf (old_buf);
557
+ std::cerr.rdbuf (old_cerr_buf);
558
+ std::cout.rdbuf (old_cout_buf);
553
559
EXPECT_EQ (ZE_RESULT_SUCCESS, status);
554
560
555
561
auto warningText = capture.str ();
0 commit comments