[MLU]: compile prior box CPU kernel when WITH_MLU is ON for SSD (#45238)

596a9405 · zhaoying9105 · GitHub · e761ea2a · 596a9405
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/operators/detection/CMakeLists.txt paddle/fluid/operators/detection/CMakeLists.txt +1 -0

未找到文件。
--- a/paddle/fluid/operators/detection/CMakeLists.txt
+++ b/paddle/fluid/operators/detection/CMakeLists.txt
@@ -47,6 +47,7 @@ if(WITH_XPU)
 elseif(WITH_MLU)
  detection_library(iou_similarity_op SRCS iou_similarity_op.cc
                    iou_similarity_op_mlu.cc)
+  detection_library(prior_box_op SRCS prior_box_op.cc)
 elseif(WITH_ASCEND_CL)
  detection_library(iou_similarity_op SRCS iou_similarity_op.cc
                    iou_similarity_op_npu.cc)