【Hackathon 6th Fundable Projects 3 No.192】hinge_loss (#65160)

* Fix

* Fix

* Fix

* Fix

paddle/phi/infermeta/binary.cc

@@ -2197,6 +2197,41 @@ void GridSampleBaseInferMeta(const MetaTensor& x,
   out->share_lod(x);
 }
 
+void HingeLossInferMeta(const MetaTensor& logits,
+                        const MetaTensor& labels,
+                        MetaTensor* loss) {
+  const auto& pred_dims = logits.dims();
+  const auto& label_dims = labels.dims();
+
+  PADDLE_ENFORCE_EQ(
+      pred_dims,
+      label_dims,
+      phi::errors::InvalidArgument(
+          "The Input(input) and Input(label) should have the same "
+          "shape, but received input shape [%s] != label shape [%s]",
+          pred_dims,
+          label_dims));
+
+  PADDLE_ENFORCE_EQ(
+      pred_dims.size(),
+      2,
+      phi::errors::InvalidArgument("Input(input) rank should be 2, "
+                                   "but received input rank(%d) != 2",
+                                   pred_dims.size()));
+
+  PADDLE_ENFORCE_EQ(pred_dims[1],
+                    1,
+                    phi::errors::InvalidArgument(
+                        "The second dimension of Input(input) should be 1, "
+                        "as each row of input contains a real value, "
+                        "but received second dimension of input (%d) != 1",
+                        pred_dims[1]));
+
+  loss->set_dims({pred_dims[0], 1});
+  loss->share_lod(logits);
+  loss->set_dtype(logits.dtype());
+}
+
 void HuberLossInferMeta(const MetaTensor& input,
                         const MetaTensor& label,
                         float delta,

paddle/phi/infermeta/binary.h

@@ -403,6 +403,10 @@ void GridSampleBaseInferMeta(const MetaTensor& x,
                              MetaTensor* out,
                              MetaConfig config = MetaConfig());
 
+void HingeLossInferMeta(const MetaTensor& logits,
+                        const MetaTensor& labels,
+                        MetaTensor* loss);
+
 void HuberLossInferMeta(const MetaTensor& input_meta,
                         const MetaTensor& label_meta,
                         float delta,

paddle/phi/kernels/cpu/hinge_loss_grad_kernel.cc

@@ -0,0 +1,19 @@
+// Copyright (c) 2024 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/kernel_registry.h"
+#include "paddle/phi/kernels/impl/hinge_loss_kernel_impl.h"
+
+PD_REGISTER_KERNEL(
+    hinge_loss_grad, CPU, ALL_LAYOUT, phi::HingeLossGradKernel, float) {}

paddle/phi/kernels/cpu/hinge_loss_kernel.cc

@@ -0,0 +1,18 @@
+// Copyright (c) 2024 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/kernel_registry.h"
+#include "paddle/phi/kernels/impl/hinge_loss_kernel_impl.h"
+
+PD_REGISTER_KERNEL(hinge_loss, CPU, ALL_LAYOUT, phi::HingeLossKernel, float) {}

paddle/phi/kernels/gpu/hinge_loss_grad_kernel.cu

@@ -0,0 +1,19 @@
+// Copyright (c) 2024 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/kernel_registry.h"
+#include "paddle/phi/kernels/impl/hinge_loss_kernel_impl.h"
+
+PD_REGISTER_KERNEL(
+    hinge_loss_grad, GPU, ALL_LAYOUT, phi::HingeLossGradKernel, float) {}

paddle/phi/kernels/gpu/hinge_loss_kernel.cu

@@ -0,0 +1,18 @@
+// Copyright (c) 2024 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/kernel_registry.h"
+#include "paddle/phi/kernels/impl/hinge_loss_kernel_impl.h"
+
+PD_REGISTER_KERNEL(hinge_loss, GPU, ALL_LAYOUT, phi::HingeLossKernel, float) {}