File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1616
1717#include " paddle/fluid/framework/details/container_cast.h"
1818#include " paddle/fluid/framework/details/variable_visitor.h"
19+ #include " paddle/fluid/platform/init_phi.h"
20+
21+ REGISTER_FILE_SYMBOLS (gather_op_handle);
1922
2023namespace phi {
2124class DenseTensor ;
Original file line number Diff line number Diff line change 1515#include " paddle/fluid/framework/op_compatible_info.h"
1616
1717#include " paddle/fluid/platform/enforce.h"
18+ #include " paddle/fluid/platform/init_phi.h"
1819#include " paddle/fluid/string/string_helper.h"
1920
21+ REGISTER_FILE_SYMBOLS (op_compatible_info);
2022namespace paddle {
2123namespace framework {
2224
Original file line number Diff line number Diff line change @@ -47,6 +47,7 @@ set(PYBIND_DEPS
4747 new_profiler
4848 fluid_jit
4949 prim_utils
50+ gather_op_handle
5051 static_tensor_operants
5152 type_info
5253 auto_parallel)
Original file line number Diff line number Diff line change @@ -222,6 +222,8 @@ PYBIND11_MAKE_OPAQUE(paddle::framework::FetchType);
222222
223223DECLARE_FILE_SYMBOLS (init_phi);
224224DECLARE_FILE_SYMBOLS (kernel_dialect);
225+ DECLARE_FILE_SYMBOLS (op_compatible_info);
226+ DECLARE_FILE_SYMBOLS (gather_op_handle);
225227namespace paddle {
226228namespace pybind {
227229
Original file line number Diff line number Diff line change @@ -263,15 +263,10 @@ cc_test(
263263 SRCS dlpack_tensor_test.cc
264264 DEPS dlpack_tensor glog )
265265
266- cc_test_old (
266+ cc_test (
267267 op_compatible_info_test
268- SRCS
269- op_compatible_info_test.cc
270- DEPS
271- op_compatible_info
272- proto_desc
273- string_helper
274- glog )
268+ SRCS op_compatible_info_test.cc
269+ DEPS op_compatible_info proto_desc string_helper glog )
275270
276271cc_test (
277272 infershape_utils_test
Original file line number Diff line number Diff line change 11paddle_test (broadcast_op_test SRCS broadcast_op_handle_test.cc )
22
3- cc_test_old (
3+ cc_test (
44 gather_op_test
5- SRCS
6- gather_op_handle_test.cc
7- DEPS
8- var_handle
9- op_handle_base
10- scope
11- phi
12- common
13- memory
14- device_context
15- gather_op_handle )
5+ SRCS gather_op_handle_test.cc
6+ DEPS var_handle
7+ op_handle_base
8+ scope
9+ phi
10+ common
11+ memory
12+ device_context
13+ gather_op_handle )
1614
1715paddle_test (fused_broadcast_op_test SRCS fused_broadcast_op_handle_test.cc )
1816paddle_test (exception_holder_test SRCS exception_holder_test.cc )
@@ -23,9 +21,9 @@ cc_test(
2321 DEPS build_strategy op_registry op_proto_maker graph string_helper )
2422
2523if (WITH_ONNXRUNTIME AND WIN32 )
26- # Copy onnxruntime for some c++ test in Windows, since the test will
27- # be build only in CI, so suppose the generator in Windows is Ninja.
28- copy_onnx (broadcast_op_test )
29- copy_onnx (fused_broadcast_op_test )
30- copy_onnx (exception_holder_test )
24+ # Copy onnxruntime for some c++ test in Windows, since the test will
25+ # be build only in CI, so suppose the generator in Windows is Ninja.
26+ copy_onnx (broadcast_op_test )
27+ copy_onnx (fused_broadcast_op_test )
28+ copy_onnx (exception_holder_test )
3129endif ()
You can’t perform that action at this time.
0 commit comments