};

  StreamExecutor* executor = GetExecutor(0);
  TF_ASSERT_OK_AND_ASSIGN(auto stream, executor->CreateStream());
  size_t size = sizeof(int);
  int src_data = 34;
  int dst_data = 2;
  DeviceMemoryBase device_src(&src_data, size);
  TF_ASSERT_OK(stream->Memcpy(&dst_data, device_src, size));
  ASSERT_EQ(dst_data, 34);
}

TEST_F(StreamExecutorTest, MemcpyFromHost) {