Function main

paddle/phi/kernels/xpu/plugin/example/example.cc:24–71

namespace xdnn = baidu::xpu::api;

int main() {
  int num = 5;
  int errcode = 0;
  auto ctx = xdnn::create_context();
  float* A = nullptr;
  errcode = xpu_malloc(reinterpret_cast<void**>(&A), num * sizeof(float));
  assert(errcode == 0);
  float* B = nullptr;
  errcode = xpu_malloc(reinterpret_cast<void**>(&B), num * sizeof(float));
  assert(errcode == 0);

  std::vector<float> A_cpu = {1, 2, 3, 4, 5};
  std::vector<float> B_cpu(num, 0.0f);
  std::vector<float> B_ref = {3, 4, 5, 6, 7};
  xpu_memcpy(reinterpret_cast<void*>(A),
             reinterpret_cast<void*>(&(A_cpu[0])),
             num * sizeof(float),
             XPUMemcpyKind::XPU_HOST_TO_DEVICE);
  errcode = xdnn::plugin::add2(ctx, A, B, num);
  assert(errcode == 0);
  xpu_memcpy(reinterpret_cast<void*>(&(B_cpu[0])),
             reinterpret_cast<void*>(B),
             num * sizeof(float),
             XPUMemcpyKind::XPU_DEVICE_TO_HOST);
  printf("A(%p):\n", A);
  for (size_t i = 0; i < num; i++) {
    printf("%f ", A_cpu[i]);
  }
  printf("\nB(%p):\n", B);
  for (size_t i = 0; i < num; i++) {
    printf("%f ", B_cpu[i]);
  }
  bool pass = true;
  for (size_t i = 0; i < num; i++) {
    if (fabs(B_cpu[i] - B_ref[i]) > 1e-5f) {
      pass = false;
      break;
    }
  }
  printf("\nCheck %s! \n", pass ? "pass" : "fail");

  destroy_context(ctx);
  errcode = xpu_free(A);
  assert(errcode == 0);
  errcode = xpu_free(B);
  assert(errcode == 0);
  return 0;
}