File tree Expand file tree Collapse file tree 2 files changed +5
-3
lines changed
Expand file tree Collapse file tree 2 files changed +5
-3
lines changed Original file line number Diff line number Diff line change @@ -20,9 +20,10 @@ ARG TVM_VERSION \
2020 TMP_DIR=/tmp/mlc
2121
2222ENV 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\n alias mlc_llm="python -m mlc_llm"\n ' >> /etc/bash.bashrc
2627
2728COPY build.sh install.sh benchmark.py wheels/ ${TMP_DIR}/
2829COPY ${MLC_PATCH} ${TMP_DIR}/patch.diff
Original file line number Diff line number Diff line change @@ -38,4 +38,5 @@ WORKDIR /
3838
3939# export TVM envs for downstream stages/users
4040ENV TVM_HOME=/opt/tvm \
41+ TVM_SOURCE_DIR=/opt/tvm \
4142 PYTHONPATH=/opt/tvm/python:${PYTHONPATH}
You can’t perform that action at this time.
0 commit comments