KernelSignature ReshapeOpArgumentMapping(const ArgumentMappingContext& ctx) {
  if (ctx.HasOutput("XShape")) {
    if (ctx.InputSize("ShapeTensor") > 0) {
      return KernelSignature(
          "reshape", {"X"}, {"ShapeTensor"}, {"Out", "XShape"});   // --> 这里要改为reshape_with_xshape
    } else if (ctx.HasInput("Shape")) {
      return KernelSignature("reshape", {"X"}, {"Shape"}, {"Out", "XShape"});
    } else {
      return KernelSignature("reshape", {"X"}, {"shape"}, {"Out", "XShape"});
    }
  } else {
    if (ctx.InputSize("ShapeTensor") > 0) {
      return KernelSignature("reshape_infer", {"X"}, {"ShapeTensor"}, {"Out"});  // ---> 这里规范为reshape 即可
    } else if (ctx.HasInput("Shape")) {
      return KernelSignature("reshape_infer", {"X"}, {"Shape"}, {"Out"});
    } else {
      return KernelSignature("reshape_infer", {"X"}, {"shape"}, {"Out"});
    }
  }
}

KernelSignature ReshapeGradOpArgumentMapping(
    const ArgumentMappingContext& ctx UNUSED) {
  return KernelSignature("reshape_grad", {"XShape", "Out@GRAD"}, {}, {"X@GRAD"});
}

  Tensor& api_output = x;  // <---- inplace 机制下，out 直接用的x，会导致infermeta后 x 的meta会覆盖
  auto kernel_out = SetKernelOutput(&api_output);
  phi::RecordEvent *infer_shape_record_event = nullptr;
  if(phi::RecordEvent::IsEnabled()){
    infer_shape_record_event = new phi::RecordEvent("reshape infer_meta", phi::TracerEventType::OperatorInner, 1);
  }
  auto origin_input_x = *input_x;     // <----- 此处浅copy了输入Tensor
  phi::MetaTensor meta_out(kernel_out, kernel_result.is_stride_kernel);
  phi::ReshapeInferMeta(MakeMetaTensor(origin_input_x), shape, &meta_out); // <--- 使用origin_input_x来infermeta
  if(infer_shape_record_event != nullptr){
    delete infer_shape_record_event;
  }
  using kernel_signature = void(*)(const phi::DeviceContext&, const phi::DenseTensor&, const phi::IntArray&, phi::DenseTensor*);
  auto* kernel_fn = kernel.GetVariadicKernelFn<kernel_signature>();
  phi::RecordEvent* kernel_record_event = nullptr;
  if(phi::RecordEvent::IsEnabled()){
    kernel_record_event = new phi::RecordEvent("reshape compute", phi::TracerEventType::OperatorInner, 1);
  }
    (*kernel_fn)(*dev_ctx, origin_input_x, phi::IntArray(shape), kernel_out);
  if(kernel_record_event != nullptr){
    delete kernel_record_event;
  }