diff --git a/src/operators/feed_op.h b/src/operators/feed_op.h index b34c7cf78b0b808e512e68e5429671bf8e9d8c4a..4766d56d9ae0b86cc28c476a17547acfd53ab02b 100644 --- a/src/operators/feed_op.h +++ b/src/operators/feed_op.h @@ -74,4 +74,5 @@ USE_OP_CPU(feed); USE_OP_MALI_GPU(feed); #endif #ifdef PADDLE_MOBILE_FPGA +USE_OP_FPGA(feed); #endif diff --git a/src/operators/fetch_op.cpp b/src/operators/fetch_op.cpp index adbd61d5ec364a40b565059ceb5d5d49999c8436..30cddceaa45da91be5ea91d70f78503c404552c3 100644 --- a/src/operators/fetch_op.cpp +++ b/src/operators/fetch_op.cpp @@ -25,4 +25,5 @@ REGISTER_OPERATOR_CPU(fetch, ops::FetchOp); REGISTER_OPERATOR_MALI_GPU(fetch, ops::FetchOp); #endif #ifdef PADDLE_MOBILE_FPGA +REGISTER_OPERATOR_FPGA(fetch, ops::FetchOp); #endif diff --git a/src/operators/fetch_op.h b/src/operators/fetch_op.h index 5614fef8fe1a5b2e234b29e6d7b52cc4c2719008..417637c80086b099395e93227991491309f656fe 100644 --- a/src/operators/fetch_op.h +++ b/src/operators/fetch_op.h @@ -54,4 +54,5 @@ USE_OP_CPU(fetch); USE_OP_MALI_GPU(fetch); #endif #ifdef PADDLE_MOBILE_FPGA +USE_OP_FPGA(fetch); #endif