diff --git a/paddle/fluid/pir/dialect/op_generator/op_gen.py b/paddle/fluid/pir/dialect/op_generator/op_gen.py index 3c4661a7caaf7..b7e33e9394244 100644 --- a/paddle/fluid/pir/dialect/op_generator/op_gen.py +++ b/paddle/fluid/pir/dialect/op_generator/op_gen.py @@ -46,7 +46,15 @@ # Note(Galaxy1458) The need_export_symbol_op_list is used # for some unittests these need to export symbol op compiled with dynamic lib. -need_export_symbol_op_list = ['AbsOp', 'FullOp', 'UniformOp'] +need_export_symbol_op_list = [ + 'AbsOp', + 'FullOp', + 'UniformOp', + 'Conv2dOp', + 'BatchNormOp', + 'FetchOp', + 'TransposeOp', +] # ===================================== # String Template for h file code gen diff --git a/test/cpp/pir/pass/CMakeLists.txt b/test/cpp/pir/pass/CMakeLists.txt index 5d43a23382b0a..3507cfd0708dd 100644 --- a/test/cpp/pir/pass/CMakeLists.txt +++ b/test/cpp/pir/pass/CMakeLists.txt @@ -1,4 +1,7 @@ -cc_test( - pass_manager_test - SRCS pass_manager_test.cc - DEPS pir op_dialect_vjp phi common gtest) +paddle_test(pass_manager_test SRCS pass_manager_test.cc DEPS common) + +if(WITH_ONNXRUNTIME AND WIN32) + # Copy onnxruntime for some c++ test in Windows, since the test will + # be build only in CI, so suppose the generator in Windows is Ninja. + copy_onnx(pass_manager_test) +endif()