Function TEST

tensorflow/stream_executor/host/host_stream_test.cc:26–48

namespace se = stream_executor;

TEST(HostStream, EnforcesFIFOOrder) {
  se::Platform* platform =
      se::MultiPlatformManager::PlatformWithName("Host").ValueOrDie();
  se::StreamExecutor* executor = platform->ExecutorForDevice(0).ValueOrDie();
  se::Stream stream(executor);
  stream.Init();

  absl::Mutex mu;
  int expected = 0;
  bool ok = true;
  for (int i = 0; i < 2000; ++i) {
    stream.ThenDoHostCallback([i, &mu, &expected, &ok]() {
      absl::MutexLock lock(&mu);
      if (expected != i) {
        ok = false;
      }
      ++expected;
    });
  }
  TF_ASSERT_OK(stream.BlockHostUntilDone());
  absl::MutexLock lock(&mu);
  EXPECT_TRUE(ok);
}