[Bugfix] Fix tvm import path for editable build

[LeiWang1999]

This pull request makes two targeted updates to improve compatibility with recent changes in the TVM library and its Python bindings. The most important changes are:

TVM Python binding path update:

• Changed the logic in tilelang/env.py to set the TVM Python binding path to third_party/tvm/python instead of just third_party/tvm, ensuring the correct directory is used for imports.

API usage update for TVM module inspection:

• Updated the test in testing/python/language/test_tilelang_language_let.py to use mod.mod.imports[0].inspect_source() instead of mod.mod.imported_modules[0].get_source(), reflecting changes in the TVM API.

Summary by CodeRabbit

• Chores
  • Updated internal test inspection method for imported module source
  • Refined Python path handling in environment configuration

[coderabbitai]

Caution

Review failed

The pull request is closed.

Walkthrough

Modified TileLang Python path handling in env.py by deriving TVM path from THIRD_PARTY_ROOT/tvm/python instead of THIRD_PARTY_ROOT/tvm. Updated corresponding test to use the new imports[0].inspect_source() API instead of imported_modules[0].get_source() for inspecting module source.

Changes

Cohort / File(s)	Summary
TVM Python path handling tilelang/env.py	Modified TVM Python path derivation to use THIRD_PARTY_ROOT/tvm/python instead of THIRD_PARTY_ROOT/tvm; removed separate tvm_python_binding variable and updated PYTHONPATH prepending logic.
Test API update testing/python/language/test_tilelang_language_let.py	Changed module source inspection from mod.mod.imported_modules[0].get_source() to mod.mod.imports[0].inspect_source() while preserving test logic and assertions.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8–12 minutes

• Verify that the TVM path derivation (THIRD_PARTY_ROOT/tvm/python) correctly points to Python bindings and maintains compatibility with downstream consumers
• Confirm that the new API surface mod.mod.imports[0].inspect_source() is the correct, supported method and behaves equivalently to the previous approach

Possibly related PRs

• tilelang#1086: Also modifies import-path handling in tilelang/env.py to prepend path entries, indicating broader refactoring of path management logic.

Poem

🐰 With tails held high and whiskers bright,
We hop through paths that feel so right,
The Python bindings find their way,
Through /python paths, hip-hip-hooray!
Old APIs fade to memories past,
New inspect_source() calls will last!

✨ 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 13bdcd6 and 445107f.

📒 Files selected for processing (2)

• testing/python/language/test_tilelang_language_let.py (1 hunks)
• tilelang/env.py (1 hunks)

---

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.}

[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! 🚀