Skip to content

[FLYDSL]: dsl and/or ast rewrite dynamic process #245

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

Closed
xudoyuan wants to merge 1 commit into from
Closed

[FLYDSL]: dsl and/or ast rewrite dynamic process #245

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
17 changes: 11 additions & 6 deletions python/flydsl/compiler/ast_rewriter.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -263,17 +263,22 @@ def rewrite_globals(cls):
"scf_if_dispatch": cls.scf_if_dispatch,
}

_REWRITE_HELPER_NAMES = {"dsl_not_", "dsl_and_", "dsl_or_",
"scf_if_dispatch", "const_expr", "type",
"bool", "isinstance", "hasattr"}
_REWRITE_HELPER_NAMES = {
"scf_if_dispatch",
"const_expr",
"type",
"bool",
"isinstance",
"hasattr",
}

@staticmethod
def _could_be_dynamic(test_node):
"""Check if an if-condition AST could produce an MLIR Value at runtime.

Calls to RewriteBoolOps helpers (dsl_not_, dsl_and_, dsl_or_) and
Python builtins are NOT considered dynamic — they just wrap Python-level
boolean logic. Only calls to user/MLIR functions can produce Values.
Calls to Python builtins are NOT considered dynamic. BoolOp helpers like
dsl_and_/dsl_or_ may still produce dynamic predicates, so they should
not be filtered out here.
"""
for child in ast.walk(test_node):
if isinstance(child, ast.Call):
Expand Down
Loading