@@ -17,9 +17,7 @@ limitations under the License. */
1717#include < unordered_map>
1818#include < vector>
1919
20- #include " paddle/fluid/operators/kron_op.h"
21- #include " paddle/fluid/platform/complex.h"
22- #include " paddle/fluid/platform/float16.h"
20+ #include " paddle/fluid/framework/op_registry.h"
2321
2422namespace paddle {
2523namespace operators {
@@ -178,27 +176,4 @@ namespace ops = paddle::operators;
178176REGISTER_OPERATOR (kron, ops::KronOp, ops::KronOpMaker,
179177 ops::KronGradOpMaker<paddle::framework::OpDesc>,
180178 ops::KronGradOpMaker<paddle::imperative::OpBase>);
181- REGISTER_OP_CPU_KERNEL (
182- kron, ops::KronKernel<paddle::platform::CPUDeviceContext, float >,
183- ops::KronKernel<paddle::platform::CPUDeviceContext, double >,
184- ops::KronKernel<paddle::platform::CPUDeviceContext,
185- paddle::platform::float16>,
186- ops::KronKernel<paddle::platform::CPUDeviceContext, int >,
187- ops::KronKernel<paddle::platform::CPUDeviceContext, int64_t >,
188- ops::KronKernel<paddle::platform::CPUDeviceContext,
189- paddle::platform::complex <float >>,
190- ops::KronKernel<paddle::platform::CPUDeviceContext,
191- paddle::platform::complex <double >>);
192-
193179REGISTER_OPERATOR (kron_grad, ops::KronGradOp);
194- REGISTER_OP_CPU_KERNEL (
195- kron_grad, ops::KronGradKernel<paddle::platform::CPUDeviceContext, float >,
196- ops::KronGradKernel<paddle::platform::CPUDeviceContext, double >,
197- ops::KronGradKernel<paddle::platform::CPUDeviceContext,
198- paddle::platform::float16>,
199- ops::KronGradKernel<paddle::platform::CPUDeviceContext, int >,
200- ops::KronGradKernel<paddle::platform::CPUDeviceContext, int64_t >,
201- ops::KronGradKernel<paddle::platform::CPUDeviceContext,
202- paddle::platform::complex <float >>,
203- ops::KronGradKernel<paddle::platform::CPUDeviceContext,
204- paddle::platform::complex <double >>);
0 commit comments