Method Run

test/cpp/pir/custom_engine/fake_cpu_engine.h:147–160

  ~CustomEngine() {}

  void Run(const phi::DeviceContext& device_ctx, const phi::Place& place) {
    PADDLE_ENFORCE_EQ(
        tensor_args_.size(),
        2u,
        common::errors::PreconditionNotMet("tensor_args.size != 2"));
    PADDLE_ENFORCE_EQ(
        return_tensor_.size(),
        1u,
        common::errors::PreconditionNotMet("return_tensor.size != 1"));
    // phi::AddKernel<float, phi::DeviceContext>(device_ctx, *(tensor_args_[0]),
    // *(tensor_args_[1]),return_tensor_[0]);
    phi::Copy(device_ctx, *(tensor_args_[0]), place, true, return_tensor_[0]);
    return;
  }

 private:
  std::vector<phi::DenseTensor*> tensor_args_;