Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[CINN] Add Div to replace Recipical in DimExpr #70376

Open
wants to merge 15 commits into
base: develop
Choose a base branch
from
Open

[CINN] Add Div to replace Recipical in DimExpr #70376

Changes from 1 commit
Commits
Show all changes
15 commits
Select commit Hold shift + click to select a range
d6109fb
add div shape op
AndPuQing Dec 14, 2024
e46a1f5
fix ut
AndPuQing Dec 14, 2024
4b2f019
Refactor SimplifyDiv to streamline numerator and denominator handling
AndPuQing Dec 15, 2024
89d9c54
fix
AndPuQing Dec 15, 2024
5c423d2
Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into…
AndPuQing Dec 20, 2024
d8c9203
Refactor Div structure and update related operations for improved cla…
AndPuQing Dec 20, 2024
6190bee
fix
AndPuQing Dec 20, 2024
3c24369
fix
AndPuQing Dec 20, 2024
684aab3
fix
AndPuQing Dec 20, 2024
4694d1f
fix
AndPuQing Dec 20, 2024
6d66e14
add BinaryExprMatchTrait
gongshaotian Dec 24, 2024
a2e49f8
fix broadcast_tree typo and fix bug of BinaryDimExprMatchTrait
gongshaotian Dec 24, 2024
becd6ac
refine code
gongshaotian Dec 25, 2024
b690af7
optimize error message and delete FlattenOperands<Div>
gongshaotian Dec 25, 2024
4b4a22b
fix bug
gongshaotian Dec 25, 2024
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
Prev Previous commit
Next Next commit
add BinaryExprMatchTrait
  • Loading branch information
@gongshaotian
gongshaotian committed Dec 24, 2024
commit 6d66e14f7bc49cd75cb2f57778e25c918a90d753
16 changes: 15 additions & 1 deletion paddle/cinn/adt/dim_expr_match_trait.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,20 @@ struct UnaryDimExprMatchTrait {
}
};

template <template <typename> class Op, typename t0>
struct BinaryDimExprMatchTrait {
using base_type = Op<DimExpr>;

static constexpr int is_template = true;

template <template <typename, typename> class Matcher>
static bool MatchChildren(const base_type& value) {
const auto& lhs = std::get<0>(value.tuple());
const auto& rhs = std::get<1>(value.tuple());
return Matcher<T0, DimExpr>::Call(lhs) && Matcher<T0, DimExpr>::Call(rhs);
}
};

template <template <typename> class Op, typename T0>
struct ListDimExprMatchTrait {
using base_type = Op<DimExpr>;
Expand Down Expand Up @@ -75,7 +89,7 @@ struct MatchTrait<DimExpr, ::symbol::Mul<T0>> final

template <typename T0>
struct MatchTrait<DimExpr, ::symbol::Div<T0>> final
: public ListDimExprMatchTrait<::symbol::Div, T0> {};
: public BinaryDimExprMatchTrait<::symbol::Div, T0> {};

template <typename T0>
struct MatchTrait<DimExpr, ::symbol::Broadcast<T0>> final
Expand Down