Skip to content

Commit f24e55f

Browse files
committed
Merge PR dusty-nv#1320: MLC Thor support, TVM source-build flow, and CUDA 13 updates
2 parents b20dbc9 + 7b784f1 commit f24e55f

File tree

2 files changed

+5
-3
lines changed

2 files changed

+5
-3
lines changed

packages/llm/mlc/Dockerfile

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,9 +20,10 @@ ARG TVM_VERSION \
2020
TMP_DIR=/tmp/mlc
2121

2222
ENV LD_LIBRARY_PATH="/usr/local/lib/python${PYTHON_VERSION}/dist-packages/tvm:/opt/venv/lib/python${PYTHON_VERSION}/site-packages/tvm:${LD_LIBRARY_PATH}" \
23-
TVM_HOME=${SOURCE_DIR}/3rdparty/tvm \
24-
TVM_SOURCE_DIR=${SOURCE_DIR}/3rdparty/tvm \
25-
MLC_LLM_SOURCE_DIR=${SOURCE_DIR}
23+
MLC_LLM_SOURCE_DIR=${SOURCE_DIR} \
24+
PYTHONPATH=${SOURCE_DIR}/python:${PYTHONPATH}
25+
26+
RUN printf '\n# MLC convenience alias\nalias mlc_llm="python -m mlc_llm"\n' >> /etc/bash.bashrc
2627

2728
COPY build.sh install.sh benchmark.py wheels/ ${TMP_DIR}/
2829
COPY ${MLC_PATCH} ${TMP_DIR}/patch.diff

packages/ml/tvm/Dockerfile

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,5 @@ WORKDIR /
3838

3939
# export TVM envs for downstream stages/users
4040
ENV TVM_HOME=/opt/tvm \
41+
TVM_SOURCE_DIR=/opt/tvm \
4142
PYTHONPATH=/opt/tvm/python:${PYTHONPATH}

0 commit comments

Comments
 (0)