diff --git a/paddle/cinn/frontend/group_pattern_util.cc b/paddle/cinn/frontend/group_pattern_util.cc index 80b0cc3130511..32e9ffff81f7f 100644 --- a/paddle/cinn/frontend/group_pattern_util.cc +++ b/paddle/cinn/frontend/group_pattern_util.cc @@ -1,12 +1,13 @@ #include "paddle/cinn/frontend/group_pattern_util.h" +#include namespace cinn::frontend { namespace { -using IS = InjectiveSourcePattern; -using R = ReductionPattern; -using PS = PartialShardablePattern; +using IS = api::InjectiveSourcePattern; +using R = api::ReductionPattern; +using PS = api::PartialShardablePattern; using InternalPattern = std::variant; diff --git a/paddle/cinn/frontend/group_pattern_util.h b/paddle/cinn/frontend/group_pattern_util.h index 1b21f6c999a26..e50ffa3004ef3 100644 --- a/paddle/cinn/frontend/group_pattern_util.h +++ b/paddle/cinn/frontend/group_pattern_util.h @@ -10,6 +10,6 @@ struct ErrorGroupPattern { std::string error_string; }; -std::variant GenerateGroupPatternFromFusionOp(const pir::FusionOp&); +std::variant GenerateGroupPatternFromFusionOp(const cinn::dialect::FusionOp&); } \ No newline at end of file