Skip to content

Commit f181d47

Browse files
authored
[Phi]Move kron kernel to phi (#40427)
* first commit * fix * fix * fix compile eeror * fix * fix complex * fix * fix * fix npu * fix * modify accroding to comments * fix
1 parent 5d08a44 commit f181d47

File tree

14 files changed

+672
-485
lines changed

14 files changed

+672
-485
lines changed

paddle/fluid/operators/gather_op_npu.cc

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ limitations under the License. */
1717
#include <string>
1818
#include <vector>
1919
#include "paddle/fluid/framework/tensor_util.h"
20-
#include "paddle/fluid/operators/kron_op.h"
2120
#include "paddle/fluid/platform/device/npu/npu_info.h"
2221
#include "paddle/fluid/platform/device/npu/npu_op_runner.h"
2322

paddle/fluid/operators/kron_op.cc

Lines changed: 1 addition & 26 deletions
Original file line numberDiff line numberDiff line change
@@ -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

2422
namespace paddle {
2523
namespace operators {
@@ -178,27 +176,4 @@ namespace ops = paddle::operators;
178176
REGISTER_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-
193179
REGISTER_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>>);

paddle/fluid/operators/kron_op.cu

Lines changed: 0 additions & 42 deletions
This file was deleted.

0 commit comments

Comments
 (0)