34 lines
1.5 KiB
Bash
34 lines
1.5 KiB
Bash
# Setup hook for ascend-toolkit (sourced when package is in buildInputs)
|
|
addAscendEnv() {
|
|
local toolkit="${1:?ascend-toolkit path required}"
|
|
|
|
export ASCEND_TOOLKIT_HOME="${toolkit}"
|
|
export ASCEND_HOME_PATH="${toolkit}"
|
|
export ASCEND_OPP_PATH="${toolkit}/opp"
|
|
export ASCEND_AICPU_PATH="${toolkit}"
|
|
export TOOLCHAIN_HOME="${toolkit}/toolkit"
|
|
|
|
addToSearchPath PATH "${toolkit}/bin"
|
|
addToSearchPath PATH "${toolkit}/tools/ccec_compiler/bin"
|
|
addToSearchPath PATH "${toolkit}/tools/profiler/bin"
|
|
addToSearchPath PATH "${toolkit}/tools/ascend_system_advisor/asys"
|
|
addToSearchPath PATH "${toolkit}/tools/show_kernel_debug_data"
|
|
addToSearchPath PATH "${toolkit}/tools/msobjdump"
|
|
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64/plugin/opskernel"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64/plugin/nnengine"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/x86_64"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/tools/aml/lib64"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/tools/aml/lib64/plugin"
|
|
addToSearchPath LD_LIBRARY_PATH "${toolkit}/devlib"
|
|
|
|
addToSearchPath PYTHONPATH "${toolkit}/python/site-packages"
|
|
addToSearchPath PYTHONPATH "${toolkit}/opp/built-in/op_impl/ai_core/tbe"
|
|
|
|
addToSearchPath CMAKE_PREFIX_PATH "${toolkit}/lib64/cmake"
|
|
addToSearchPath CMAKE_PREFIX_PATH "${toolkit}/toolkit/tools/tikicpulib/lib/cmake"
|
|
}
|
|
|
|
addAscendEnv "@ASCEND_TOOLKIT@"
|