Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
feifei-111 committed Mar 20, 2024
2 parents 9f804b7 + 2ec08f8 commit 4a767ec
Show file tree
Hide file tree
Showing 3 changed files with 302 additions and 451 deletions.
8 changes: 4 additions & 4 deletions paddle/cinn/frontend/cluster_ops/clustering_engine.h
Original file line number Diff line number Diff line change
Expand Up @@ -36,7 +36,7 @@ class ClusteringEngine {
const std::function<size_t(const pir::Operation*)>& OrderValue4Op);

template <typename DoEachComponentT>
void ClusteringEngine::VisitConnectedComponent(
void VisitConnectedComponent(
const common::BfsWalker<const StmtPattern*>& walker,
const std::vector<StmtPattern>& stmt_patterns,
const DoEachComponentT& DoEachComponent) {
Expand Down Expand Up @@ -97,7 +97,7 @@ class ClusteringEngine {
const std::list<const StmtPattern*> sinks);

template <typename DoEachStmtAndTopoClosureUpstreamsT>
void ClusteringEngine::VisitStmtTopoClosureUpstreams(
void VisitStmtTopoClosureUpstreams(
const common::TopoWalker<const StmtPattern*>& entire_topo_walker,
const TopoClosure& topo_closure,
const DoEachStmtAndTopoClosureUpstreamsT&
Expand Down Expand Up @@ -147,7 +147,7 @@ class ClusteringEngine {
const std::vector<StmtPattern>& stmt_patterns);

template <typename DoEachComponentT>
void ClusteringEngine::VisitClusterStmts(const common::TopoWalker<const StmtPattern*>& walker,
void VisitClusterStmts(const common::TopoWalker<const StmtPattern*>& walker,
const std::vector<StmtPattern>& stmt_patterns,
const DoEachComponentT& DoEachComponent) {
std::vector<const StmtPattern*> stmt_ptrs = [&] {
Expand Down Expand Up @@ -179,7 +179,7 @@ class ClusteringEngine {
}

template <typename DoEachComponentT>
void ClusteringEngine::VisitInferedClusterStmts(
void VisitInferedClusterStmts(
const common::TopoWalker<const StmtPattern*>& entire_topo_walker,
const std::vector<const StmtPattern*>& stmt_ptrs,
const DoEachComponentT& DoEachComponent) {
Expand Down
Loading

0 comments on commit 4a767ec

Please sign in to comment.