refine signature of group_pattern.InferShardableAxes

feifei-111 · Mar 10, 2024 · 8e74d2e · 8e74d2e
1 parent cff8bb6
commit 8e74d2e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/paddle/cinn/frontend/group_pattern.h b/paddle/cinn/frontend/group_pattern.h
@@ -143,6 +143,6 @@ namespace cinn::frontend {
 using ErrorGroupPattern = api::ErrorPattern<frontend::FrontendPattern>;
 using GroupPattern = api::OpTopoPattern<frontend::FrontendPattern>;
 
-std::unordered_map<pir::Value, ShardableAxes> InferShardableAxes(const cinn::pir::FusionOp& fusion_op);
+std::unordered_map<pir::Value, ShardableAxes> InferShardableAxes(const std::vector<const pir::Operation*>& ops);
 
 }