🌐 AI搜索 & 代理 主页
Skip to content

[GraphPartition] cache get_free_symbol_uses #166338

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
BoyuanFeng wants to merge 14 commits into from
Closed

[GraphPartition] cache get_free_symbol_uses #166338

Changes from 1 commit
Commits
Show all changes
14 commits
Select commit Hold shift + click to select a range
0361c22
cache get_free_symbol_uses
BoyuanFeng Oct 27, 2025
46110df
lint
BoyuanFeng Oct 27, 2025
47d229f
Merge branch 'main' into bf/partition-cache-free-symbols
BoyuanFeng Oct 27, 2025
ab23224
fix ConcatKernel
BoyuanFeng Oct 27, 2025
e4bc3e1
add obj id to cache key
BoyuanFeng Oct 27, 2025
1bb964f
Merge branch 'main' into bf/partition-cache-free-symbols
BoyuanFeng Oct 28, 2025
7c59352
check that FlexibleLayout does not change free symbol uses
BoyuanFeng Oct 28, 2025
dbc76f6
use exec to make cache faster
BoyuanFeng Oct 28, 2025
261e6eb
nit
BoyuanFeng Oct 28, 2025
261b4c1
lint
BoyuanFeng Oct 28, 2025
565ff78
Merge branch 'main' into bf/partition-cache-free-symbols
BoyuanFeng Oct 29, 2025
48172eb
nit
BoyuanFeng Oct 29, 2025
f8b2bb5
store on self.key
BoyuanFeng Oct 29, 2025
0d670c8
Merge branch 'main' into bf/partition-cache-free-symbols
BoyuanFeng Oct 31, 2025
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
fix ConcatKernel
  • Loading branch information
@BoyuanFeng
BoyuanFeng committed Oct 27, 2025
commit ab232246e3e905fa0f5e557cea24a4a41bc1602c
2 changes: 1 addition & 1 deletion torch/_inductor/ir.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5529,7 +5529,7 @@ def can_realize_into_without_copy(
and not isinstance(src.data, ExternKernelAlloc)
)

@cache_on_self
@cache_on_self_and_args
def get_free_symbol_uses(
self, unbacked_only: bool = False
) -> OrderedSet[sympy.Symbol]:
Expand Down
Loading