Skip to content

Make tilelang a Linux-only dependency (like triton)#4469

Merged
lvhan028 merged 1 commit intomainfrom
copilot/fix-issue-4459
Mar 26, 2026
Merged

Make tilelang a Linux-only dependency (like triton)#4469
lvhan028 merged 1 commit intomainfrom
copilot/fix-issue-4459

Conversation

Copy link
Contributor

Copilot AI commented Mar 26, 2026

Motivation

tilelang was added as an unconditional dependency in 0.12.2, but it is painful to build on Windows and requires a C++ compiler at runtime. It is only used by CUDA-specific SSM kernels (GatedDeltaRule, causal_conv1d) which are irrelevant on non-Linux platforms.

Modification

Add the same platform marker to tilelang in requirements/runtime_cuda.txt that already governs triton:

tilelang; sys_platform == "linux" and "aarch64" not in platform_machine and "arm" not in platform_machine

The Python call-sites already gate on has_tilelang(), so no runtime behaviour changes on Linux.

BC-breaking (Optional)

No. Windows and ARM Linux users were already unable to use tilelang-backed kernels; this just stops the install from failing.

Use cases (Optional)

N/A

Checklist

  1. Pre-commit or other linting tools are used to fix the potential lint issues.
  2. The modification is covered by complete unit tests. If not, please add more unit tests to ensure the correctness.
  3. If the modification has a dependency on downstream projects of a newer version, this PR should be tested with all supported versions of downstream projects.
  4. The documentation has been modified accordingly, like docstring or example tutorials.

@lvhan028 lvhan028 linked an issue Mar 26, 2026 that may be closed by this pull request
@lvhan028 lvhan028 marked this pull request as ready for review March 26, 2026 05:02
Copilot AI review requested due to automatic review settings March 26, 2026 05:02
Copy link
Contributor

Copilot AI left a comment

Choose a reason for hiding this comment

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

Pull request overview

Updates CUDA runtime dependencies so tilelang is only installed on supported platforms (Linux, non-ARM), matching the existing triton platform marker behavior and preventing install/build failures on Windows and ARM.

Changes:

  • Add a Linux/non-ARM PEP 508 environment marker to the tilelang dependency in requirements/runtime_cuda.txt.

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.

@lvhan028 lvhan028 force-pushed the copilot/fix-issue-4459 branch from 836cf4f to 2b950a4 Compare March 26, 2026 09:56
@lvhan028 lvhan028 requested a review from grimoire March 26, 2026 10:34
@lvhan028 lvhan028 merged commit 9df6a7e into main Mar 26, 2026
6 checks passed
@lvhan028 lvhan028 deleted the copilot/fix-issue-4459 branch March 26, 2026 11:48
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Issue] lmdeploy 0.12.2 has a dependency to 'tilelang'

4 participants