chore: import upstream snapshot with attribution

test/cpp/fluid/platform/device_event_test.cc

@@ -0,0 +1,147 @@
+// Copyright (c) 2021 PaddlePaddle Authors. All Rights Reserved.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+#include "paddle/phi/core/platform/device_event.h"
+
+#include "glog/logging.h"
+#include "gtest/gtest.h"
+#include "paddle/phi/common/place.h"
+
+using ::paddle::platform::kCPU;
+using ::paddle::platform::kCUDA;
+
+using paddle::platform::DeviceEvent;
+using phi::DeviceContextPool;
+
+#ifdef PADDLE_WITH_CUDA
+#include <cuda_runtime.h>
+
+TEST(DeviceEvent, CUDA) {
+  VLOG(1) << "In Test";
+  using phi::GPUPlace;
+
+  auto& pool = DeviceContextPool::Instance();
+  auto place = GPUPlace(0);
+  auto* context = static_cast<phi::GPUContext*>(pool.Get(place));
+
+  ASSERT_NE(context, nullptr);
+  // case 1. test for event_creator
+  DeviceEvent event(place, paddle::platform::GenerateDeviceEventFlag());
+  ASSERT_NE(event.GetEvent().get(), nullptr);
+  bool status = event.Query();
+  ASSERT_EQ(status, true);
+  // case 2. test for event_recorder
+  event.Record(context);
+  // case 3. test for event_finisher
+  event.Finish();
+  status = event.Query();
+  ASSERT_EQ(status, true);
+
+  // case 4. test for event_waiter
+  float *src_fp32, *dst_fp32;
+  int size = 1000000 * sizeof(float);
+  cudaMallocHost(reinterpret_cast<void**>(&src_fp32), size);
+  cudaMalloc(reinterpret_cast<void**>(&dst_fp32), size);
+  cudaMemcpyAsync(
+      dst_fp32, src_fp32, size, cudaMemcpyHostToDevice, context->stream());
+  event.Record(context);  // step 1. record it
+  status = event.Query();
+  ASSERT_EQ(status, false);
+
+  event.Wait(kCUDA, context);  // step 2. add streamWaitEvent
+  status = event.Query();
+  ASSERT_EQ(status, false);  // async
+
+  event.Wait(kCPU, context);  // step 3. EventSynchronize
+  status = event.Query();
+  ASSERT_EQ(status, true);  // sync
+
+  // release resource
+  cudaFree(dst_fp32);
+  cudaFreeHost(src_fp32);
+}
+#endif
+
+#ifdef PADDLE_WITH_HIP
+#include <hip/hip_runtime.h>
+
+TEST(DeviceEvent, CUDA) {
+  VLOG(1) << "In Test";
+  using phi::GPUPlace;
+
+  auto& pool = DeviceContextPool::Instance();
+  auto place = GPUPlace(0);
+  auto* context = static_cast<phi::GPUContext*>(pool.Get(place));
+
+  ASSERT_NE(context, nullptr);
+  // case 1. test for event_creator
+  DeviceEvent event(place, paddle::platform::GenerateDeviceEventFlag());
+  ASSERT_NE(event.GetEvent().get(), nullptr);
+  bool status = event.Query();
+  ASSERT_EQ(status, true);
+  // case 2. test for event_recorder
+  event.Record(context);
+  status = event.Query();
+  ASSERT_EQ(status, false);
+  // case 3. test for event_finisher
+  event.Finish();
+  status = event.Query();
+  ASSERT_EQ(status, true);
+
+  // case 4. test for event_waiter
+  float *src_fp32, *dst_fp32;
+  int size = 1000000 * sizeof(float);
+  hipMallocHost(reinterpret_cast<void**>(&src_fp32), size);
+  hipMalloc(reinterpret_cast<void**>(&dst_fp32), size);
+  hipMemcpyAsync(
+      dst_fp32, src_fp32, size, hipMemcpyHostToDevice, context->stream());
+  event.Record(context);  // step 1. record it
+  status = event.Query();
+  ASSERT_EQ(status, false);
+
+  event.Wait(kCUDA, context);  // step 2. add streamWaitEvent
+  status = event.Query();
+  ASSERT_EQ(status, false);  // async
+
+  event.Wait(kCPU, context);  // step 3. EventSynchronize
+  status = event.Query();
+  ASSERT_EQ(status, true);  // sync
+
+  // release resource
+  hipFree(dst_fp32);
+  hipFreeHost(src_fp32);
+}
+#endif
+
+TEST(DeviceEvent, CPU) {
+  using phi::CPUPlace;
+  auto place = CPUPlace();
+  DeviceEvent event(place, paddle::platform::GenerateDeviceEventFlag());
+  auto& pool = DeviceContextPool::Instance();
+  auto* context = pool.Get(place);
+
+  // TODO(Aurelius84): All DeviceContext should has Record/Wait
+  event.Record(context);
+  event.SetFinished();
+  bool status = event.Query();
+  ASSERT_EQ(status, true);
+
+  // test for Record again
+  event.Record(context);
+  status = event.Query();
+  ASSERT_EQ(status, false);  // SCHEDULED
+
+  event.Reset();
+  ASSERT_EQ(status, false);  // INITIALIZED
+}