Skip to content

Commit 99fafb8

Browse files
authored
[XPU] fix query_ner performence degradation issue (#6076)
1 parent 7da838b commit 99fafb8

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

lite/kernels/xpu/gru_compute.cc

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,14 @@
1414

1515
#include "lite/kernels/xpu/gru_compute.h"
1616

17-
// REGISTER_LITE_KERNEL(
18-
// gru, kXPU, kFloat, kNCHW, paddle::lite::kernels::xpu::GRUCompute, def)
19-
// .BindInput("Input", {LiteType::GetTensorTy(TARGET(kXPU))})
20-
// .BindInput("H0", {LiteType::GetTensorTy(TARGET(kXPU))})
21-
// .BindInput("Weight", {LiteType::GetTensorTy(TARGET(kXPU))})
22-
// .BindInput("Bias", {LiteType::GetTensorTy(TARGET(kXPU))})
23-
// .BindOutput("BatchGate", {LiteType::GetTensorTy(TARGET(kXPU))})
24-
// .BindOutput("BatchResetHiddenPrev", {LiteType::GetTensorTy(TARGET(kXPU))})
25-
// .BindOutput("BatchHidden", {LiteType::GetTensorTy(TARGET(kXPU))})
26-
// .BindOutput("Hidden", {LiteType::GetTensorTy(TARGET(kXPU))})
27-
// .Finalize();
17+
REGISTER_LITE_KERNEL(
18+
gru, kXPU, kFloat, kNCHW, paddle::lite::kernels::xpu::GRUCompute, def)
19+
.BindInput("Input", {LiteType::GetTensorTy(TARGET(kXPU))})
20+
.BindInput("H0", {LiteType::GetTensorTy(TARGET(kXPU))})
21+
.BindInput("Weight", {LiteType::GetTensorTy(TARGET(kXPU))})
22+
.BindInput("Bias", {LiteType::GetTensorTy(TARGET(kXPU))})
23+
.BindOutput("BatchGate", {LiteType::GetTensorTy(TARGET(kXPU))})
24+
.BindOutput("BatchResetHiddenPrev", {LiteType::GetTensorTy(TARGET(kXPU))})
25+
.BindOutput("BatchHidden", {LiteType::GetTensorTy(TARGET(kXPU))})
26+
.BindOutput("Hidden", {LiteType::GetTensorTy(TARGET(kXPU))})
27+
.Finalize();

0 commit comments

Comments
 (0)