Skip to content

Commit b31c062

Browse files
committed
mlc: adjust config for FlashInfer 0.2.5 support
1 parent eddb2ed commit b31c062

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

packages/llm/mlc/config.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,6 @@ def mlc(commit, patch=None, version='0.1', tvm='0.15', llvm=17, depends=[], requ
3838
mlc('385cef2', 'patches/385cef2.diff', version='0.1.4', tvm='0.19.0', requires='>=36'), # 12/14/2024
3939
mlc('cf7ae82', 'patches/cf7ae82.diff', version='0.19.0', tvm='0.19.0', requires='>=36'), # 01/09/2025
4040
# Thor-specific: commit = (SM110)
41-
mlc('b3dea5581f28ce029bc776b3b6d299582e4562a4', 'patches/empty.diff', version='thor', tvm='0.20.0', requires='>=36', depends=['tvm:builder', 'flashinfer:0.2.6.post1'], default=(L4T_VERSION.major >= 36 and 110 in CUDA_ARCHITECTURES)),
42-
mlc('d2118b3', 'patches/d2118b3.diff', version='0.20.0', tvm='0.20.0', requires='>=36', depends=['tvm', 'flashinfer:0.2.6.post1']), # 5/1/2025
41+
mlc('b3dea5581f28ce029bc776b3b6d299582e4562a4', 'patches/empty.diff', version='thor', tvm='0.20.0', requires='>=36', depends=['tvm:builder', 'flashinfer:0.2.5'], default=(L4T_VERSION.major >= 36)),
42+
mlc('d2118b3', 'patches/d2118b3.diff', version='0.20.0', tvm='0.20.0', requires='>=36', depends=['tvm', 'flashinfer:0.2.5']), # 5/1/2025
4343
]

0 commit comments

Comments
 (0)