[Fix] Fix a type that make wrong T.macro backtrace

[kurisu6912]

This pr fix a typo in ast rewrite process, and make the backtrace more readable in T.macro. Without this patch, tilelang generates a backtrace always at the Line 1 of the file when using an undefined variable.

@T.macro
def foo():
    xxx
@T.prim_func
def tmp():
    foo()

[github-actions]

👋 Hi! Thank you for contributing to the TileLang project.

Please remember to run pre-commit run --all-files in the root directory of the project to ensure your changes are properly linted and formatted. This will help ensure your contribution passes the format check.

We appreciate you taking this step! Our team will review your contribution, and we look forward to your awesome work! 🚀

[coderabbitai]

Walkthrough

Two minor changes to the tilelang module: reorganized imports in the version-detection block to rely on implicit module loading, and modified the visit_Name method in the AST handler to pass the actual node object instead of its identifier string as a parameter.

Changes

Cohort / File(s)	Change Summary
Import refactoring tilelang/__init__.py	Replaced explicit version_provider import with direct import of dynamic_metadata from the module, enabling implicit module loading
AST node parameter modification tilelang/language/v2/ast.py	Changed visit_Name Load context handler to pass the AST node object itself to __tb.rval instead of passing the identifier string value

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

• tilelang/init.py: Verify that the implicit import behavior correctly handles version detection without side effects
• tilelang/language/v2/ast.py: Confirm that __tb.rval correctly handles the node object parameter and that this change doesn't break downstream logic that may have expected a string identifier

Poem

A rabbit hops through code so neat,
Imports refined, the flow's complete,
AST nodes dance in proper form,
Changes small, but oh so warm!
🐰 ✨

Pre-merge checks and finishing touches

❌ Failed checks (1 warning)

Check name	Status	Explanation	Resolution
Docstring Coverage	⚠️ Warning	Docstring coverage is 0.00% which is insufficient. The required threshold is 80.00%.	You can run @coderabbitai generate docstrings to improve docstring coverage.

✅ Passed checks (2 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title directly addresses fixing a typo that causes incorrect T.macro backtraces, which aligns with the actual code changes modifying AST node handling in the backtrace logic.

✨ Finishing touches

• 📝 Generate docstrings

🧪 Generate unit tests (beta)

• Create PR with unit tests
• Post copyable unit tests in a comment

---

📜 Recent review details

Configuration used: CodeRabbit UI

Review profile: CHILL

Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 454a9df and 93ac1f4.

📒 Files selected for processing (2)

• tilelang/__init__.py (0 hunks)
• tilelang/language/v2/ast.py (1 hunks)

💤 Files with no reviewable changes (1)

• tilelang/init.py

⏰ Context from checks skipped due to timeout of 90000ms. You can increase the timeout in your CodeRabbit configuration to a maximum of 15 minutes (900000ms). (1)

• GitHub Check: Quick Lint

🔇 Additional comments (1)

tilelang/language/v2/ast.py (1)

554-557: LGTM! Excellent fix for preserving source location information.

By passing the original AST node object instead of just the identifier string, this change preserves the lineno, col_offset, end_lineno, and end_col_offset attributes, which are essential for accurate error backtraces. The old implementation would have created a new Name node without this metadata, making it impossible to provide precise source locations in T.macro error messages.

---

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

• X
• Mastodon
• Reddit
• LinkedIn

_{Comment @coderabbitai help to get the list of available commands and usage tips.}