fix matrix_rank

Author: Zjq9409

paddle/fluid/operators/eigh_op.cc

@@ -145,3 +145,11 @@ REGISTER_OPERATOR(eigh, ops::EighOp, ops::EignOpMaker,
  ops::EighGradOpMaker<paddle::framework::OpDesc>,
  ops::EighGradOpMaker<paddle::imperative::OpBase>);
 REGISTER_OPERATOR(eigh_grad, ops::EighGradOp);
+
+REGISTER_OP_CPU_KERNEL(
+ eigh_grad, ops::EighGradKernel<paddle::platform::CPUDeviceContext, float>,
+ ops::EighGradKernel<paddle::platform::CPUDeviceContext, double>,
+ ops::EighGradKernel<paddle::platform::CPUDeviceContext,
+ paddle::platform::complex<float>>,
+ ops::EighGradKernel<paddle::platform::CPUDeviceContext,
+ paddle::platform::complex<double>>);

paddle/fluid/operators/eigh_op.cu

@@ -0,0 +1,22 @@
+/* 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/fluid/operators/eigh_op.h"
+
+namespace ops = paddle::operators;
+
+REGISTER_OP_CUDA_KERNEL(
+ eigh_grad, ops::EighGradKernel<paddle::platform::CUDADeviceContext, float>,
+ ops::EighGradKernel<paddle::platform::CUDADeviceContext, double>,
+ ops::EighGradKernel<paddle::platform::CUDADeviceContext,
+ paddle::platform::complex<float>>,
+ ops::EighGradKernel<paddle::platform::CUDADeviceContext,
+ paddle::platform::complex<double>>);