Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion paddle/fluid/ir_adaptor/translator/type_translator.h
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ class TypeTranslator {
using VarType = paddle::framework::proto::VarType;

private:
TypeTranslator(); // Disallow instantiation outside of the class.
TEST_API TypeTranslator(); // Disallow instantiation outside of the class.
std::unordered_map<VarType::Type, TypeTranslateFn> handlers;

public:
Expand Down
2 changes: 1 addition & 1 deletion paddle/fluid/pir/dialect/op_generator/op_gen.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,7 +41,7 @@

# 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']
need_export_symbol_op_list = ['AbsOp', 'FullOp', 'UniformOp']

# =====================================
# String Template for h file code gen
Expand Down
10 changes: 5 additions & 5 deletions paddle/fluid/pir/dialect/operator/utils/op_yaml_info_parser.h
Original file line number Diff line number Diff line change
Expand Up @@ -23,13 +23,13 @@ class OpYamlInfoParser {
public:
OpYamlInfoParser() = delete;

explicit OpYamlInfoParser(const OpInfoTuple& op_info_tuple,
bool is_legacy_op = false);
TEST_API explicit OpYamlInfoParser(const OpInfoTuple& op_info_tuple,
bool is_legacy_op = false);

bool IsTensorAttribute(size_t index) const;
size_t InputTensorNumber() const;
TEST_API bool IsTensorAttribute(size_t index) const;
TEST_API size_t InputTensorNumber() const;

const std::string& AttrTypeName(const std::string& name) const;
TEST_API const std::string& AttrTypeName(const std::string& name) const;
const std::string& TensorAttrTypeName(const std::string& name) const;

const std::vector<std::string>& TensorParams(bool is_kernel = false) const;
Expand Down
22 changes: 3 additions & 19 deletions test/cpp/pir/core/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -43,26 +43,10 @@ cc_test(
SRCS ir_parser_test.cc
DEPS gtest op_dialect_vjp pir)

cc_test_old(ir_op_info_test SRCS op_info_test.cc DEPS gtest pir)
cc_test_old(
ir_op_yaml_info_parser_test
SRCS
op_yaml_info_parser_test.cc
DEPS
gtest
op_dialect
op_dialect_vjp
pir)
paddle_test(ir_op_info_test SRCS op_info_test.cc)
paddle_test(ir_op_yaml_info_parser_test SRCS op_yaml_info_parser_test.cc)

cc_test_old(
ir_type_converter_test
SRCS
ir_type_converter_test.cc
DEPS
gtest
program_translator
op_dialect_vjp
pir)
paddle_test(ir_type_converter_test SRCS ir_type_converter_test.cc)

paddle_test(type_interface_test SRCS type_interface_test.cc DEPS test_dialect
gtest)
Expand Down