Skip to content

Commit be48553

Browse files
author
zhangyue
committed
fix(ci): add Ascend toolkit environment variables to CI Dockerfile
1 parent f1ab525 commit be48553

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

.ci/images/ascend/Dockerfile

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,4 +18,12 @@ RUN pip install --no-cache-dir --progress off \
1818
pytest-xdist \
1919
ruff
2020

21+
ENV ASCEND_TOOLKIT_HOME=/usr/local/Ascend/ascend-toolkit/latest
22+
ENV LD_LIBRARY_PATH=${ASCEND_TOOLKIT_HOME}/lib64:${ASCEND_TOOLKIT_HOME}/lib64/plugin/opskernel:${ASCEND_TOOLKIT_HOME}/lib64/plugin/nnengine:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/aarch64:${LD_LIBRARY_PATH}
23+
ENV PYTHONPATH=${ASCEND_TOOLKIT_HOME}/python/site-packages:${ASCEND_TOOLKIT_HOME}/opp/built-in/op_impl/ai_core/tbe:${PYTHONPATH}
24+
ENV PATH=${ASCEND_TOOLKIT_HOME}/bin:${ASCEND_TOOLKIT_HOME}/compiler/ccec_compiler/bin:${PATH}
25+
ENV ASCEND_AICPU_PATH=${ASCEND_TOOLKIT_HOME}
26+
ENV ASCEND_OPP_PATH=${ASCEND_TOOLKIT_HOME}/opp
27+
ENV TOOLCHAIN_HOME=${ASCEND_TOOLKIT_HOME}/toolkit
28+
2129
WORKDIR /workspace

0 commit comments

Comments
 (0)