Merge pull request PaddlePaddle#45 from tc20042008/xk-cinn-trivalop-fuse

declare group_pattern.InferShardableAxes
feifei-111 · Mar 10, 2024 · c6bcf2d · c6bcf2d
2 parents 6be51d0 + 8e74d2e
commit c6bcf2d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/paddle/cinn/frontend/group_pattern.h b/paddle/cinn/frontend/group_pattern.h
@@ -143,4 +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 std::vector<const pir::Operation*>& ops);
+
 }