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.

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

clean up imports of jit_ext.py #1588

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

clean up imports of jit_ext.py #1588

wants to merge 2 commits into from

Conversation

crcrpar
Copy link
Collaborator

@crcrpar crcrpar commented Dec 27, 2024

What does this PR do?

some cleaning of jit_ext's imports

crcrpar and others added 2 commits January 3, 2025 01:21
@crcrpar crcrpar force-pushed the crpa/jit-ext-imports branch from d3254d5 to d4b35e7 Compare January 2, 2025 16:21
@@ -1671,7 +1634,7 @@ def is_variableified_tensorproxy(v: Variable | Proxy) -> Proxy:

prim(*args)

cache_info = thunder._get_cache_info()
cache_info = _get_cache_info()
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

What's the advantage here?
To my mind, the original version is clearer because you immediately see that it is a function from a different module (and which). I can see how the shorter way is important for things that get used a lot, but this seems distributing the information that this is a utility function from the thunder/__init__.py int two lines that are 300 lines apart.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants